X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/b02fc7b2227e923beb4d47daee7949c94b90da5f..1e03ea4765bee6db9331a1265c8abf2c3e45caff:/services/nodemanager/tests/testutil.py diff --git a/services/nodemanager/tests/testutil.py b/services/nodemanager/tests/testutil.py index df8c7c7d79..555144c4d0 100644 --- a/services/nodemanager/tests/testutil.py +++ b/services/nodemanager/tests/testutil.py @@ -55,7 +55,7 @@ def cloud_object_mock(name_id, **extra): def cloud_node_fqdn(node): # We intentionally put the FQDN somewhere goofy to make sure tested code is # using this function for lookups. - return node.extra.get('testname', 'NoTestName') + return node.extra.get('testname', node.name+'.NoTestName.invalid') def ip_address_mock(last_octet): return '10.20.30.{}'.format(last_octet) @@ -80,7 +80,7 @@ class MockShutdownTimer(object): class MockSize(object): def __init__(self, factor): self.id = 'z{}.test'.format(factor) - self.name = self.id + self.name = 'test size '+self.id self.ram = 128 * factor self.disk = factor # GB self.scratch = 1000 * factor # MB @@ -123,7 +123,10 @@ class ActorTestMixin(object): pykka.ActorRegistry.stop_all() def stop_proxy(self, proxy): - return proxy.actor_ref.stop(timeout=self.TIMEOUT) + th = proxy.get_thread().get() + t = proxy.actor_ref.stop(timeout=self.TIMEOUT) + th.join() + return t def wait_for_assignment(self, proxy, attr_name, unassigned=None, timeout=TIMEOUT):