fix(test): assign workspaces to admin & annotator only

argilla-io · Jul 13, 2023 · 08a9a64 · 08a9a64
1 parent 5341a40
commit 08a9a64
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/client/sdk/v1/workspaces/test_api.py b/tests/client/sdk/v1/workspaces/test_api.py
@@ -39,7 +39,7 @@ async def test_get_workspace(role: UserRole) -> None:
 @pytest.mark.parametrize("role", [UserRole.owner, UserRole.admin, UserRole.annotator])
 async def test_list_workspaces_me(role: UserRole) -> None:
     workspaces = await WorkspaceFactory.create_batch(size=5)
-    user = await UserFactory.create(role=role, workspaces=workspaces)
+    user = await UserFactory.create(role=role, workspaces=workspaces if role != UserRole.owner else [])
 
     httpx_client = ArgillaSingleton.init(api_key=user.api_key).http_client.httpx
 

diff --git a/tests/server/api/v1/test_workspaces.py b/tests/server/api/v1/test_workspaces.py
@@ -81,7 +81,7 @@ async def test_get_workspace_with_nonexistent_workspace_id(client: TestClient, o
 @pytest.mark.parametrize("role", [UserRole.owner, UserRole.admin, UserRole.annotator])
 async def test_list_workspaces_me(client: TestClient, role: UserRole) -> None:
     workspaces = await WorkspaceFactory.create_batch(size=5)
-    user = await UserFactory.create(role=role, workspaces=workspaces)
+    user = await UserFactory.create(role=role, workspaces=workspaces if role != UserRole.owner else [])
 
     response = client.get("/api/v1/me/workspaces", headers={API_KEY_HEADER_NAME: user.api_key})