test(backend): convert existing app suites to unittest

apps/time_entries/tests/test_services.py

@@ -1,78 +1,87 @@
 from datetime import timedelta
 
-import pytest
+from django.test import TestCase
 from django.utils import timezone
 from rest_framework.exceptions import ValidationError
 
 from apps.projects.models import Project
 from apps.tags.models import Tag
-from apps.time_entries.services.time_entries import create_time_entry, stop_time_entry, update_time_entry
+from apps.time_entries.services.time_entries import (
+    create_time_entry,
+    stop_time_entry,
+    update_time_entry,
+)
 from apps.users.models import User
 from apps.workspaces.models import Workspace
 
 
-@pytest.fixture
-def workspace_owner(db):
-    user = User.objects.create_user(mobile="09121111111", password="secret123")
-    workspace = Workspace.objects.create(name="Core", owner=user)
-    return user, workspace
+class TimeEntryServiceTests(TestCase):
+    @classmethod
+    def setUpTestData(cls):
+        cls.user = User.objects.create_user(mobile="09121111111", password="secret123")
+        cls.workspace = Workspace.objects.create(name="Core", owner=cls.user)
 
-
-def test_create_time_entry_allows_only_one_running_timer_per_workspace(workspace_owner):
-    user, workspace = workspace_owner
-
-    create_time_entry(
-        user=user,
-        workspace_id=workspace.id,
-        start_time=timezone.now(),
-    )
-
-    with pytest.raises(ValidationError):
+    def test_create_time_entry_allows_only_one_running_timer_per_workspace(self):
         create_time_entry(
-            user=user,
-            workspace_id=workspace.id,
-            start_time=timezone.now() + timedelta(minutes=5),
+            user=self.user,
+            workspace_id=self.workspace.id,
+            start_time=timezone.now(),
         )
 
+        with self.assertRaises(ValidationError):
+            create_time_entry(
+                user=self.user,
+                workspace_id=self.workspace.id,
+                start_time=timezone.now() + timedelta(minutes=5),
+            )
 
-def test_stop_time_entry_sets_end_time_and_duration(workspace_owner):
-    user, workspace = workspace_owner
-    entry = create_time_entry(
-        user=user,
-        workspace_id=workspace.id,
-        start_time=timezone.now() - timedelta(hours=1),
-    )
+    def test_stop_time_entry_sets_end_time_and_duration(self):
+        entry = create_time_entry(
+            user=self.user,
+            workspace_id=self.workspace.id,
+            start_time=timezone.now() - timedelta(hours=1),
+        )
 
-    stopped_entry = stop_time_entry(entry, end_time=timezone.now())
+        stopped_entry = stop_time_entry(entry, end_time=timezone.now())
 
-    assert stopped_entry.end_time is not None
-    assert stopped_entry.duration is not None
+        self.assertIsNotNone(stopped_entry.end_time)
+        self.assertIsNotNone(stopped_entry.duration)
 
+    def test_update_time_entry_preserves_deleted_project_and_tags(self):
+        project = Project.objects.create(workspace=self.workspace, name="Deleted project")
+        tag = Tag.objects.create(
+            workspace=self.workspace,
+            name="Deleted tag",
+            color="#0f172a",
+        )
+        entry = create_time_entry(
+            user=self.user,
+            workspace_id=self.workspace.id,
+            start_time=timezone.now() - timedelta(hours=1),
+            end_time=timezone.now(),
+            project=project,
+            tags=[tag],
+            description="Before delete",
+        )
 
-def test_update_time_entry_preserves_deleted_project_and_tags(workspace_owner):
-    user, workspace = workspace_owner
-    project = Project.objects.create(workspace=workspace, name="Deleted project")
-    tag = Tag.objects.create(workspace=workspace, name="Deleted tag", color="#0f172a")
-    entry = create_time_entry(
-        user=user,
-        workspace_id=workspace.id,
-        start_time=timezone.now() - timedelta(hours=1),
-        end_time=timezone.now(),
-        project=project,
-        tags=[tag],
-        description="Before delete",
-    )
+        project.delete()
+        tag.delete()
 
-    project.delete()
-    tag.delete()
+        updated_entry = update_time_entry(
+            entry,
+            project=Project.all_objects.get(id=project.id),
+            tags=[Tag.all_objects.get(id=tag.id)],
+            description="After delete",
+        )
 
-    updated_entry = update_time_entry(
-        entry,
-        project=Project.all_objects.get(id=project.id),
-        tags=[Tag.all_objects.get(id=tag.id)],
-        description="After delete",
-    )
-
-    assert updated_entry.description == "After delete"
-    assert updated_entry.project_id == project.id
-    assert list(Tag.all_objects.filter(time_entries=updated_entry).values_list("id", flat=True)) == [tag.id]
+        self.assertEqual(updated_entry.description, "After delete")
+        self.assertEqual(updated_entry.project_id, project.id)
+        self.assertEqual(
+            list(
+                Tag.all_objects.filter(time_entries=updated_entry).values_list(
+                    "id",
+                    flat=True,
+                )
+            ),
+            [tag.id],
+        )