mirror of
https://github.com/arcan1s/ahriman.git
synced 2025-07-13 22:15:48 +00:00
50 lines
1.5 KiB
Python
50 lines
1.5 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.views.service.add import AddView
|
|
|
|
|
|
async def test_get_permission() -> None:
|
|
"""
|
|
must return correct permission for the request
|
|
"""
|
|
for method in ("POST",):
|
|
request = pytest.helpers.request("", "", method)
|
|
assert await AddView.get_permission(request) == UserAccess.Write
|
|
|
|
|
|
async def test_post(client: TestClient, mocker: MockerFixture) -> None:
|
|
"""
|
|
must call post request correctly
|
|
"""
|
|
add_mock = mocker.patch("ahriman.core.spawn.Spawn.packages_add")
|
|
response = await client.post("/service-api/v1/add", json={"packages": ["ahriman"]})
|
|
|
|
assert response.ok
|
|
add_mock.assert_called_with(["ahriman"], now=True)
|
|
|
|
|
|
async def test_post_exception(client: TestClient, mocker: MockerFixture) -> None:
|
|
"""
|
|
must raise exception on missing packages payload
|
|
"""
|
|
add_mock = mocker.patch("ahriman.core.spawn.Spawn.packages_add")
|
|
response = await client.post("/service-api/v1/add")
|
|
|
|
assert response.status == 400
|
|
add_mock.assert_not_called()
|
|
|
|
|
|
async def test_post_update(client: TestClient, mocker: MockerFixture) -> None:
|
|
"""
|
|
must call post request correctly for alias
|
|
"""
|
|
add_mock = mocker.patch("ahriman.core.spawn.Spawn.packages_add")
|
|
response = await client.post("/service-api/v1/update", json={"packages": ["ahriman"]})
|
|
|
|
assert response.ok
|
|
add_mock.assert_called_with(["ahriman"], now=True)
|