def test_database_initializing(self, task_register_mock): DatabaseHelper.create(status=Database.INITIALIZING) task_register_mock.return_value = self.task_history check_databases_status() self.assertEqual(self.task_history.task_status, 'SUCCESS') self.assertIn('All databases were checked.', self.task_history.details)
def test_database_alive(self, task_register_mock): DatabaseHelper.create(status=Database.ALIVE, _quantity=4) task_register_mock.return_value = self.task_history check_databases_status() self.assertEqual(self.task_history.task_status, 'SUCCESS') self.assertIn('All databases are alive.', self.task_history.details)
def test_database_dead(self, check_database_is_alive, task_register_mock): database = DatabaseHelper.create(status=Database.DEAD) task_register_mock.return_value = self.task_history check_databases_status() check_database_is_alive.assert_called_with(database) self.assertEqual(self.task_history.task_status, 'SUCCESS') self.assertIn('All databases were checked.', self.task_history.details)
def test_databases_with_different_status(self, task_register_mock): DatabaseHelper.create(status=Database.INITIALIZING) DatabaseHelper.create(status=Database.ALIVE) database_alert = DatabaseHelper.create(status=Database.ALERT) database_dead = DatabaseHelper.create(status=Database.DEAD) task_register_mock.return_value = self.task_history check_databases_status() self.assertEqual(self.task_history.task_status, 'ERROR') self.assertIn('Alert', self.task_history.details) self.assertIn('Dead', self.task_history.details) self.assertIn(database_dead.name, self.task_history.details) self.assertIn(database_alert.name, self.task_history.details)
def test_database_dead(self, task_register_mock): DatabaseHelper.create( name=seq('test'), status=Database.DEAD, _quantity=4 ) task_register_mock.return_value = self.task_history check_databases_status() self.assertEqual(self.task_history.task_status, 'ERROR') self.assertIn('Dead', self.task_history.details) self.assertIn('test1', self.task_history.details) self.assertIn('test2', self.task_history.details) self.assertIn('test3', self.task_history.details) self.assertIn('test4', self.task_history.details)
def test_databases_with_different_status(self, check_database_is_alive, task_register_mock): DatabaseHelper.create(status=Database.INITIALIZING) DatabaseHelper.create(status=Database.ALIVE) database_alert = DatabaseHelper.create(status=Database.ALERT) database_dead = DatabaseHelper.create(status=Database.DEAD) task_register_mock.return_value = self.task_history calls = [call(database_dead), call(database_alert)] check_databases_status() check_database_is_alive.assert_has_calls(calls) self.assertEqual(self.task_history.task_status, 'SUCCESS') self.assertIn('All databases were checked.', self.task_history.details)