class TestEPUHarness(object):
def setup(self):
self.amqp_uri = "memory://test-epuharness"
self.pidantic_dir = tempfile.mkdtemp()
self.epuharness = EPUHarness(amqp_uri=self.amqp_uri,
pidantic_dir=self.pidantic_dir)
self.epuharness._setup_factory()
def test_start_pd(self):
pd_name = "testpd"
engines = {}
assert not self.epuharness.factory.reload_instances()
self.epuharness._start_process_dispatcher(pd_name,
engines,
exe_name="echo")
assert len(self.epuharness.factory.reload_instances()) == 1
def test_start_eeagent(self):
ee_name = "testeeagent"
pd_name = "testpd"
node_name = "somenode"
launch_type = "fork"
assert not self.epuharness.factory.reload_instances()
self.epuharness._start_eeagent(ee_name,
pd_name,
node_name,
launch_type,
exe_name="echo")
assert len(self.epuharness.factory.reload_instances()) == 1
def test_announce_node(self):
raise SkipTest("TODO")
got_announce = False
def dt_state(self):
pass
dashi = self.epuharness.dashi
raise Exception("TODO")
def teardown(self):
instances = self.epuharness.factory.reload_instances()
for instance in instances.values():
instance.cleanup()
self.epuharness.factory.terminate()
shutil.rmtree(self.pidantic_dir)
class TestEPUHarness(object):
def setup(self):
self.amqp_uri = "memory://test-epuharness"
self.pidantic_dir = tempfile.mkdtemp()
self.epuharness = EPUHarness(amqp_uri=self.amqp_uri, pidantic_dir=self.pidantic_dir)
self.epuharness._setup_factory()
def test_start_pd(self):
pd_name = "testpd"
engines = {}
assert not self.epuharness.factory.reload_instances()
self.epuharness._start_process_dispatcher(pd_name, engines, exe_name="echo")
assert len(self.epuharness.factory.reload_instances()) == 1
def test_start_eeagent(self):
ee_name = "testeeagent"
pd_name = "testpd"
node_name = "somenode"
launch_type = "fork"
assert not self.epuharness.factory.reload_instances()
self.epuharness._start_eeagent(ee_name, pd_name, node_name,
launch_type, exe_name="echo")
assert len(self.epuharness.factory.reload_instances()) == 1
def test_announce_node(self):
raise SkipTest("TODO")
got_announce = False
def dt_state(self):
pass
dashi = self.epuharness.dashi
raise Exception("TODO")
def teardown(self):
instances = self.epuharness.factory.reload_instances()
for instance in instances.values():
instance.cleanup()
self.epuharness.factory.terminate()
shutil.rmtree(self.pidantic_dir)
