def test_start_already_started(self, scheduler): descriptor = Mock(main=Mock(threads=4)) scheduler.return_value.isAlive.return_value = True # test plugin = Plugin(descriptor, '') plugin.attach = Mock() plugin.start() # validation self.assertFalse(plugin.attach.called) self.assertFalse(scheduler.return_value.start.called)
def test_start(self, scheduler): descriptor = Mock(main=Mock(threads=4)) scheduler.return_value.isAlive.return_value = False # test plugin = Plugin(descriptor, '') plugin.attach = Mock() plugin.start() # validation plugin.attach.assert_called_once_with() scheduler.return_value.start.assert_called_once_with()
def test_attach(self, pool, model, consumer, node): queue = 'test' descriptor = Mock(main=Mock(threads=4)) pool.return_value.run.side_effect = lambda fn: fn() model.return_value.queue = queue # test plugin = Plugin(descriptor, '') plugin.authenticator = Mock() plugin.detach = Mock() plugin.refresh = Mock() plugin.attach() # validation plugin.detach.assert_called_once_with(False) model.assert_called_with(plugin) model.return_value.setup.assert_called_once_with() node.assert_called_once_with(queue) consumer.assert_called_once_with(node.return_value, plugin) consumer = consumer.return_value consumer.start.assert_called_once_with() self.assertEqual(consumer.authenticator, plugin.authenticator) self.assertEqual(plugin.consumer, consumer)