38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
|
import pytest
|
||
|
|
from starlette.responses import PlainTextResponse
|
||
|
|
from starlette.testclient import TestClient
|
||
|
|
from halfapi.lib.routes import route_acl_decorator
|
||
|
|
from halfapi.lib import acl
|
||
|
|
|
||
|
|
def test_acl_Check(dummy_app, token_debug_false_builder):
|
||
|
|
"""
|
||
|
|
A request with ?check should always return a 200 status code
|
||
|
|
"""
|
||
|
|
|
||
|
|
@route_acl_decorator(params=[{'acl':acl.public}])
|
||
|
|
async def test_route_public(request, **kwargs):
|
||
|
|
raise Exception('Should not raise')
|
||
|
|
return PlainTextResponse('ok')
|
||
|
|
|
||
|
|
dummy_app.add_route('/test_public', test_route_public)
|
||
|
|
test_client = TestClient(dummy_app)
|
||
|
|
|
||
|
|
resp = test_client.get('/test_public?check')
|
||
|
|
assert resp.status_code == 200
|
||
|
|
|
||
|
|
@route_acl_decorator(params=[{'acl':acl.private}])
|
||
|
|
async def test_route_private(request, **kwargs):
|
||
|
|
raise Exception('Should not raise')
|
||
|
|
return PlainTextResponse('ok')
|
||
|
|
|
||
|
|
dummy_app.add_route('/test_private', test_route_private)
|
||
|
|
test_client = TestClient(dummy_app)
|
||
|
|
|
||
|
|
resp = test_client.get('/test_private')
|
||
|
|
assert resp.status_code == 401
|
||
|
|
|
||
|
|
resp = test_client.get('/test_private?check')
|
||
|
|
assert resp.status_code == 200
|
||
|
|
|
||
|
|
|