meadsteve / lagom

tests/integrations/test_flask.py

from flask import Flask

from lagom import injectable, Container
from lagom.integrations.flask import FlaskIntegration


class ComplexDep:
    def __init__(self, message):
        self.message = message


def test_flask_container_provides_a_route_decorator(container: Container):
    app = Flask(__name__)
    container[ComplexDep] = ComplexDep("hello from dep")

The variable ComplexDep does not seem to be defined.

    flask_integration = FlaskIntegration(app, container)

    @flask_integration.route("/")
    def _some_handler(dep: ComplexDep = injectable):
        return dep.message

    with app.test_client() as client:
        resp = client.get("/")
        assert resp.get_data(as_text=True) == "hello from dep"


def test_the_route_decorator_can_have_request_level_singletons(container: Container):
    app = Flask(__name__)
    container[ComplexDep] = lambda: ComplexDep("hello from dep")

The variable ComplexDep does not seem to be defined.

    flask_integration = FlaskIntegration(
        app, container, request_singletons=[ComplexDep]

The variable ComplexDep does not seem to be defined.

    )

    @flask_integration.route("/")
    def _some_handler(
        dep_one: ComplexDep = injectable, dep_two: ComplexDep = injectable
    ):
        return "singleton" if dep_one is dep_two else "nope"

    with app.test_client() as client:
        resp = client.get("/")
        assert resp.get_data(as_text=True) == "singleton"