tests/test_middleware.py

import pytest
from starlette.testclient import TestClient


@pytest.mark.parametrize(
    "test_server_path, test_client_path, method, status_code, user, response_body", [
        ('/dataset1/resource2', '/dataset1/resource2', 'GET', 200, 'alice', 'ok'),
        ('/dataset1/resource2', '/dataset1/resource2', 'GET', 403, 'notalice', 'Forbidden'),
        ('/dataset1/resource2', '/dataset1/resource2', 'OPTIONS', 200, 'notalice', 'ok'),
        ('/dataset1/resource1', '/dataset1/resource1', 'POST', 200, 'alice', 'ok'),
    ]
)
def test_middleware_authed(app_fixture, test_server_path, test_client_path, method, status_code, user, response_body):
    @getattr(app_fixture, method.lower())(test_server_path)
    async def index():
        return 'ok'

    test_client = TestClient(app_fixture)

    test_response = getattr(test_client, method.lower())(test_client_path, auth=(user, 'password'))

    assert test_response.status_code == status_code
    assert test_response.json() == response_body


@pytest.mark.parametrize(
    "test_server_path, test_client_path, method, status_code, response_body", [
        ('/login', '/login', 'GET', 200, 'ok'),
        ('/', '/', 'GET', 200, 'ok')
    ]
)
def test_middleware_not_authed(app_fixture, test_server_path, test_client_path, method, status_code, response_body):
    @getattr(app_fixture, method.lower())(test_server_path)
    async def index():
        return 'ok'

    test_client = TestClient(app_fixture)

    test_response = getattr(test_client, method.lower())(test_client_path)

    assert test_response.status_code == status_code
    assert test_response.json() == response_body


if __name__ == '__main__':
    pytest.main()