87 lines
4.0 KiB
Python
87 lines
4.0 KiB
Python
from django.test import TestCase, RequestFactory
|
|
from django.contrib.auth import get_user_model
|
|
from unittest.mock import patch
|
|
from rest_framework.test import APIClient
|
|
|
|
from modules.task_management.views import TaskViewSet
|
|
from modules.task_management.serializers import TaskSerializer
|
|
from modules.task_management.models import Task, TaskTemplate, TaskTemplateEntry
|
|
from modules.fleet.models import Vehicle
|
|
|
|
|
|
class TaskViewSetTests(TestCase):
|
|
def setUp(self):
|
|
User = get_user_model()
|
|
self.user = User.objects.create_user(username="u1", email="u1@example.test", password="pass")
|
|
self.factory = RequestFactory()
|
|
self.view = TaskViewSet()
|
|
self.view.request = self.factory.post("/")
|
|
self.view.request.user = self.user
|
|
self.vehicle = Vehicle.objects.create(registration_number="ZG-TV-01", asset_type='crane')
|
|
self.api_client = APIClient()
|
|
self.api_client.force_authenticate(user=self.user)
|
|
|
|
@patch("modules.task_management.views.TaskService.create_task_entry")
|
|
def test_perform_create_calls_service(self, mock_create):
|
|
data = {"title": "Task from view", "description": "", "status": "aktivan", "vehicle": str(self.vehicle.id)}
|
|
serializer = TaskSerializer(data=data, context={"request": self.view.request})
|
|
assert serializer.is_valid(), serializer.errors
|
|
|
|
# Call perform_create, which should delegate to TaskService.create_task_entry
|
|
self.view.perform_create(serializer)
|
|
mock_create.assert_called_once()
|
|
args, kwargs = mock_create.call_args
|
|
assert "data" in kwargs and kwargs["user"] == self.user
|
|
|
|
@patch("modules.task_management.views.TaskService.update_task_entry")
|
|
def test_perform_update_calls_service_and_saves(self, mock_update):
|
|
# Create an instance and use serializer to update
|
|
task = Task.objects.create(title="Old", description="", assigned_to=self.user)
|
|
data = {"title": "Updated", "description": "new", "status": "aktivan"}
|
|
serializer = TaskSerializer(instance=task, data=data, partial=True, context={"request": self.view.request})
|
|
assert serializer.is_valid(), serializer.errors
|
|
|
|
# attach view.request and instance then call perform_update
|
|
self.view.request = self.factory.put("/")
|
|
self.view.request.user = self.user
|
|
serializer.instance = task
|
|
|
|
self.view.perform_update(serializer)
|
|
mock_update.assert_called_once_with(instance=task, data=serializer.validated_data, user=self.view.request.user)
|
|
# confirm instance saved with new title
|
|
task.refresh_from_db()
|
|
assert task.title == "Updated"
|
|
|
|
def test_perform_destroy_soft_deletes(self):
|
|
task = Task.objects.create(title="ToDelete", description="", assigned_to=self.user)
|
|
self.view.perform_destroy(task)
|
|
task.refresh_from_db()
|
|
assert task.is_active is False
|
|
|
|
def test_template_created_task_is_visible_in_service_records_endpoint(self):
|
|
template = TaskTemplate.objects.create(
|
|
code='template-api-service-records',
|
|
title='Template API',
|
|
description='Opis',
|
|
status='aktivan',
|
|
)
|
|
TaskTemplateEntry.objects.create(
|
|
task_template=template,
|
|
sort_order=1,
|
|
title='Korak API',
|
|
work_description='Opis API koraka',
|
|
)
|
|
|
|
response = self.api_client.post(
|
|
"/api/tasks/tasks/",
|
|
data={"template_id": str(template.id), "vehicle": str(self.vehicle.id)},
|
|
format="json",
|
|
)
|
|
self.assertEqual(response.status_code, 201, response.content)
|
|
created_task_id = str(response.json()["id"])
|
|
|
|
service_records_response = self.api_client.get("/api/fleet/service-records/")
|
|
self.assertEqual(service_records_response.status_code, 200, service_records_response.content)
|
|
payload = service_records_response.json()
|
|
rows = payload if isinstance(payload, list) else payload.get("results", [])
|
|
self.assertTrue(any(str(row.get("task") or "") == created_task_id for row in rows)) |