def test_engine_entry_log_failed_with_exception(self):
# noinspection PyTypeChecker
db_logger = DbLoggerWrapper(ExecuteLogManager())
finish_time = datetime.now(pytz.utc)
exception_message = u'Something Not Specified'
failed_message = u'Execute Failed'
db_logger.started(finish_time)
try:
raise Exception(exception_message)
except:
db_logger.failed(failed_message, *sys.exc_info())
db_logger.finished(finish_time, None)
with DBSession() as db:
execute = db.query(Execute).first()
db.expunge(execute)
assert execute.finish_time == finish_time
assert execute.status == 'finished'
assert execute.failed_message is None
with DBSession() as db:
entries = db.query(ExecuteLog).all()
db.expunge_all()
assert len(entries) == 1
assert exception_message in entries[0].message
assert failed_message in entries[0].message
assert 'failed' == entries[0].level
def test_engine_entry_log_mixed(self):
# noinspection PyTypeChecker
db_logger = DbLoggerWrapper(ExecuteLogManager())
finish_time = datetime.now(pytz.utc)
message1 = u'Inf 1'
message2 = u'Downloaded 1'
message3 = u'Failed 1'
db_logger.started(finish_time)
entry_time = datetime.now(pytz.utc)
db_logger.info(message1)
db_logger.downloaded(message2, None)
db_logger.failed(message3)
db_logger.finished(finish_time, None)
with DBSession() as db:
execute = db.query(Execute).first()
db.expunge(execute)
self.assertEqual(execute.finish_time, finish_time)
self.assertEqual(execute.status, 'finished')
self.assertIsNone(execute.failed_message)
with DBSession() as db:
entries = db.query(ExecuteLog).all()
db.expunge_all()
self.assertEqual(len(entries), 3)
self.assertEqual(entries[0].message, message1)
self.assertEqual(entries[1].message, message2)
self.assertEqual(entries[2].message, message3)
# 1 seconds is enought precision for call and log results
self.assertAlmostEqual(entries[0].time,
entry_time,
delta=timedelta(seconds=1))
self.assertAlmostEqual(entries[1].time,
entry_time,
delta=timedelta(seconds=1))
self.assertAlmostEqual(entries[2].time,
entry_time,
delta=timedelta(seconds=1))
self.assertEqual(entries[0].level, 'info')
self.assertEqual(entries[1].level, 'downloaded')
self.assertEqual(entries[2].level, 'failed')
self.assertEqual(entries[0].execute_id, execute.id)
self.assertEqual(entries[1].execute_id, execute.id)
self.assertEqual(entries[2].execute_id, execute.id)
def test_manual_execute_shouldnt_reset_timeout_for_whole_execute(self):
executed = Event()
# noinspection PyUnusedLocal
def execute(*args, **kwargs):
executed.set()
execute_mock = Mock(side_effect=execute)
mock_tracker = Mock()
mock_tracker.get_topics = Mock(return_value=[Topic()])
mock_tracker.execute = execute_mock
self.trackers_manager.trackers = {'mock.tracker': mock_tracker}
with DBSession() as db:
db.add(ExecuteSettings(interval=1, last_execute=None))
self.create_runner()
sleep(0.5)
# start manual
self.engine_runner.execute([1, 2, 3])
executed.wait(0.3)
executed.clear()
sleep(0.5)
executed.wait(0.3)
self.assertTrue(executed.is_set)
self.stop_runner()
self.assertEqual(2, execute_mock.call_count)
def test_last_execute_set_should_update_persisted_value(self):
# arrange
last_execute_expected = datetime.now(pytz.utc)
self.create_runner()
self.assertIsNone(self.engine_runner.last_execute)
# act
self.engine_runner.last_execute = last_execute_expected
self.stop_runner()
# assert
with DBSession() as db:
settings = db.query(ExecuteSettings).first()
self.assertEqual(settings.last_execute, last_execute_expected)
def test_engine_entry_finished(self):
# noinspection PyTypeChecker
db_logger = DbLoggerWrapper(ExecuteLogManager())
finish_time = datetime.now(pytz.utc)
db_logger.started(finish_time)
db_logger.finished(finish_time, None)
with DBSession() as db:
execute = db.query(Execute).first()
db.expunge(execute)
self.assertEqual(execute.finish_time, finish_time)
self.assertEqual(execute.status, 'finished')
self.assertIsNone(execute.failed_message)
def test_interval_set_should_update_timer_and_persisted_value(
self, expected_value, create_timer_mock):
# arrange
cancel_mock = Mock()
create_timer_mock.return_value = cancel_mock
self.create_runner()
# act
self.engine_runner.interval = expected_value
self.stop_runner()
# assert
self.assertEqual(2, create_timer_mock.call_count)
self.assertEqual(2, cancel_mock.call_count)
with DBSession() as db:
settings = db.query(ExecuteSettings).first()
self.assertEqual(settings.interval, expected_value)
def test_engine_entry_log_multiple_executes(self):
# noinspection PyTypeChecker
db_logger = DbLoggerWrapper(ExecuteLogManager())
finish_time_1 = datetime.now(pytz.utc)
finish_time_2 = finish_time_1 + timedelta(seconds=10)
message1 = u'Inf 1'
message2 = u'Downloaded 1'
message3 = u'Failed 1'
message4 = u'Failed 2'
exception = Exception('Some exception message')
db_logger.started(finish_time_1)
entry_time_1 = datetime.now(pytz.utc)
db_logger.info(message1)
db_logger.downloaded(message2, None)
db_logger.failed(message3)
db_logger.finished(finish_time_1, None)
db_logger.started(finish_time_2)
entry_time_2 = datetime.now(pytz.utc)
db_logger.failed(message4)
db_logger.finished(finish_time_2, exception)
with DBSession() as db:
executes = db.query(Execute).all()
execute1 = executes[0]
execute2 = executes[1]
db.expunge_all()
self.assertEqual(execute1.finish_time, finish_time_1)
self.assertEqual(execute1.status, 'finished')
self.assertIsNone(execute1.failed_message)
self.assertEqual(execute2.finish_time, finish_time_2)
self.assertEqual(execute2.status, 'failed')
self.assertEqual(execute2.failed_message, str(exception))
with DBSession() as db:
entries = db.query(ExecuteLog).all()
db.expunge_all()
self.assertEqual(len(entries), 4)
self.assertEqual(entries[0].message, message1)
self.assertEqual(entries[1].message, message2)
self.assertEqual(entries[2].message, message3)
self.assertEqual(entries[3].message, message4)
# 1 seconds is enought precision for call and log results
self.assertAlmostEqual(entries[0].time,
entry_time_1,
delta=timedelta(seconds=1))
self.assertAlmostEqual(entries[1].time,
entry_time_1,
delta=timedelta(seconds=1))
self.assertAlmostEqual(entries[2].time,
entry_time_1,
delta=timedelta(seconds=1))
self.assertAlmostEqual(entries[3].time,
entry_time_2,
delta=timedelta(seconds=1))
self.assertEqual(entries[0].level, 'info')
self.assertEqual(entries[1].level, 'downloaded')
self.assertEqual(entries[2].level, 'failed')
self.assertEqual(entries[3].level, 'failed')
self.assertEqual(entries[0].execute_id, execute1.id)
self.assertEqual(entries[1].execute_id, execute1.id)
self.assertEqual(entries[2].execute_id, execute1.id)
self.assertEqual(entries[3].execute_id, execute2.id)