class ConnectorTestCase(TestCase):
@patch('alamo_worker.connector.SchedulerConnection.open',
MagicMock(return_value=None))
def setUp(self):
self.address = settings.SCHEDULERS[0]
self.scheduler = SchedulerConnection(self.address)
@patch('alamo_worker.connector.Context.socket')
@patch('alamo_worker.connector.Poller')
def test_load_plugins_with_correct_plugin_list(self, poller_mock,
socket_mock):
poller, socket_mocked = MagicMock(), MagicMock()
poller_mock.return_value = poller
socket_mock.return_value = socket_mocked
self.scheduler.open()
socket_mocked.connect.assert_called_with(self.address)
poller.register.assert_called_with(socket_mocked, POLLIN)
def test_str_and_repr(self):
self.assertEqual(str(self.scheduler), self.address)
self.assertEqual(
repr(self.scheduler), 'Scheduler({})'.format(self.address)
)
@override_settings(SCHEDULER_MONITOR=True)
@patch('alamo_worker.connector.Context.socket')
@patch('alamo_worker.connector.Poller')
def test_(self, poller_mock, socket_mock):
poller, monitor_poller = MagicMock(), MagicMock()
sock = MagicMock()
poller_mock.side_effect = [poller, monitor_poller]
socket_mock.return_value = sock
self.scheduler.open()
self.assertTrue(sock.get_monitor_socket.called)
self.assertTrue(monitor_poller.register)
@data(False, True)
def test_connector_close_method(self, monitor):
self.scheduler.pull = MagicMock()
self.scheduler.poller = MagicMock()
self.scheduler.monitor_socket = MagicMock()
self.scheduler.monitor_poller = MagicMock()
with override_settings(SCHEDULER_MONITOR=monitor):
self.scheduler.close()
self.assertTrue(self.scheduler.pull.close.called)
self.assertTrue(self.scheduler.poller.unregister.called)
self.assertEqual(self.scheduler.monitor_socket.close.called,
monitor)
self.assertEqual(self.scheduler.monitor_poller.unregister.called,
monitor)
def setUp(self):
self.address = settings.SCHEDULERS[0]
self.scheduler = SchedulerConnection(self.address)
