def test_get_block(self):
p = Process("proc", MagicMock())
p.process_block["remoteBlocks"].set_value(['name1'])
b1 = p.get_block("name1")
self.assertEqual(b1.status, "Waiting for connection...")
self.assertEqual(p.get_block("name1"), b1)
b2 = Block()
b2.set_process_path(p, ("name2", ))
c = MagicMock()
p.add_block(b2, c)
self.assertEqual(p.get_block("name2"), b2)
self.assertEqual(p.get_controller("name2"), c)
def test_get_block(self):
p = Process("proc", MagicMock())
p.process_block["remoteBlocks"].set_value(['name1'])
b1 = p.get_block("name1")
self.assertEqual(b1.status, "Waiting for connection...")
self.assertEqual(p.get_block("name1"), b1)
b2 = Block()
b2.set_process_path(p, ("name2",))
c = MagicMock()
p.add_block(b2, c)
self.assertEqual(p.get_block("name2"), b2)
self.assertEqual(p.get_controller("name2"), c)
class TestProcess(unittest.TestCase):
def setUp(self):
self.o = Process("proc")
self.o.start()
def tearDown(self):
self.o.stop(timeout=1)
def test_init(self):
assert self.o.name == "proc"
def test_add_controller(self):
controller = MagicMock()
self.o.add_controller("mri", controller)
assert self.o.get_controller("mri") == controller
def test_init_controller(self):
class InitController(Controller):
init = False
@Process.Init
def do_init(self):
self.init = True
c = InitController(self.o, "mri", [])
self.o.add_controller("mri", c)
assert c.init == True
def test_publish_controller(self):
class PublishController(Controller):
published = []
@Process.Publish
def do_publish(self, published):
self.published = published
c = PublishController(self.o, "mri", [])
self.o.add_controller("mri", c)
assert c.published == ["mri"]
self.o.add_controller("mri2", MagicMock())
assert c.published == ["mri", "mri2"]
self.o.add_controller("mri3", MagicMock(), False)
assert c.published == ["mri", "mri2"]
self.o.remove_controller("mri2")
assert c.published == ["mri"]
