feat: added printing module

api_v1/views/printing/test_api.py

@@ -0,0 +1,400 @@
+"""
+PrintingOrder API 测试
+"""
+from django.test import TestCase
+from rest_framework.test import APIClient
+from rest_framework import status
+from django.contrib.auth import get_user_model
+from django.contrib.auth.models import Permission
+from basic_info import models as basic_models
+from printing import models as printing_models
+
+User = get_user_model()
+
+
+class PrintingOrderAPITestCase(TestCase):
+    """测试 PrintingOrder 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',
+            job_type=basic_models.EmployeeTypeEnum.PRINTER,
+            status=basic_models.EmployeeStatusEnum.ACTIVE
+        )
+        
+        # 创建客户
+        self.customer = basic_models.Customer.objects.create(
+            merchant=self.merchant,
+            name='测试客户',
+            mobile='13900139000',
+            area='测试地区'
+        )
+        
+        # 认证用户
+        self.client.force_authenticate(user=self.user)
+        
+        # 给用户添加基础权限
+        view_perm = Permission.objects.get(codename='view_printingorder')
+        add_perm = Permission.objects.get(codename='add_printingorder')
+        change_perm = Permission.objects.get(codename='change_printingorder')
+        self.user.user_permissions.add(view_perm, add_perm, change_perm)
+    
+    def test_create_printing_order(self):
+        """测试创建印染订单"""
+        data = {
+            'customer': self.customer.id,
+            'fabric': '纯棉布料',
+            'width': '150cm',
+            'is_urgent': True,
+            'area': '广州',
+            'address': '白云区xxx',
+            'fabric_source': '客户提供',
+            'is_fabric_received': False,
+            'craft': '活性印花',
+            'description': '测试订单描述',
+            'outgoing_date': '2025-11-20',
+            'printing_warn': '注意颜色',
+            'rolling_warn': '注意温度',
+            'production_warn': '质量检查'
+        }
+        
+        response = self.client.post('/api/v1/printing-orders/', data, format='json')
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+        
+        # 检查响应数据
+        self.assertIn('fabric', response.data)
+        self.assertEqual(response.data['fabric'], '纯棉布料')
+        self.assertEqual(response.data['is_urgent'], True)
+        
+        # 验证订单已创建 - 通过fabric查找，因为响应可能不包含id
+        order = printing_models.PrintingOrder.objects.filter(fabric='纯棉布料').first()
+        self.assertIsNotNone(order)
+        self.assertEqual(order.customer.id, self.customer.id)
+        self.assertEqual(order.fabric, '纯棉布料')
+    
+    def test_list_printing_orders(self):
+        """测试获取订单列表"""
+        # 创建测试订单
+        printing_models.PrintingOrder.objects.create(
+            customer=self.customer,
+            fabric='布料1',
+            width='150cm'
+        )
+        printing_models.PrintingOrder.objects.create(
+            customer=self.customer,
+            fabric='布料2',
+            width='160cm'
+        )
+        
+        response = self.client.get('/api/v1/printing-orders/')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        # 不使用分页参数时，返回列表
+        self.assertEqual(len(response.data), 2)
+    
+    def test_retrieve_printing_order(self):
+        """测试获取订单详情"""
+        order = printing_models.PrintingOrder.objects.create(
+            customer=self.customer,
+            fabric='测试布料',
+            width='150cm',
+            craft='活性印花',
+            description='详情测试'
+        )
+        
+        response = self.client.get(f'/api/v1/printing-orders/{order.id}/')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertEqual(response.data['fabric'], '测试布料')
+        self.assertEqual(response.data['craft'], '活性印花')
+        self.assertIn('customer_name', response.data)
+        self.assertEqual(response.data['customer_name'], self.customer.name)
+    
+    def test_update_printing_order(self):
+        """测试更新订单"""
+        order = printing_models.PrintingOrder.objects.create(
+            customer=self.customer,
+            fabric='旧布料',
+            width='150cm'
+        )
+        
+        update_data = {
+            'customer': self.customer.id,
+            'fabric': '新布料',
+            'width': '160cm',
+            'is_urgent': True,
+            'craft': '更新的工艺'
+        }
+        
+        response = self.client.put(
+            f'/api/v1/printing-orders/{order.id}/',
+            update_data,
+            format='json'
+        )
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        
+        order.refresh_from_db()
+        self.assertEqual(order.fabric, '新布料')
+        self.assertEqual(order.width, '160cm')
+        self.assertEqual(order.is_urgent, True)
+    
+    def test_partial_update_printing_order(self):
+        """测试部分更新订单"""
+        order = printing_models.PrintingOrder.objects.create(
+            customer=self.customer,
+            fabric='原布料',
+            width='150cm',
+            is_urgent=False
+        )
+        
+        patch_data = {
+            'is_urgent': True,
+            'craft': '新工艺'
+        }
+        
+        response = self.client.patch(
+            f'/api/v1/printing-orders/{order.id}/',
+            patch_data,
+            format='json'
+        )
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        
+        order.refresh_from_db()
+        self.assertEqual(order.is_urgent, True)
+        self.assertEqual(order.craft, '新工艺')
+        self.assertEqual(order.fabric, '原布料')  # 未修改字段保持不变
+    
+    def test_delete_printing_order_forbidden(self):
+        """测试删除订单被禁用"""
+        # 添加删除权限以便测试destroy方法的自定义逻辑
+        delete_perm = Permission.objects.get(codename='delete_printingorder')
+        self.user.user_permissions.add(delete_perm)
+        
+        order = printing_models.PrintingOrder.objects.create(
+            customer=self.customer,
+            fabric='测试布料',
+            width='150cm'
+        )
+        
+        response = self.client.delete(f'/api/v1/printing-orders/{order.id}/')
+        self.assertEqual(response.status_code, status.HTTP_405_METHOD_NOT_ALLOWED)
+        self.assertIn('不支持删除', response.data['detail'])
+        
+        # 验证订单仍然存在
+        self.assertTrue(
+            printing_models.PrintingOrder.objects.filter(id=order.id).exists()
+        )
+    
+    def test_invalidate_printing_order(self):
+        """测试作废订单"""
+        # 添加作废权限
+        invalidate_perm = Permission.objects.get(codename='can_invalidate_printingorder')
+        self.user.user_permissions.add(invalidate_perm)
+        
+        order = printing_models.PrintingOrder.objects.create(
+            customer=self.customer,
+            fabric='测试布料',
+            width='150cm',
+            is_invalid=False
+        )
+        
+        response = self.client.post(f'/api/v1/printing-orders/{order.id}/invalidate/')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertIn('已作废', response.data['detail'])
+        
+        order.refresh_from_db()
+        self.assertTrue(order.is_invalid)
+    
+    def test_invalidate_without_permission(self):
+        """测试没有权限时作废订单失败"""
+        order = printing_models.PrintingOrder.objects.create(
+            customer=self.customer,
+            fabric='测试布料',
+            width='150cm'
+        )
+        
+        response = self.client.post(f'/api/v1/printing-orders/{order.id}/invalidate/')
+        self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
+        self.assertIn('没有权限', response.data['detail'])
+    
+    def test_activate_printing_order(self):
+        """测试恢复订单"""
+        # 添加恢复权限
+        activate_perm = Permission.objects.get(codename='can_activate_printingorder')
+        self.user.user_permissions.add(activate_perm)
+        
+        order = printing_models.PrintingOrder.objects.create(
+            customer=self.customer,
+            fabric='测试布料',
+            width='150cm',
+            is_invalid=True
+        )
+        
+        response = self.client.post(f'/api/v1/printing-orders/{order.id}/activate/')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertIn('已恢复', response.data['detail'])
+        
+        order.refresh_from_db()
+        self.assertFalse(order.is_invalid)
+    
+    def test_activate_without_permission(self):
+        """测试没有权限时恢复订单失败"""
+        order = printing_models.PrintingOrder.objects.create(
+            customer=self.customer,
+            fabric='测试布料',
+            width='150cm',
+            is_invalid=True
+        )
+        
+        response = self.client.post(f'/api/v1/printing-orders/{order.id}/activate/')
+        self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
+        self.assertIn('没有权限', response.data['detail'])
+    
+    def test_mark_fabric_received(self):
+        """测试标记布料已收"""
+        order = printing_models.PrintingOrder.objects.create(
+            customer=self.customer,
+            fabric='测试布料',
+            width='150cm',
+            is_fabric_received=False
+        )
+        
+        response = self.client.post(f'/api/v1/printing-orders/{order.id}/mark_fabric_received/')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertIn('已标记布料已收', response.data['detail'])
+        
+        order.refresh_from_db()
+        self.assertTrue(order.is_fabric_received)
+    
+    def test_mark_fabric_received_already_received(self):
+        """测试重复标记布料已收"""
+        order = printing_models.PrintingOrder.objects.create(
+            customer=self.customer,
+            fabric='测试布料',
+            width='150cm',
+            is_fabric_received=True
+        )
+        
+        response = self.client.post(f'/api/v1/printing-orders/{order.id}/mark_fabric_received/')
+        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
+        self.assertIn('已经标记', response.data['detail'])
+    
+    def test_filter_by_customer(self):
+        """测试按客户过滤"""
+        customer2 = basic_models.Customer.objects.create(
+            merchant=self.merchant,
+            name='客户2',
+            mobile='13900139001'
+        )
+        
+        printing_models.PrintingOrder.objects.create(
+            customer=self.customer,
+            fabric='布料1',
+            width='150cm'
+        )
+        printing_models.PrintingOrder.objects.create(
+            customer=customer2,
+            fabric='布料2',
+            width='160cm'
+        )
+        
+        response = self.client.get(f'/api/v1/printing-orders/?customer={self.customer.id}')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertEqual(len(response.data), 1)
+    
+    def test_filter_by_urgent(self):
+        """测试按紧急状态过滤"""
+        printing_models.PrintingOrder.objects.create(
+            customer=self.customer,
+            fabric='布料1',
+            width='150cm',
+            is_urgent=True
+        )
+        printing_models.PrintingOrder.objects.create(
+            customer=self.customer,
+            fabric='布料2',
+            width='160cm',
+            is_urgent=False
+        )
+        
+        response = self.client.get('/api/v1/printing-orders/?is_urgent=true')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertEqual(len(response.data), 1)
+        self.assertTrue(response.data[0]['is_urgent'])
+    
+    def test_search_by_fabric(self):
+        """测试按布料搜索"""
+        printing_models.PrintingOrder.objects.create(
+            customer=self.customer,
+            fabric='纯棉布料',
+            width='150cm'
+        )
+        printing_models.PrintingOrder.objects.create(
+            customer=self.customer,
+            fabric='涤纶布料',
+            width='160cm'
+        )
+        
+        response = self.client.get('/api/v1/printing-orders/?search=纯棉')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertEqual(len(response.data), 1)
+        self.assertIn('纯棉', response.data[0]['fabric'])
+    
+    def test_ordering(self):
+        """测试排序"""
+        order1 = printing_models.PrintingOrder.objects.create(
+            customer=self.customer,
+            fabric='布料1',
+            width='150cm'
+        )
+        order2 = printing_models.PrintingOrder.objects.create(
+            customer=self.customer,
+            fabric='布料2',
+            width='160cm'
+        )
+        
+        # 按 ID 升序
+        response = self.client.get('/api/v1/printing-orders/?ordering=id')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        results = response.data
+        self.assertEqual(results[0]['id'], order1.id)
+        self.assertEqual(results[1]['id'], order2.id)
+        
+        # 按 ID 降序
+        response = self.client.get('/api/v1/printing-orders/?ordering=-id')
+        results = response.data
+        self.assertEqual(results[0]['id'], order2.id)
+        self.assertEqual(results[1]['id'], order1.id)
+    
+    def test_human_id_generation(self):
+        """测试 human_id 自动生成"""
+        from datetime import date
+        
+        order = printing_models.PrintingOrder.objects.create(
+            customer=self.customer,
+            fabric='测试布料',
+            width='150cm'
+        )
+        
+        # human_id 格式: YYYYMMDD000001
+        self.assertIsNotNone(order.human_id)
+        self.assertTrue(len(order.human_id) >= 14)
+        today = date.today().strftime('%Y%m%d')
+        self.assertTrue(order.human_id.startswith(today))