def test_publish_controller(self):
class PublishController(Controller):
published = []
def on_hook(self, hook):
if isinstance(hook, ProcessPublishHook):
hook(self.do_publish)
@add_call_types
def do_publish(self, published):
# type: (APublished) -> None
self.published = published
class UnpublishableController(Controller):
def on_hook(self, hook):
if isinstance(hook, ProcessStartHook):
hook(self.on_start)
def on_start(self):
return UnpublishedInfo(self.mri)
c = PublishController("mri")
self.o.add_controller(c)
assert c.published == ["mri"]
self.o.add_controller(Controller(mri="mri2"))
assert c.published == ["mri", "mri2"]
self.o.add_controller(UnpublishableController("mri3"))
assert c.published == ["mri", "mri2"]
def test_publish_controller(self):
c = PublishController("mri")
self.o.add_controller(c)
assert c.published == ["mri"]
self.o.add_controller(Controller(mri="mri2"))
assert c.published == ["mri", "mri2"]
self.o.add_controller(UnpublishableController("mri3"))
assert c.published == ["mri", "mri2"]
def setUp(self):
self.c = Controller('block', MagicMock())
self.b = self.c.block
def setUp(self):
params = Controller.MethodMeta.prepare_input_map(mri="mri1")
self.c = Controller(MagicMock(), {}, params)
self.b = self.c.block
