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