prvi kod za live uporabu

backend/modules/fleet/tests/trest_travel_expenses.py

@@ -0,0 +1,30 @@
+import pytest
+from django.core.exceptions import ValidationError
+from modules.fleet.models import Vehicle, WorkOrder, VehicleServiceRecord
+
+@pytest.mark.django_db
+def test_work_order_updates_vehicle_mileage():
+    # 1. Kreiraj vozilo (koristeći tvoj model 'Vehicle')
+    vehicle = Vehicle.objects.create(
+        registration_number="ZG-123-ABC", 
+        current_mileage=5000
+    )
+    
+    # 2. Kreiraj radni/putni nalog
+    # Pretpostavljam da ti treba i 'creator' zbog ForeignKey relacije
+    from django.contrib.auth import get_user_model
+    User = get_user_model()
+    creator = User.objects.create_user(username="testuser")
+    
+    order = WorkOrder.objects.create(
+        vehicle=vehicle,
+        creator=creator,
+        start_mileage=5000,
+        end_mileage=5500,
+        purpose="Delivery"
+    )
+    
+    # 3. Nalog je već pozvao save() pri kreiranju, 
+    # pa bi vozilo trebalo biti ažurirano
+    vehicle.refresh_from_db()
+    assert vehicle.current_mileage == 5500