mirror of https://github.com/tiangolo/fastapi.git
59 lines
1.3 KiB
Python
59 lines
1.3 KiB
Python
from unittest.mock import Mock, call, patch
|
|
|
|
import pytest
|
|
from fastapi import FastAPI, Security
|
|
from fastapi.security import SecurityScopes
|
|
from fastapi.testclient import TestClient
|
|
|
|
|
|
async def security1(scopes: SecurityScopes):
|
|
return _security1(scopes.scopes)
|
|
|
|
|
|
def _security1(scopes): # pragma: no cover
|
|
pass
|
|
|
|
|
|
async def security2(scopes: SecurityScopes):
|
|
return _security2(scopes.scopes)
|
|
|
|
|
|
def _security2(scopes): # pragma: no cover
|
|
pass
|
|
|
|
|
|
async def dep3(
|
|
dep=Security(security1, scopes=["scope1"]),
|
|
dep2=Security(security2, scopes=["scope2"]),
|
|
):
|
|
return {}
|
|
|
|
|
|
@pytest.fixture
|
|
def mocks():
|
|
with patch.dict(globals(), {"_security1": Mock()}):
|
|
with patch.dict(globals(), {"_security2": Mock()}):
|
|
yield
|
|
|
|
|
|
app = FastAPI()
|
|
|
|
|
|
@app.get("/recursive_scopes")
|
|
def recursive_scopes(dep=Security(dep3, scopes=["scope3"])):
|
|
return {}
|
|
|
|
|
|
client = TestClient(app)
|
|
|
|
|
|
# issue https://github.com/tiangolo/fastapi/issues/5623
|
|
def test_recursive_scopes(mocks):
|
|
"""
|
|
Test that scope recursion properly applies. Scopes added to a dependency should propagate
|
|
and be prepended correctly to all sub-dependencies.
|
|
"""
|
|
client.get("recursive_scopes")
|
|
_security1.assert_has_calls([call(["scope3", "scope1"])])
|
|
_security2.assert_has_calls([call(["scope3", "scope2"])])
|