test(backend): convert existing app suites to unittest

apps/projects/tests/test_views.py

@@ -1,5 +1,4 @@
-import pytest
-from rest_framework.test import APIClient
+from rest_framework.test import APITestCase
 
 from apps.clients.models import Client
 from apps.projects.models import Project
@@ -7,69 +6,65 @@ from apps.users.models import User
 from apps.workspaces.models import Workspace, WorkspaceMembership
 
 
-@pytest.fixture()
-def api_client():
-    return APIClient()
+class ProjectViewTests(APITestCase):
+    @classmethod
+    def setUpTestData(cls):
+        cls.owner = User.objects.create_user(
+            mobile="09121110001",
+            password="secret123",
+            first_name="Owner",
+        )
+        cls.workspace = Workspace.objects.create(name="Projects", owner=cls.owner)
+        cls.member = User.objects.create_user(
+            mobile="09121110002",
+            password="secret123",
+            first_name="Member",
+        )
+        WorkspaceMembership.objects.create(
+            workspace=cls.workspace,
+            user=cls.member,
+            role=WorkspaceMembership.Role.MEMBER,
+            is_active=True,
+        )
+        cls.first_client = Client.objects.create(workspace=cls.workspace, name="Acme")
+        cls.second_client = Client.objects.create(workspace=cls.workspace, name="Globex")
+        cls.third_client = Client.objects.create(workspace=cls.workspace, name="Initech")
+        Project.objects.create(
+            workspace=cls.workspace,
+            client=cls.first_client,
+            name="Alpha",
+        )
+        Project.objects.create(
+            workspace=cls.workspace,
+            client=cls.second_client,
+            name="Beta",
+        )
+        Project.objects.create(
+            workspace=cls.workspace,
+            client=cls.third_client,
+            name="Gamma",
+        )
 
+    def test_project_list_supports_multi_client_filter(self):
+        self.client.force_authenticate(user=self.member)
 
-@pytest.fixture()
-def owner(db):
-    return User.objects.create_user(mobile="09121110001", password="secret123", first_name="Owner")
+        response = self.client.get(
+            "/api/projects/",
+            [
+                ("workspace", str(self.workspace.id)),
+                ("clients", str(self.first_client.id)),
+                ("clients", str(self.second_client.id)),
+            ],
+        )
 
-
-@pytest.fixture()
-def workspace(owner):
-    return Workspace.objects.create(name="Projects", owner=owner)
-
-
-@pytest.fixture()
-def member(db, workspace):
-    user = User.objects.create_user(mobile="09121110002", password="secret123", first_name="Member")
-    WorkspaceMembership.objects.create(
-        workspace=workspace,
-        user=user,
-        role=WorkspaceMembership.Role.MEMBER,
-        is_active=True,
-    )
-    return user
-
-
-@pytest.fixture()
-def clients(workspace):
-    first = Client.objects.create(workspace=workspace, name="Acme")
-    second = Client.objects.create(workspace=workspace, name="Globex")
-    third = Client.objects.create(workspace=workspace, name="Initech")
-    return first, second, third
-
-
-@pytest.fixture()
-def projects(workspace, clients):
-    first, second, third = clients
-    return [
-        Project.objects.create(workspace=workspace, client=first, name="Alpha"),
-        Project.objects.create(workspace=workspace, client=second, name="Beta"),
-        Project.objects.create(workspace=workspace, client=third, name="Gamma"),
-    ]
-
-
-def test_project_list_supports_multi_client_filter(api_client, member, workspace, clients, projects):
-    api_client.force_authenticate(user=member)
-    first, second, _ = clients
-
-    response = api_client.get(
-        "/api/projects/",
-        [
-            ("workspace", str(workspace.id)),
-            ("clients", str(first.id)),
-            ("clients", str(second.id)),
-        ],
-    )
-
-    assert response.status_code == 200
-    items = (
-        response.data
-        if isinstance(response.data, list)
-        else response.data.get("results") or response.data.get("items", [])
-    )
-    result_ids = {str(item["client"]["id"]) for item in items}
-    assert result_ids == {str(first.id), str(second.id)}
+        self.assertEqual(response.status_code, 200)
+        items = (
+            response.data
+            if isinstance(response.data, list)
+            else response.data.get("results") or response.data.get("items", [])
+        )
+        result_ids = {str(item["client"]["id"]) for item in items}
+        self.assertEqual(
+            result_ids,
+            {str(self.first_client.id), str(self.second_client.id)},
+        )