def test_event_listeners(self):
client = mock.Mock()
task_server = self.task_server
tc = TaskComputer(task_server, use_docker_manager=False)
tc.lock_config(True)
tc.lock_config(False)
listener = ClientTaskComputerEventListener(client)
tc.register_listener(listener)
tc.lock_config(True)
client.lock_config.assert_called_with(True)
tc.lock_config(False)
client.lock_config.assert_called_with(False)
def test_event_listeners(self):
client = mock.Mock()
task_server = mock.MagicMock()
task_server.config_desc = config_desc()
tc = TaskComputer("ABC", task_server, use_docker_machine_manager=False)
tc.lock_config(True)
tc.lock_config(False)
listener = ClientTaskComputerEventListener(client)
tc.register_listener(listener)
tc.lock_config(True)
client.lock_config.assert_called_with(True)
tc.lock_config(False)
client.lock_config.assert_called_with(False)