mirror of
https://github.com/arcan1s/ahriman.git
synced 2025-04-28 09:17:17 +00:00
93 lines
3.4 KiB
Python
93 lines
3.4 KiB
Python
import pytest
|
|
|
|
from aiohttp.test_utils import TestClient
|
|
from pytest_mock import MockerFixture
|
|
|
|
from ahriman.models.user_access import UserAccess
|
|
from ahriman.web.schemas.error_schema import ErrorSchema
|
|
from ahriman.web.schemas.pgp_key_id_schema import PGPKeyIdSchema
|
|
from ahriman.web.schemas.pgp_key_schema import PGPKeySchema
|
|
from ahriman.web.views.service.pgp import PGPView
|
|
|
|
|
|
async def test_get_permission() -> None:
|
|
"""
|
|
must return correct permission for the request
|
|
"""
|
|
for method in ("GET",):
|
|
request = pytest.helpers.request("", "", method)
|
|
assert await PGPView.get_permission(request) == UserAccess.Reporter
|
|
for method in ("POST",):
|
|
request = pytest.helpers.request("", "", method)
|
|
assert await PGPView.get_permission(request) == UserAccess.Full
|
|
|
|
|
|
async def test_get(client: TestClient, mocker: MockerFixture) -> None:
|
|
"""
|
|
must retrieve key from the keyserver
|
|
"""
|
|
import_mock = mocker.patch("ahriman.core.sign.gpg.GPG.key_download", return_value="imported")
|
|
request_schema = PGPKeyIdSchema()
|
|
response_schema = PGPKeySchema()
|
|
|
|
payload = {"key": "0xdeadbeaf", "server": "keyserver.ubuntu.com"}
|
|
assert not request_schema.validate(payload)
|
|
response = await client.get("/api/v1/service/pgp", params=payload)
|
|
assert response.ok
|
|
import_mock.assert_called_once_with("keyserver.ubuntu.com", "0xdeadbeaf")
|
|
assert not response_schema.validate(await response.json())
|
|
assert await response.json() == {"key": "imported"}
|
|
|
|
|
|
async def test_get_empty(client: TestClient, mocker: MockerFixture) -> None:
|
|
"""
|
|
must raise 400 on missing parameters
|
|
"""
|
|
import_mock = mocker.patch("ahriman.core.sign.gpg.GPG.key_download")
|
|
response_schema = ErrorSchema()
|
|
|
|
response = await client.get("/api/v1/service/pgp")
|
|
assert response.status == 400
|
|
assert not response_schema.validate(await response.json())
|
|
import_mock.assert_not_called()
|
|
|
|
|
|
async def test_get_process_exception(client: TestClient, mocker: MockerFixture) -> None:
|
|
"""
|
|
must raise 404 on invalid PGP server response
|
|
"""
|
|
import_mock = mocker.patch("ahriman.core.sign.gpg.GPG.key_download", side_effect=Exception())
|
|
response_schema = ErrorSchema()
|
|
|
|
response = await client.get("/api/v1/service/pgp", params={"key": "0xdeadbeaf", "server": "keyserver.ubuntu.com"})
|
|
assert response.status == 404
|
|
assert not response_schema.validate(await response.json())
|
|
import_mock.assert_called_once_with("keyserver.ubuntu.com", "0xdeadbeaf")
|
|
|
|
|
|
async def test_post(client: TestClient, mocker: MockerFixture) -> None:
|
|
"""
|
|
must call post request correctly
|
|
"""
|
|
import_mock = mocker.patch("ahriman.core.spawn.Spawn.key_import")
|
|
request_schema = PGPKeyIdSchema()
|
|
|
|
payload = {"key": "0xdeadbeaf", "server": "keyserver.ubuntu.com"}
|
|
assert not request_schema.validate(payload)
|
|
response = await client.post("/api/v1/service/pgp", json=payload)
|
|
assert response.ok
|
|
import_mock.assert_called_once_with("0xdeadbeaf", "keyserver.ubuntu.com")
|
|
|
|
|
|
async def test_post_exception(client: TestClient, mocker: MockerFixture) -> None:
|
|
"""
|
|
must raise exception on missing key payload
|
|
"""
|
|
import_mock = mocker.patch("ahriman.core.spawn.Spawn.key_import")
|
|
response_schema = ErrorSchema()
|
|
|
|
response = await client.post("/api/v1/service/pgp")
|
|
assert response.status == 400
|
|
assert not response_schema.validate(await response.json())
|
|
import_mock.assert_not_called()
|