26 lines
979 B
Python
26 lines
979 B
Python
import pytest
|
|
from django.core.exceptions import ValidationError
|
|
from modules.fleet.models import Vehicle
|
|
from django.contrib.auth import get_user_model
|
|
|
|
@pytest.mark.django_db
|
|
def test_vehicle_servicer_many_to_one_allows_multiple_vehicles(test_user):
|
|
Vehicle.objects.create(registration_number="ZG-111-AA", assigned_servicer=test_user)
|
|
|
|
Vehicle.objects.create(registration_number="ZG-222-BB", assigned_servicer=test_user)
|
|
|
|
assert Vehicle.objects.filter(assigned_servicer=test_user).count() == 2
|
|
|
|
@pytest.mark.django_db
|
|
def test_vehicle_next_service_logic():
|
|
# current=12000, interval=15000 -> next_service_at = 15000
|
|
vehicle = Vehicle.objects.create(
|
|
registration_number="ZG-001",
|
|
current_mileage=12000,
|
|
service_interval_km=15000
|
|
)
|
|
assert vehicle.next_service_at() == 15000
|
|
|
|
# current=16000, interval=15000 -> next_service_at = 30000
|
|
vehicle.current_mileage = 16000
|
|
assert vehicle.next_service_at() == 30000 |