Function simulate_postgres_error

posthog/test/test_health.py:212–219

Causes any call to cursor to raise the upper most Error in djangos db Exception hierachy

()

@contextmanager
def simulate_postgres_error():
    """
    Causes any call to cursor to raise the upper most Error in djangos db
    Exception hierachy
    """
    with patch.object(connections[DEFAULT_DB_ALIAS], "cursor") as cursor_mock:
        cursor_mock.side_effect = DjangoDatabaseError  # This should be the most general
        yield


@contextmanager