mirror of
https://github.com/arcan1s/ahriman.git
synced 2025-04-24 15:27:17 +00:00
52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
import pytest
|
|
|
|
from aiohttp.test_utils import TestClient
|
|
from aiohttp.web import HTTPUnauthorized
|
|
from pytest_mock import MockerFixture
|
|
|
|
from ahriman.models.user_access import UserAccess
|
|
from ahriman.web.views.user.logout import LogoutView
|
|
|
|
|
|
async def test_get_permission() -> None:
|
|
"""
|
|
must return correct permission for the request
|
|
"""
|
|
for method in ("POST",):
|
|
request = pytest.helpers.request("", "", method)
|
|
assert await LogoutView.get_permission(request) == UserAccess.Unauthorized
|
|
|
|
|
|
async def test_post(client_with_auth: TestClient, mocker: MockerFixture) -> None:
|
|
"""
|
|
must log out user correctly
|
|
"""
|
|
mocker.patch("aiohttp_security.check_authorized")
|
|
forget_mock = mocker.patch("aiohttp_security.forget")
|
|
|
|
response = await client_with_auth.post("/api/v1/logout")
|
|
assert response.ok
|
|
forget_mock.assert_called_once_with(pytest.helpers.anyvar(int), pytest.helpers.anyvar(int))
|
|
|
|
|
|
async def test_post_unauthorized(client_with_auth: TestClient, mocker: MockerFixture) -> None:
|
|
"""
|
|
must raise exception if unauthorized
|
|
"""
|
|
mocker.patch("aiohttp_security.check_authorized", side_effect=HTTPUnauthorized())
|
|
forget_mock = mocker.patch("aiohttp_security.forget")
|
|
response_schema = pytest.helpers.schema_response(LogoutView.post, code=401)
|
|
|
|
response = await client_with_auth.post("/api/v1/logout", headers={"accept": "application/json"})
|
|
assert response.status == 401
|
|
assert not response_schema.validate(await response.json())
|
|
forget_mock.assert_not_called()
|
|
|
|
|
|
async def test_post_disabled(client: TestClient) -> None:
|
|
"""
|
|
must raise exception if auth is disabled
|
|
"""
|
|
response = await client.post("/api/v1/logout")
|
|
assert response.ok
|