meadsteve / lagom

tests/integrations/test_fastapi.py

import pytest
from starlette.testclient import TestClient

from .fastapi_app import app, deps, Inner, ContextLoaded


def test_request_singletons_are_the_same_within_a_request_context():
    client = TestClient(app)
    response = client.get("/")
    data = response.json()
    assert data["outer_one"] == data["outer_two"]


def test_request_singletons_are_different_for_new_requests():
    client = TestClient(app)
    data_one = client.get("/").json()
    data_two = client.get("/").json()

    assert data_one["outer_one"] != data_two["outer_one"]


def test_request_singletons_also_have_request_class():
    client = TestClient(app)
    data = client.get("/request_injected_request_singleton").json()

    assert data["data"] == "/request_injected_request_singleton"


def test_deps_can_be_overridden_during_test():
    client = TestClient(app)
    with deps.override_for_test() as c:
        c[Inner] = Inner("test_message")

The variable Inner does not seem to be defined.

        call_under_test = client.get("/inner").json()
    call_after_test = client.get("/inner").json()

    assert call_under_test["data"] == "test_message"
    assert call_after_test["data"] != "test_message"


@pytest.fixture
def fixture_fake_deps():
    with deps.override_for_test() as test_container:
        test_container[Inner] = Inner("fixture_deps")

The variable Inner does not seem to be defined.

        yield test_container


def test_overriding_with_fixtures_works(fixture_fake_deps):
    client = TestClient(app)
    resp = client.get("/inner").json()
    assert resp["data"] == "fixture_deps"


def test_deps_can_be_overridden_during_test_multiple_times():
    client = TestClient(app)
    with deps.override_for_test() as c1:
        with deps.override_for_test() as c2:
            c1[Inner] = Inner("first_level")

The variable Inner does not seem to be defined.

            c2[Inner] = Inner("second_level")
            second = client.get("/inner").json()
        first = client.get("/inner").json()
    outer = client.get("/inner").json()

    assert outer["data"] is None
    assert first["data"] == "first_level"
    assert second["data"] == "second_level"


def test_deps_can_use_contexts_for_cleanup_tasks():
    ContextLoaded.cleaned_up = False
    client = TestClient(app)
    response = client.get("with_some_context")
    assert response.json() == {"cleaned_up": "False"}
    assert ContextLoaded.cleaned_up