prvi kod za live uporabu

backend/modules/task_management/tests/test_services.py

@@ -0,0 +1,53 @@
+from django.test import TestCase
+from django.contrib.auth import get_user_model
+from rest_framework.exceptions import ValidationError
+from modules.task_management.services import TaskService
+from modules.task_management.models import Task
+from modules.fleet.models import Vehicle, WorkOrder
+
+
+class TaskServiceTests(TestCase):
+    def setUp(self):
+        User = get_user_model()
+        self.user = User.objects.create_user(username="user1", email="u1@example.test", password="pass")
+        self.other_user = User.objects.create_user(username="user2", email="u2@example.test", password="pass")
+        self.staff = User.objects.create_user(username="admin", email="admin@example.test", password="pass", is_staff=True)
+
+    def test_create_task_entry_creates_task_and_assigns_user(self):
+        data = {"title": "Do work", "description": "Something", "status": "aktivan"}
+        entry = TaskService.create_task_entry(data=data, user=self.user)
+        assert isinstance(entry, Task)
+        assert entry.assigned_to == self.user
+        assert entry.title == "Do work"
+
+    def test_create_task_entry_short_title_raises(self):
+        data = {"title": "x", "description": "", "status": "aktivan"}
+        with self.assertRaises(ValidationError):
+            TaskService.create_task_entry(data=data, user=self.user)
+
+    def test_update_task_entry_permission_denied_for_non_owner(self):
+        task = Task.objects.create(title="T", description="", assigned_to=self.other_user)
+        with self.assertRaises(ValidationError):
+            TaskService.update_task_entry(instance=task, data={"title": "New"}, user=self.user)
+
+    def test_update_task_entry_allowed_for_owner(self):
+        task = Task.objects.create(title="Old", description="", assigned_to=self.user)
+        updated = TaskService.update_task_entry(instance=task, data={"title": "New"}, user=self.user)
+        assert updated.title == "New"
+
+    def test_update_task_entry_rejects_closing_without_work_order(self):
+        task = Task.objects.create(title="Za zatvaranje", description="", assigned_to=self.user, status='servis')
+        with self.assertRaises(ValidationError):
+            TaskService.update_task_entry(instance=task, data={"status": "neaktivan"}, user=self.user)
+
+    def test_update_task_entry_allows_closing_with_work_order(self):
+        vehicle = Vehicle.objects.create(registration_number="ZG-TASK-01", asset_type='crane')
+        work_order = WorkOrder.objects.create(vehicle=vehicle, creator=self.user, purpose='defektaza')
+        task = Task.objects.create(title="Za zatvaranje", description="", assigned_to=self.user, status='servis')
+        updated = TaskService.update_task_entry(
+            instance=task,
+            data={"status": "neaktivan", "work_order": work_order},
+            user=self.user,
+        )
+        assert updated.status == "neaktivan"
+        assert updated.work_order == work_order