def test_cron_jobs_clear_logs_handler__delete_data_store(self): date_tag = datetime.utcnow() handler = ClearLogsHandler() handler._ClearLogsHandler__delete_data_store(date_tag) LogModel.all().filter.assert_called_with('update_time <=', date_tag) LogModel.all().filter().fetch.assert_called_with(1000) db.delete.assert_called_with(LogModel.all().filter().fetch())
def test_cron_jobs_clear_logs_handler_get(self): ApplicationModel.all.return_value = [MagicMock()] handler = ClearLogsHandler() handler._ClearLogsHandler__delete_text_search = MagicMock() handler._ClearLogsHandler__delete_data_store = MagicMock() handler.get() self.assertTrue(ApplicationModel.all.called) self.assertTrue(handler._ClearLogsHandler__delete_text_search.called) self.assertTrue(handler._ClearLogsHandler__delete_data_store.called)