def test_cursor_error(self, mocked_connection):
mocked_conn = mock.MagicMock()
mocked_connection.return_value.__enter__.return_value = mocked_conn
mocked_conn.execute.side_effect = Exception('Connection Refused')
results = PostgresCheck.run()
assert results['POSTGRES'].is_healthy is False
assert results['POSTGRES'].severity == Result.ERROR
def get_status():
status = {}
status.update(CronsCheck.run())
status.update(EventsCheck.run())
status.update(HPSearchCheck.run())
status.update(PipelinesCheck.run())
status.update(PostgresCheck.run())
status.update(RabbitMQCheck.run())
status.update(RedisCheck.run())
status.update(SchedulerCheck.run())
status.update(StreamsCheck.run())
return {k: v.to_dict() for k, v in status.items()}
def test_bad_connection(self, mocked_connection):
mocked_connection.return_value.__enter__.return_value = None
results = PostgresCheck.run()
assert results['POSTGRES'].is_healthy is False
assert results['POSTGRES'].severity == Result.WARNING
def test_database_is_healthy(self):
results = PostgresCheck.run()
assert results['POSTGRES'].is_healthy is True