feat: mission category

basic_info/tests.py

@@ -7,7 +7,7 @@ from django.core.exceptions import ValidationError
 from .models import (
     Merchant, MerchantTypeEnum, WareHouse, WarehouseTypeEnum, 
     WareHouseModeEnum, EmployeeType, Employee, EmployeeStatusEnum,
-    FrontendPage, FrontendPageTypeEnum
+    FrontendPage, FrontendPageTypeEnum, TransportVehicle, TransportVehicleMaterialCapacity
 )
 
 
@@ -171,6 +171,45 @@ class EmployeeTypeTestCase(TestCase):
         self.assertEqual(emp_type.employees.count(), 2)
 
 
+class TransportVehicleTestCase(TestCase):
+    def setUp(self):
+        self.merchant = Merchant.objects.create(
+            name='运输商户',
+            type=MerchantTypeEnum.STORE
+        )
+
+    def test_create_transport_vehicle_with_material_capacity(self):
+        vehicle = TransportVehicle.objects.create(
+            merchant=self.merchant,
+            name='9.6米厢车',
+            license_plate='浙A12345'
+        )
+        capacity = TransportVehicleMaterialCapacity.objects.create(
+            merchant=self.merchant,
+            transport_vehicle=vehicle,
+            material_name='棉布',
+            capacity=120
+        )
+
+        self.assertEqual(vehicle.license_plate, '浙A12345')
+        self.assertEqual(str(vehicle), '9.6米厢车 (浙A12345)')
+        self.assertEqual(capacity.capacity, 120)
+        self.assertEqual(vehicle.material_capacities.count(), 1)
+
+    def test_transport_vehicle_license_plate_unique_per_merchant(self):
+        TransportVehicle.objects.create(
+            merchant=self.merchant,
+            name='一号车',
+            license_plate='浙A99999'
+        )
+        with self.assertRaises(Exception):
+            TransportVehicle.objects.create(
+                merchant=self.merchant,
+                name='二号车',
+                license_plate='浙A99999'
+            )
+
+
 class FrontendPageTestCase(TestCase):
     """测试前端页面模型"""