fix: added merchant_id to printing_order and plate_order

api_v1/views/shipment/test_api.py

@@ -0,0 +1,266 @@
+"""
+Shipment API 测试
+"""
+from decimal import Decimal
+
+from django.test import TestCase
+from django.conf import settings
+from rest_framework.test import APIClient
+from rest_framework import status
+from django.contrib.auth import get_user_model
+
+from basic_info import models as basic_models
+from printing import models as printing_models
+from shipment import models as shipment_models
+from stateflow import models as stateflow_models
+
+User = get_user_model()
+
+
+class SalesItemByPrintingOrderAPITestCase(TestCase):
+    """测试通过生产订单查询销售品 API"""
+    
+    def setUp(self):
+        self.client = APIClient()
+        
+        # 创建商户
+        self.merchant = basic_models.Merchant.objects.create(
+            name='测试印花厂',
+            type=basic_models.MerchantTypeEnum.FACTORY
+        )
+        
+        # 创建用户
+        self.user = User.objects.create_user(
+            username='testuser',
+            password='testpass123',
+            email='test@example.com'
+        )
+        
+        # 创建员工并关联商户
+        self.employee = basic_models.Employee.objects.create(
+            sys_user=self.user,
+            merchant=self.merchant,
+            name='测试员工',
+            mobile='13800138000',
+            status=basic_models.EmployeeStatusEnum.ACTIVE
+        )
+        
+        # 创建客户
+        self.customer = basic_models.Customer.objects.create(
+            merchant=self.merchant,
+            name='测试客户',
+            mobile='13900139000',
+            area='测试地区'
+        )
+        
+        # 创建流程
+        self.state1 = stateflow_models.State.objects.create(name='待印染')
+        self.state2 = stateflow_models.State.objects.create(name='印染中')
+        self.state3 = stateflow_models.State.objects.create(name='已完成')
+        
+        self.process = stateflow_models.Process.objects.create(name='印染流程')
+        self.process.replace_nodes([self.state1, self.state2, self.state3])
+        
+        # 创建产品分类
+        self.category = basic_models.ProductCategory.objects.create(
+            merchant=self.merchant,
+            name='测试分类',
+        )
+        
+        # 创建产品
+        self.product = basic_models.Product.objects.create(
+            merchant=self.merchant,
+            category=self.category,
+            name='测试产品',
+            human_id='TEST001',
+        )
+        
+        # 创建印染订单
+        self.printing_order = printing_models.PrintingOrder.objects.create(
+            merchant=self.merchant,
+            customer=self.customer,
+            fabric='测试面料',
+            width='150cm',
+            process=self.process,
+            created_by=self.user,
+        )
+        
+        # 创建印染任务
+        self.printing_job1 = printing_models.PrintingJob.objects.create(
+            merchant=self.merchant,
+            printing_order=self.printing_order,
+            product=self.product,
+            quantity=100,
+            unit='米',
+            created_by=self.user,
+        )
+        
+        self.printing_job2 = printing_models.PrintingJob.objects.create(
+            merchant=self.merchant,
+            printing_order=self.printing_order,
+            product=self.product,
+            quantity=200,
+            unit='米',
+            created_by=self.user,
+        )
+        
+        # 创建出货单
+        self.shipment = shipment_models.Shipment.objects.create(
+            customer=self.customer,
+            shipment_date='2026-01-14',
+            created_by=self.user,
+        )
+        
+        # 创建销售品 - 未关联出货单
+        self.sales_item1 = shipment_models.SalesItem.objects.create(
+            name='销售品1',
+            quantity=Decimal('50.00'),
+            unit=shipment_models.UnitChoices.METER,
+            printing_job_id=self.printing_job1.id,
+            created_by=self.user,
+        )
+        
+        self.sales_item2 = shipment_models.SalesItem.objects.create(
+            name='销售品2',
+            quantity=Decimal('30.00'),
+            unit=shipment_models.UnitChoices.METER,
+            printing_job_id=self.printing_job1.id,
+            position='A1-01',
+            remark='备注信息',
+            created_by=self.user,
+        )
+        
+        # 创建销售品 - 已关联出货单
+        self.sales_item3 = shipment_models.SalesItem.objects.create(
+            name='销售品3（已出货）',
+            quantity=Decimal('100.00'),
+            unit=shipment_models.UnitChoices.METER,
+            printing_job_id=self.printing_job2.id,
+            shipment=self.shipment,
+            created_by=self.user,
+        )
+        
+        # 创建与该订单无关的销售品
+        self.sales_item_other = shipment_models.SalesItem.objects.create(
+            name='其它销售品',
+            quantity=Decimal('999.00'),
+            unit=shipment_models.UnitChoices.PIECE,
+            printing_job_id=99999,  # 不存在的 job
+            created_by=self.user,
+        )
+        
+        # 认证用户
+        self.client.force_authenticate(user=self.user)
+    
+    def test_get_sales_items_by_printing_order_exclude_shipped(self):
+        """测试查询销售品 - 默认不包含已出货的"""
+        url = f'/api/v1/shipment/sales-items/by-printing-order/{self.printing_order.id}/'
+        response = self.client.get(url)
+        
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        data = response.json()
+        
+        # 应该只返回2个未出货的销售品
+        self.assertEqual(data['count'], 2)
+        
+        # 检查返回的销售品
+        item_ids = [item['id'] for item in data['results']]
+        self.assertIn(self.sales_item1.id, item_ids)
+        self.assertIn(self.sales_item2.id, item_ids)
+        self.assertNotIn(self.sales_item3.id, item_ids)  # 已出货的不应该在列表中
+        self.assertNotIn(self.sales_item_other.id, item_ids)  # 其它订单的也不在
+    
+    def test_get_sales_items_by_printing_order_include_shipped(self):
+        """测试查询销售品 - 包含已出货的"""
+        url = f'/api/v1/shipment/sales-items/by-printing-order/{self.printing_order.id}/?include_already_has_shipment=true'
+        response = self.client.get(url)
+        
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        data = response.json()
+        
+        # 应该返回3个销售品（包含已出货的）
+        self.assertEqual(data['count'], 3)
+        
+        # 检查返回的销售品
+        item_ids = [item['id'] for item in data['results']]
+        self.assertIn(self.sales_item1.id, item_ids)
+        self.assertIn(self.sales_item2.id, item_ids)
+        self.assertIn(self.sales_item3.id, item_ids)  # 已出货的也应该在列表中
+        self.assertNotIn(self.sales_item_other.id, item_ids)  # 其它订单的依然不在
+    
+    def test_get_sales_items_response_format(self):
+        """测试返回数据格式"""
+        url = f'/api/v1/shipment/sales-items/by-printing-order/{self.printing_order.id}/'
+        response = self.client.get(url)
+        
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        data = response.json()
+        
+        # 找到 sales_item2（包含 position 和 remark）
+        item = next(item for item in data['results'] if item['id'] == self.sales_item2.id)
+        
+        # 检查所有字段
+        self.assertEqual(item['name'], '销售品2')
+        self.assertEqual(Decimal(item['quantity']), Decimal('30.00'))
+        self.assertEqual(item['unit'], shipment_models.UnitChoices.METER)
+        self.assertEqual(item['unit_display'], '米')
+        self.assertEqual(item['position'], 'A1-01')
+        self.assertEqual(item['remark'], '备注信息')
+        self.assertEqual(item['printing_job_id'], self.printing_job1.id)
+        self.assertIsNone(item['shipment_id'])
+        self.assertIsNone(item['shipment_date'])
+        self.assertIsNotNone(item['created_at'])
+        self.assertEqual(item['created_by_id'], self.user.id)
+    
+    def test_get_sales_items_shipped_item_format(self):
+        """测试已出货的销售品返回格式"""
+        url = f'/api/v1/shipment/sales-items/by-printing-order/{self.printing_order.id}/?include_already_has_shipment=true'
+        response = self.client.get(url)
+        
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        data = response.json()
+        
+        # 找到已出货的销售品
+        item = next(item for item in data['results'] if item['id'] == self.sales_item3.id)
+        
+        # 检查出货单信息
+        self.assertEqual(item['shipment_id'], self.shipment.id)
+        self.assertEqual(item['shipment_date'], '2026-01-14')
+    
+    def test_get_sales_items_printing_order_not_found(self):
+        """测试生产订单不存在"""
+        url = '/api/v1/shipment/sales-items/by-printing-order/99999/'
+        response = self.client.get(url)
+        
+        self.assertEqual(response.status_code, status.HTTP_404_NOT_FOUND)
+        self.assertIn('不存在', response.json()['detail'])
+    
+    def test_get_sales_items_empty_result(self):
+        """测试生产订单没有关联销售品"""
+        # 创建一个没有销售品的订单
+        empty_order = printing_models.PrintingOrder.objects.create(
+            merchant=self.merchant,
+            customer=self.customer,
+            fabric='测试面料2',
+            width='150cm',
+            process=self.process,
+            created_by=self.user,
+        )
+        
+        url = f'/api/v1/shipment/sales-items/by-printing-order/{empty_order.id}/'
+        response = self.client.get(url)
+        
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        data = response.json()
+        
+        self.assertEqual(data['count'], 0)
+        self.assertEqual(data['results'], [])
+    
+    def test_get_sales_items_unauthenticated(self):
+        """测试未认证用户"""
+        self.client.logout()
+        
+        url = f'/api/v1/shipment/sales-items/by-printing-order/{self.printing_order.id}/'
+        response = self.client.get(url)
+        
+        self.assertEqual(response.status_code, status.HTTP_401_UNAUTHORIZED)