From 018b27b4cbd6aec7e77e9dba44651e46865e4a17 Mon Sep 17 00:00:00 2001 From: Victor Skvortsov Date: Tue, 25 Nov 2025 12:30:52 +0500 Subject: [PATCH] Assert not authenticated 401 --- src/tests/_internal/server/routers/test_files.py | 4 ++-- src/tests/_internal/server/routers/test_fleets.py | 14 +++++++------- .../_internal/server/routers/test_gateways.py | 2 +- .../_internal/server/routers/test_instances.py | 2 +- .../_internal/server/routers/test_prometheus.py | 2 +- src/tests/_internal/server/routers/test_volumes.py | 10 +++++----- 6 files changed, 17 insertions(+), 17 deletions(-) diff --git a/src/tests/_internal/server/routers/test_files.py b/src/tests/_internal/server/routers/test_files.py index 931a24b568..28f851e1c8 100644 --- a/src/tests/_internal/server/routers/test_files.py +++ b/src/tests/_internal/server/routers/test_files.py @@ -27,7 +27,7 @@ async def test_returns_403_if_not_authenticated(self, client: AsyncClient): "/api/files/get_archive_by_hash", json={"hash": "blob_hash"}, ) - assert response.status_code == 403 + assert response.status_code in [401, 403] async def test_returns_400_if_archive_does_not_exist( self, session: AsyncSession, client: AsyncClient @@ -81,7 +81,7 @@ async def test_returns_403_if_not_authenticated(self, client: AsyncClient): "/api/files/upload_archive", files={"file": self.file}, ) - assert response.status_code == 403 + assert response.status_code in [401, 403] async def test_returns_existing_archive( self, session: AsyncSession, client: AsyncClient, default_storage_mock: Mock diff --git a/src/tests/_internal/server/routers/test_fleets.py b/src/tests/_internal/server/routers/test_fleets.py index 934f333b63..43301aea69 100644 --- a/src/tests/_internal/server/routers/test_fleets.py +++ b/src/tests/_internal/server/routers/test_fleets.py @@ -58,7 +58,7 @@ async def test_returns_40x_if_not_authenticated( self, test_db, session: AsyncSession, client: AsyncClient ): response = await client.post("/api/fleets/list") - assert response.status_code == 403 + assert response.status_code in [401, 403] @pytest.mark.asyncio @pytest.mark.parametrize("test_db", ["sqlite", "postgres"], indirect=True) @@ -147,7 +147,7 @@ async def test_returns_40x_if_not_authenticated( self, test_db, session: AsyncSession, client: AsyncClient ): response = await client.post("/api/project/main/fleets/list") - assert response.status_code == 403 + assert response.status_code in [401, 403] @pytest.mark.asyncio @pytest.mark.parametrize("test_db", ["sqlite", "postgres"], indirect=True) @@ -188,7 +188,7 @@ async def test_returns_40x_if_not_authenticated( self, test_db, session: AsyncSession, client: AsyncClient ): response = await client.post("/api/project/main/fleets/get") - assert response.status_code == 403 + assert response.status_code in [401, 403] @pytest.mark.asyncio @pytest.mark.parametrize("test_db", ["sqlite", "postgres"], indirect=True) @@ -309,7 +309,7 @@ async def test_returns_40x_if_not_authenticated( self, test_db, session: AsyncSession, client: AsyncClient ): response = await client.post("/api/project/main/fleets/apply") - assert response.status_code == 403 + assert response.status_code in [401, 403] @pytest.mark.asyncio @pytest.mark.parametrize("test_db", ["sqlite", "postgres"], indirect=True) @@ -829,7 +829,7 @@ async def test_returns_40x_if_not_authenticated( self, test_db, session: AsyncSession, client: AsyncClient ): response = await client.post("/api/project/main/fleets/delete") - assert response.status_code == 403 + assert response.status_code in [401, 403] @pytest.mark.asyncio @pytest.mark.parametrize("test_db", ["sqlite", "postgres"], indirect=True) @@ -942,7 +942,7 @@ async def test_returns_40x_if_not_authenticated( self, test_db, session: AsyncSession, client: AsyncClient ): response = await client.post("/api/project/main/fleets/delete_instances") - assert response.status_code == 403 + assert response.status_code in [401, 403] @pytest.mark.asyncio @pytest.mark.parametrize("test_db", ["sqlite", "postgres"], indirect=True) @@ -1036,7 +1036,7 @@ async def test_returns_40x_if_not_authenticated( self, test_db, session: AsyncSession, client: AsyncClient ): response = await client.post("/api/project/main/fleets/get_plan") - assert response.status_code == 403 + assert response.status_code in [401, 403] @pytest.mark.asyncio @pytest.mark.parametrize("test_db", ["sqlite", "postgres"], indirect=True) diff --git a/src/tests/_internal/server/routers/test_gateways.py b/src/tests/_internal/server/routers/test_gateways.py index 6d06e69692..0bee1e6f06 100644 --- a/src/tests/_internal/server/routers/test_gateways.py +++ b/src/tests/_internal/server/routers/test_gateways.py @@ -26,7 +26,7 @@ async def test_returns_40x_if_not_authenticated( self, test_db, session: AsyncSession, client: AsyncClient ): response = await client.post("/api/project/main/gateways/list") - assert response.status_code == 403 + assert response.status_code in [401, 403] @pytest.mark.asyncio @pytest.mark.parametrize("test_db", ["sqlite", "postgres"], indirect=True) diff --git a/src/tests/_internal/server/routers/test_instances.py b/src/tests/_internal/server/routers/test_instances.py index 72198331d0..f4fe924e4d 100644 --- a/src/tests/_internal/server/routers/test_instances.py +++ b/src/tests/_internal/server/routers/test_instances.py @@ -265,7 +265,7 @@ async def test_pagination( async def test_not_authenticated(self, client: AsyncClient, data) -> None: resp = await client.post("/api/instances/list", json={}) - assert resp.status_code == 403 + assert resp.status_code in [401, 403] @pytest.mark.asyncio diff --git a/src/tests/_internal/server/routers/test_prometheus.py b/src/tests/_internal/server/routers/test_prometheus.py index 0678d0197e..ab9549965d 100644 --- a/src/tests/_internal/server/routers/test_prometheus.py +++ b/src/tests/_internal/server/routers/test_prometheus.py @@ -375,7 +375,7 @@ async def test_returns_403_if_not_authenticated( else: headers = None response = await client.get("/metrics", headers=headers) - assert response.status_code == 403 + assert response.status_code in [401, 403] async def test_returns_200_if_token_is_valid( self, monkeypatch: pytest.MonkeyPatch, client: AsyncClient diff --git a/src/tests/_internal/server/routers/test_volumes.py b/src/tests/_internal/server/routers/test_volumes.py index fa55e97a7f..1949747f62 100644 --- a/src/tests/_internal/server/routers/test_volumes.py +++ b/src/tests/_internal/server/routers/test_volumes.py @@ -32,7 +32,7 @@ async def test_returns_40x_if_not_authenticated( self, test_db, session: AsyncSession, client: AsyncClient ): response = await client.post("/api/volumes/list") - assert response.status_code == 403 + assert response.status_code in [401, 403] @pytest.mark.asyncio @pytest.mark.parametrize("test_db", ["sqlite", "postgres"], indirect=True) @@ -198,7 +198,7 @@ async def test_returns_40x_if_not_authenticated( self, test_db, session: AsyncSession, client: AsyncClient ): response = await client.post("/api/project/main/volumes/list") - assert response.status_code == 403 + assert response.status_code in [401, 403] @pytest.mark.asyncio @pytest.mark.parametrize("test_db", ["sqlite", "postgres"], indirect=True) @@ -249,7 +249,7 @@ async def test_returns_40x_if_not_authenticated( self, test_db, session: AsyncSession, client: AsyncClient ): response = await client.post("/api/project/main/volumes/get") - assert response.status_code == 403 + assert response.status_code in [401, 403] @pytest.mark.asyncio @pytest.mark.parametrize("test_db", ["sqlite", "postgres"], indirect=True) @@ -316,7 +316,7 @@ async def test_returns_40x_if_not_authenticated( self, test_db, session: AsyncSession, client: AsyncClient ): response = await client.post("/api/project/main/volumes/create") - assert response.status_code == 403 + assert response.status_code in [401, 403] @pytest.mark.asyncio @pytest.mark.parametrize("test_db", ["sqlite", "postgres"], indirect=True) @@ -366,7 +366,7 @@ async def test_returns_40x_if_not_authenticated( self, test_db, session: AsyncSession, client: AsyncClient ): response = await client.post("/api/project/main/volumes/delete") - assert response.status_code == 403 + assert response.status_code in [401, 403] @pytest.mark.asyncio @pytest.mark.parametrize("test_db", ["sqlite", "postgres"], indirect=True)