@ -11,11 +11,156 @@ import mitogen.parent
import ansible_mitogen . affinity
class NullFixedPolicy ( ansible_mitogen . affinity . FixedPolicy ) :
def _set_cpu_mask ( self , mask ) :
self . mask = mask
class FixedPolicyTest ( testlib . TestCase ) :
klass = NullFixedPolicy
def test_assign_controller_1core ( self ) :
# Uniprocessor .
policy = self . klass ( cpu_count = 1 )
policy . assign_controller ( )
self . assertEquals ( 0x1 , policy . mask )
def test_assign_controller_2core ( self ) :
# Small SMP gets 1.. % cpu_count
policy = self . klass ( cpu_count = 2 )
policy . assign_controller ( )
self . assertEquals ( 0x2 , policy . mask )
policy . assign_controller ( )
self . assertEquals ( 0x2 , policy . mask )
policy . assign_controller ( )
def test_assign_controller_3core ( self ) :
# Small SMP gets 1.. % cpu_count
policy = self . klass ( cpu_count = 3 )
policy . assign_controller ( )
self . assertEquals ( 0x2 , policy . mask )
policy . assign_controller ( )
self . assertEquals ( 0x4 , policy . mask )
policy . assign_controller ( )
self . assertEquals ( 0x2 , policy . mask )
policy . assign_controller ( )
self . assertEquals ( 0x4 , policy . mask )
policy . assign_controller ( )
def test_assign_controller_4core ( self ) :
# Big SMP gets a dedicated core.
policy = self . klass ( cpu_count = 4 )
policy . assign_controller ( )
self . assertEquals ( 0x2 , policy . mask )
policy . assign_controller ( )
self . assertEquals ( 0x2 , policy . mask )
def test_assign_muxprocess_1core ( self ) :
# Uniprocessor .
policy = self . klass ( cpu_count = 1 )
policy . assign_muxprocess ( )
self . assertEquals ( 0x1 , policy . mask )
def test_assign_muxprocess_2core ( self ) :
# Small SMP gets dedicated core.
policy = self . klass ( cpu_count = 2 )
policy . assign_muxprocess ( )
self . assertEquals ( 0x1 , policy . mask )
policy . assign_muxprocess ( )
self . assertEquals ( 0x1 , policy . mask )
policy . assign_muxprocess ( )
def test_assign_muxprocess_3core ( self ) :
# Small SMP gets a dedicated core.
policy = self . klass ( cpu_count = 3 )
policy . assign_muxprocess ( )
self . assertEquals ( 0x1 , policy . mask )
policy . assign_muxprocess ( )
self . assertEquals ( 0x1 , policy . mask )
def test_assign_muxprocess_4core ( self ) :
# Big SMP gets a dedicated core.
policy = self . klass ( cpu_count = 4 )
policy . assign_muxprocess ( )
self . assertEquals ( 0x1 , policy . mask )
policy . assign_muxprocess ( )
self . assertEquals ( 0x1 , policy . mask )
def test_assign_worker_1core ( self ) :
# Balance n % 1
policy = self . klass ( cpu_count = 1 )
policy . assign_worker ( )
self . assertEquals ( 0x1 , policy . mask )
policy . assign_worker ( )
self . assertEquals ( 0x1 , policy . mask )
def test_assign_worker_2core ( self ) :
# Balance n % 1
policy = self . klass ( cpu_count = 2 )
policy . assign_worker ( )
self . assertEquals ( 0x2 , policy . mask )
policy . assign_worker ( )
self . assertEquals ( 0x2 , policy . mask )
def test_assign_worker_3core ( self ) :
# Balance n % 1
policy = self . klass ( cpu_count = 3 )
policy . assign_worker ( )
self . assertEquals ( 0x2 , policy . mask )
policy . assign_worker ( )
self . assertEquals ( 0x4 , policy . mask )
policy . assign_worker ( )
self . assertEquals ( 0x2 , policy . mask )
def test_assign_worker_4core ( self ) :
# Balance n % 1
policy = self . klass ( cpu_count = 4 )
policy . assign_worker ( )
self . assertEquals ( 4 , policy . mask )
policy . assign_worker ( )
self . assertEquals ( 8 , policy . mask )
policy . assign_worker ( )
self . assertEquals ( 4 , policy . mask )
def test_assign_subprocess_1core ( self ) :
# allow all except reserved.
policy = self . klass ( cpu_count = 1 )
policy . assign_subprocess ( )
self . assertEquals ( 0x1 , policy . mask )
policy . assign_subprocess ( )
self . assertEquals ( 0x1 , policy . mask )
def test_assign_subprocess_2core ( self ) :
# allow all except reserved.
policy = self . klass ( cpu_count = 2 )
policy . assign_subprocess ( )
self . assertEquals ( 0x2 , policy . mask )
policy . assign_subprocess ( )
self . assertEquals ( 0x2 , policy . mask )
def test_assign_subprocess_3core ( self ) :
# allow all except reserved.
policy = self . klass ( cpu_count = 3 )
policy . assign_subprocess ( )
self . assertEquals ( 0x2 + 0x4 , policy . mask )
policy . assign_subprocess ( )
self . assertEquals ( 0x2 + 0x4 , policy . mask )
def test_assign_subprocess_4core ( self ) :
# allow all except reserved.
policy = self . klass ( cpu_count = 4 )
policy . assign_subprocess ( )
self . assertEquals ( 0x4 + 0x8 , policy . mask )
policy . assign_subprocess ( )
self . assertEquals ( 0x4 + 0x8 , policy . mask )
@unittest2.skipIf (
reason = ' Linux/SMP only ' ,
condition = ( not (
os . uname ( ) [ 0 ] == ' Linux ' and
multiprocessing . cpu_count ( ) > = 4
multiprocessing . cpu_count ( ) > 2
) )
)
class LinuxPolicyTest ( testlib . TestCase ) :
@ -33,12 +178,15 @@ class LinuxPolicyTest(testlib.TestCase):
finally :
fp . close ( )
def test_set_clear ( self ) :
before = self . _get_cpus ( )
self . policy . _set_cpu ( 3 )
self . assertEquals ( self . _get_cpus ( ) , 1 << 3 )
self . policy . _clear ( )
self . assertEquals ( self . _get_cpus ( ) , before )
def test_set_cpu_mask ( self ) :
self . policy . _set_cpu_mask ( 0x1 )
self . assertEquals ( 0x1 , self . _get_cpus ( ) )
self . policy . _set_cpu_mask ( 0x2 )
self . assertEquals ( 0x2 , self . _get_cpus ( ) )
self . policy . _set_cpu_mask ( 0x3 )
self . assertEquals ( 0x3 , self . _get_cpus ( ) )
def test_clear_on_popen ( self ) :
tf = tempfile . NamedTemporaryFile ( )