packagers support (#100)

tests/ahriman/web/views/service/test_views_service_add.py

@@ -2,6 +2,7 @@ import pytest
 
 from aiohttp.test_utils import TestClient
 from pytest_mock import MockerFixture
+from unittest.mock import AsyncMock
 
 from ahriman.models.user_access import UserAccess
 from ahriman.web.views.service.add import AddView
@@ -21,13 +22,16 @@ async def test_post(client: TestClient, mocker: MockerFixture) -> None:
     must call post request correctly
     """
     add_mock = mocker.patch("ahriman.core.spawn.Spawn.packages_add")
+    user_mock = AsyncMock()
+    user_mock.return_value = "username"
+    mocker.patch("ahriman.web.views.base.BaseView.username", side_effect=user_mock)
     request_schema = pytest.helpers.schema_request(AddView.post)
 
     payload = {"packages": ["ahriman"]}
     assert not request_schema.validate(payload)
     response = await client.post("/api/v1/service/add", json=payload)
     assert response.ok
-    add_mock.assert_called_once_with(["ahriman"], now=True)
+    add_mock.assert_called_once_with(["ahriman"], "username", now=True)
 
 
 async def test_post_empty(client: TestClient, mocker: MockerFixture) -> None:

tests/ahriman/web/views/service/test_views_service_rebuild.py

@@ -2,6 +2,7 @@ import pytest
 
 from aiohttp.test_utils import TestClient
 from pytest_mock import MockerFixture
+from unittest.mock import AsyncMock
 
 from ahriman.models.user_access import UserAccess
 from ahriman.web.views.service.rebuild import RebuildView
@@ -21,13 +22,16 @@ async def test_post(client: TestClient, mocker: MockerFixture) -> None:
     must call post request correctly
     """
     rebuild_mock = mocker.patch("ahriman.core.spawn.Spawn.packages_rebuild")
+    user_mock = AsyncMock()
+    user_mock.return_value = "username"
+    mocker.patch("ahriman.web.views.base.BaseView.username", side_effect=user_mock)
     request_schema = pytest.helpers.schema_request(RebuildView.post)
 
     payload = {"packages": ["python", "ahriman"]}
     assert not request_schema.validate(payload)
     response = await client.post("/api/v1/service/rebuild", json=payload)
     assert response.ok
-    rebuild_mock.assert_called_once_with("python")
+    rebuild_mock.assert_called_once_with("python", "username")
 
 
 async def test_post_exception(client: TestClient, mocker: MockerFixture) -> None:

tests/ahriman/web/views/service/test_views_service_request.py

@@ -2,6 +2,7 @@ import pytest
 
 from aiohttp.test_utils import TestClient
 from pytest_mock import MockerFixture
+from unittest.mock import AsyncMock
 
 from ahriman.models.user_access import UserAccess
 from ahriman.web.views.service.request import RequestView
@@ -21,13 +22,16 @@ async def test_post(client: TestClient, mocker: MockerFixture) -> None:
     must call post request correctly
     """
     add_mock = mocker.patch("ahriman.core.spawn.Spawn.packages_add")
+    user_mock = AsyncMock()
+    user_mock.return_value = "username"
+    mocker.patch("ahriman.web.views.base.BaseView.username", side_effect=user_mock)
     request_schema = pytest.helpers.schema_request(RequestView.post)
 
     payload = {"packages": ["ahriman"]}
     assert not request_schema.validate(payload)
     response = await client.post("/api/v1/service/request", json=payload)
     assert response.ok
-    add_mock.assert_called_once_with(["ahriman"], now=False)
+    add_mock.assert_called_once_with(["ahriman"], "username", now=False)
 
 
 async def test_post_exception(client: TestClient, mocker: MockerFixture) -> None:

tests/ahriman/web/views/service/test_views_service_update.py

@@ -1,5 +1,6 @@
 from aiohttp.test_utils import TestClient
 from pytest_mock import MockerFixture
+from unittest.mock import AsyncMock
 
 
 async def test_post_update(client: TestClient, mocker: MockerFixture) -> None:
@@ -7,7 +8,10 @@ async def test_post_update(client: TestClient, mocker: MockerFixture) -> None:
     must call post request correctly for alias
     """
     update_mock = mocker.patch("ahriman.core.spawn.Spawn.packages_update")
+    user_mock = AsyncMock()
+    user_mock.return_value = "username"
+    mocker.patch("ahriman.web.views.base.BaseView.username", side_effect=user_mock)
 
     response = await client.post("/api/v1/service/update")
     assert response.ok
-    update_mock.assert_called_once_with()
+    update_mock.assert_called_once_with("username")

tests/ahriman/web/views/test_views_base.py

@@ -2,6 +2,8 @@ import pytest
 
 from multidict import MultiDict
 from aiohttp.test_utils import TestClient
+from pytest_mock import MockerFixture
+from unittest.mock import AsyncMock
 
 from ahriman.models.user_access import UserAccess
 from ahriman.web.views.base import BaseView
@@ -146,3 +148,22 @@ async def test_head_not_allowed(client: TestClient) -> None:
     """
     response = await client.head("/api/v1/service/add")
     assert response.status == 405
+
+
+async def test_username(base: BaseView, mocker: MockerFixture) -> None:
+    """
+    must return identity of logged-in user
+    """
+    policy = AsyncMock()
+    policy.identify.return_value = "identity"
+    mocker.patch("aiohttp.web.Application.get", return_value=policy)
+
+    assert await base.username() == "identity"
+    policy.identify.assert_called_once_with(base.request)
+
+
+async def test_username_no_auth(base: BaseView) -> None:
+    """
+    must return None in case if auth is disabled
+    """
+    assert await base.username() is None