30 lines
986 B
Python
30 lines
986 B
Python
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 |