Function test_retryable_query_default_session

tests/app/test_utils.py:203–228

(caplog)

def test_retryable_query_default_session(caplog):
    mock_default_session = mock.Mock(autospec=Session)
    n = 0

    @retryable_query(default_retry_attempts=2, exception_cls=_ExampleException)
    def dummy_function(foo, bar=123, session=mock_default_session):
        assert foo == "baz"
        assert bar == 123

        session.execute(321)
        nonlocal n
        n += 1

        if n < 2:
            raise _ExampleException("example reason")

    dummy_function("baz")

    assert mock_default_session.mock_calls == [
        mock.call.execute(321),
        mock.call.rollback(),
        mock.call.execute(321),
    ]
    assert caplog.messages == [
        "Attempt 0 of query failed",
    ]


def test_retryable_query_positional_session(caplog):