32 lines
1.7 KiB
Python
32 lines
1.7 KiB
Python
from django.test import TestCase
|
|
from modules.task_management.serializers import TaskSerializer
|
|
from rest_framework.exceptions import ValidationError
|
|
from django.contrib.auth import get_user_model
|
|
from modules.task_management.models import Task
|
|
from modules.fleet.models import Vehicle, WorkOrder
|
|
|
|
|
|
class TaskSerializerTests(TestCase):
|
|
def test_validate_title_too_short_raises(self):
|
|
serializer = TaskSerializer(data={"title": "ab", "description": "", "status": "aktivan"})
|
|
with self.assertRaises(ValidationError):
|
|
serializer.is_valid(raise_exception=True)
|
|
|
|
def test_validate_title_ok(self):
|
|
serializer = TaskSerializer(data={"title": "Valid title", "description": "", "status": "aktivan"})
|
|
assert serializer.is_valid(), serializer.errors
|
|
|
|
def test_cannot_close_without_work_order(self):
|
|
serializer = TaskSerializer(data={"title": "Task 1", "description": "", "status": "neaktivan"})
|
|
with self.assertRaises(ValidationError):
|
|
serializer.is_valid(raise_exception=True)
|
|
|
|
def test_can_close_with_work_order(self):
|
|
User = get_user_model()
|
|
user = User.objects.create_user(username="task-user", email="task-user@example.test", password="pass")
|
|
vehicle = Vehicle.objects.create(registration_number="ZG-TSK-02", asset_type='crane')
|
|
work_order = WorkOrder.objects.create(vehicle=vehicle, creator=user, purpose='defektaza')
|
|
task = Task.objects.create(title="Task 2", description="", status="servis", assigned_to=user)
|
|
|
|
serializer = TaskSerializer(instance=task, data={"status": "neaktivan", "work_order": str(work_order.id)}, partial=True)
|
|
assert serializer.is_valid(), serializer.errors |