def test_executors_events_are_emitted_to_clients(self, mock_run,
mock_emit):
ui = SchedulerUI(self.scheduler)
ui.start()
# Executor addition.
self.scheduler.add_executor(ThreadPoolExecutor(max_workers=1),
alias='thread_pool')
mock_emit.assert_called_once()
self.assertEqual('executor_event', mock_emit.call_args[0][0])
emitted_event = mock_emit.call_args[0][1]
self.assertEqual('thread_pool', emitted_event['executor_name'])
self.assertEqual('executor_added', emitted_event['event_name'])
self.assertIsInstance(
datetime.strptime(emitted_event['event_ts'],
'%Y-%m-%d %H:%M:%S.%f'), datetime)
# Executor removal.
mock_emit.reset_mock()
self.scheduler.remove_executor('thread_pool')
mock_emit.assert_called_once()
emitted_event = mock_emit.call_args[0][1]
self.assertEqual('thread_pool', emitted_event['executor_name'])
self.assertEqual('executor_removed', emitted_event['event_name'])
self.assertIsInstance(
datetime.strptime(emitted_event['event_ts'],
'%Y-%m-%d %H:%M:%S.%f'), datetime)
def test_jobstore_events_are_emitted_to_clients(self, mock_run, mock_emit):
ui = SchedulerUI(self.scheduler)
ui.start()
# Job store addition.
self.scheduler.add_jobstore(MemoryJobStore(), alias='in_memory')
mock_emit.assert_called_once()
self.assertEqual('jobstore_event', mock_emit.call_args[0][0])
emitted_event = mock_emit.call_args[0][1]
self.assertEqual('in_memory', emitted_event['jobstore_name'])
self.assertEqual('jobstore_added', emitted_event['event_name'])
self.assertIsInstance(
datetime.strptime(emitted_event['event_ts'],
'%Y-%m-%d %H:%M:%S.%f'), datetime)
# Job store removal.
mock_emit.reset_mock()
self.scheduler.remove_jobstore('in_memory')
mock_emit.assert_called_once()
emitted_event = mock_emit.call_args[0][1]
self.assertEqual('in_memory', emitted_event['jobstore_name'])
self.assertEqual('jobstore_removed', emitted_event['event_name'])
self.assertIsInstance(
datetime.strptime(emitted_event['event_ts'],
'%Y-%m-%d %H:%M:%S.%f'), datetime)
def test_job_events_are_emitted_to_clients(self, mock_run, mock_emit):
ui = SchedulerUI(self.scheduler)
ui.start()
self.scheduler.add_job(lambda: time.sleep(0.1),
id='waiting_job',
name='Waiting job',
trigger='interval',
seconds=0.2,
next_run_time=datetime.now() +
timedelta(milliseconds=50))
mock_emit.assert_called_once()
emitted_event = mock_emit.call_args[0][1]
self.assertEqual('job_added', emitted_event['event_name'])
self.assertIn('properties', emitted_event)
self.assertIsInstance(
datetime.strptime(emitted_event['event_ts'],
'%Y-%m-%d %H:%M:%S.%f'), datetime)
mock_emit.reset_mock()
time.sleep(0.1)
# Job submission event.
mock_emit.assert_called_once()
def test_scheduler_events_are_emitted_to_clients(self, mock_run,
mock_emit):
ui = SchedulerUI(self.scheduler)
ui.start()
mock_run.assert_called()
# Pause scheduler.
self.scheduler.pause()
mock_emit.assert_called_once()
self.assertEqual('scheduler_paused',
mock_emit.call_args[0][1]['event_name'])
mock_emit.reset_mock()
# Resume it.
self.scheduler.resume()
self.assertEqual('scheduler_resumed',
mock_emit.call_args[0][1]['event_name'])
# Stop it.
mock_emit.reset_mock()
self.scheduler.shutdown()
self.assertEqual('scheduler_shutdown',
mock_emit.call_args[0][1]['event_name'])
# Start it again.
mock_emit.reset_mock()
self.scheduler.start()
self.assertEqual('scheduler_started',
mock_emit.call_args[0][1]['event_name'])
def test_webserver_start(self, mock_run):
ui = SchedulerUI(self.scheduler)
self.assertEqual(0, len(ui._scheduler_listener.listeners))
ui.start()
self.assertEqual(1, len(ui._scheduler_listener.listeners))
self.assertEqual(ui, ui._scheduler_listener.listeners[0],
'Webserver should register itself as listener')
# SocketIO.run should be called by the web server thread on start.
mock_run.assert_called_with(ui._web_server, host='0.0.0.0', port=5000)
def test_connected_clients_get_initialized(self, mock_emit):
ui = SchedulerUI(self.scheduler, capabilities={'run_job': True})
ui.start(port=5001, host='localhost')
time.sleep(0.1)
import socketio
socket_client = socketio.Client()
socket_client.connect('ws://localhost:5001')
socket_client.emit(
'connected'
) # Notify server that we're now connected, as frontend would do.
time.sleep(0.1)
self.assertEqual(2, mock_emit.call_count,
'emit should be called twice when a client connects')
first_call = mock_emit.call_args_list[0]
second_call = mock_emit.call_args_list[1]
self.assertEqual(
'init_jobs', first_call[0][0],
'First argument of the first emit should be event name')
self.assertEqual(
'init_capabilities', second_call[0][0],
'First argument of the second emit shoud be the init_capabilities event name'
)
self.assertEqual(
ui.capabilities, second_call[0][1],
"Second argument of init_capabilities should equal the web server's capabilities"
)
socket_client.disconnect()