feat: shipment customer aou

2026-03-30 22:03:14 +08:00
parent 38f40644a3
commit aff10b8925
7 changed files with 690 additions and 6 deletions
--- a/api_v1/views/shipment/test_api.py
+++ b/api_v1/views/shipment/test_api.py
@@ -6,7 +6,7 @@ from decimal import Decimal

 from django.test import TestCase
 from django.conf import settings
-from rest_framework.test import APIClient
+from rest_framework.test import APIClient, APITestCase
 from rest_framework import status
 from django.contrib.auth import get_user_model

@@ -279,6 +279,311 @@ class SalesItemByPrintingOrderAPITestCase(TestCase):
        self.assertEqual(response.status_code, status.HTTP_401_UNAUTHORIZED)


+class ShipmentSalesItemCustomersAPITestCase(TestCase):
+    """测试由未出货销售品汇总客户 API"""
+
+    def setUp(self):
+        self.client = APIClient()
+
+        self.merchant = basic_models.Merchant.objects.create(
+            name="测试印花厂", type=basic_models.MerchantTypeEnum.FACTORY
+        )
+        self.other_merchant = basic_models.Merchant.objects.create(
+            name="其它商户", type=basic_models.MerchantTypeEnum.FACTORY
+        )
+
+        self.user = User.objects.create_user(
+            username="shipment_customer_user",
+            password="testpass123",
+            email="shipment_customer@example.com",
+        )
+        self.employee = basic_models.Employee.objects.create(
+            sys_user=self.user,
+            merchant=self.merchant,
+            name="测试员工",
+            mobile="13800138020",
+            status=basic_models.EmployeeStatusEnum.ACTIVE,
+        )
+
+        self.customer1 = basic_models.Customer.objects.create(
+            merchant=self.merchant,
+            name="客户A",
+            mobile="13900139020",
+            area="杭州",
+        )
+        self.customer2 = basic_models.Customer.objects.create(
+            merchant=self.merchant,
+            name="客户B",
+            mobile="13900139021",
+            area="绍兴",
+        )
+        self.other_customer = basic_models.Customer.objects.create(
+            merchant=self.other_merchant,
+            name="客户C",
+            mobile="13900139022",
+            area="苏州",
+        )
+
+        shipment_models.SalesItem.objects.create(
+            merchant=self.merchant,
+            name="客户A销售品1",
+            quantity=Decimal("10.00"),
+            unit=shipment_models.UnitChoices.METER,
+            customer_id=self.customer1.id,
+            created_by=self.user,
+        )
+        shipment_models.SalesItem.objects.create(
+            merchant=self.merchant,
+            name="客户A销售品2",
+            quantity=Decimal("20.00"),
+            unit=shipment_models.UnitChoices.METER,
+            customer_id=self.customer1.id,
+            created_by=self.user,
+        )
+        shipment_models.SalesItem.objects.create(
+            merchant=self.merchant,
+            name="客户B销售品1",
+            quantity=Decimal("30.00"),
+            unit=shipment_models.UnitChoices.METER,
+            customer_id=self.customer2.id,
+            created_by=self.user,
+        )
+
+        shipped = shipment_models.Shipment.objects.create(
+            merchant=self.merchant,
+            customer=self.customer1,
+            shipment_date="2026-01-14",
+            created_by=self.user,
+        )
+        shipment_models.SalesItem.objects.create(
+            merchant=self.merchant,
+            name="客户A已出货销售品",
+            quantity=Decimal("40.00"),
+            unit=shipment_models.UnitChoices.METER,
+            customer_id=self.customer1.id,
+            shipment=shipped,
+            created_by=self.user,
+        )
+        shipment_models.SalesItem.objects.create(
+            merchant=self.merchant,
+            name="无客户销售品",
+            quantity=Decimal("50.00"),
+            unit=shipment_models.UnitChoices.METER,
+            created_by=self.user,
+        )
+        shipment_models.SalesItem.objects.create(
+            merchant=self.other_merchant,
+            name="其它商户销售品",
+            quantity=Decimal("60.00"),
+            unit=shipment_models.UnitChoices.METER,
+            customer_id=self.other_customer.id,
+            created_by=self.user,
+        )
+
+        self.client.force_authenticate(user=self.user)
+
+    def test_list_customers_with_unshipped_sales_items(self):
+        resp = self.client.get("/api/v1/shipment/sales-items/customers/")
+
+        self.assertEqual(resp.status_code, status.HTTP_200_OK)
+        data = resp.json()
+
+        self.assertEqual(data["count"], 2)
+        self.assertIn("next", data)
+        self.assertIn("previous", data)
+
+        by_customer_id = {
+            item["customer_id"]: item
+            for item in data["results"]
+        }
+        self.assertEqual(set(by_customer_id.keys()), {self.customer1.id, self.customer2.id})
+        self.assertEqual(by_customer_id[self.customer1.id]["customer_name"], "客户A")
+        self.assertEqual(by_customer_id[self.customer1.id]["unshipped_sales_items_count"], 2)
+        self.assertEqual(by_customer_id[self.customer2.id]["unshipped_sales_items_count"], 1)
+
+    def test_list_customers_with_unshipped_sales_items_supports_limit_offset(self):
+        resp = self.client.get("/api/v1/shipment/sales-items/customers/?limit=1&offset=1")
+
+        self.assertEqual(resp.status_code, status.HTTP_200_OK)
+        data = resp.json()
+
+        self.assertEqual(data["count"], 2)
+        self.assertEqual(len(data["results"]), 1)
+        self.assertEqual(data["results"][0]["customer_id"], self.customer2.id)
+
+    def test_list_customers_with_unshipped_sales_items_unauthenticated(self):
+        self.client.logout()
+
+        resp = self.client.get("/api/v1/shipment/sales-items/customers/")
+
+        self.assertEqual(resp.status_code, status.HTTP_401_UNAUTHORIZED)
+
+
+class SalesItemByCustomerAPITestCase(TestCase):
+    """测试按客户查询销售品 API"""
+
+    def setUp(self):
+        self.client = APIClient()
+
+        self.merchant = basic_models.Merchant.objects.create(
+            name="测试印花厂", type=basic_models.MerchantTypeEnum.FACTORY
+        )
+        self.other_merchant = basic_models.Merchant.objects.create(
+            name="其它印花厂", type=basic_models.MerchantTypeEnum.FACTORY
+        )
+
+        self.user = User.objects.create_user(
+            username="sales_item_by_customer_user",
+            password="testpass123",
+            email="sales_item_by_customer@example.com",
+        )
+        self.employee = basic_models.Employee.objects.create(
+            sys_user=self.user,
+            merchant=self.merchant,
+            name="测试员工",
+            mobile="13800138030",
+            status=basic_models.EmployeeStatusEnum.ACTIVE,
+        )
+
+        self.customer = basic_models.Customer.objects.create(
+            merchant=self.merchant,
+            name="测试客户",
+            mobile="13900139030",
+            area="杭州",
+        )
+        self.other_customer = basic_models.Customer.objects.create(
+            merchant=self.merchant,
+            name="其它客户",
+            mobile="13900139031",
+            area="绍兴",
+        )
+        self.foreign_customer = basic_models.Customer.objects.create(
+            merchant=self.other_merchant,
+            name="外部客户",
+            mobile="13900139032",
+            area="苏州",
+        )
+
+        self.state = stateflow_models.State.objects.create(name="待生产")
+        self.process = stateflow_models.Process.objects.create(name="生产流程")
+        self.process.replace_nodes([self.state])
+        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="PROD001",
+        )
+        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_job = printing_models.PrintingJob.objects.create(
+            merchant=self.merchant,
+            printing_order=self.printing_order,
+            product=self.product,
+            quantity=100,
+            unit="米",
+            created_by=self.user,
+        )
+
+        self.sales_item1 = shipment_models.SalesItem.objects.create(
+            merchant=self.merchant,
+            name="销售品1",
+            quantity=Decimal("10.00"),
+            unit=shipment_models.UnitChoices.METER,
+            printing_job_id=self.printing_job.id,
+            customer_id=self.customer.id,
+            position="A1-01",
+            created_by=self.user,
+        )
+        self.sales_item2 = shipment_models.SalesItem.objects.create(
+            merchant=self.merchant,
+            name="销售品2",
+            quantity=Decimal("20.00"),
+            unit=shipment_models.UnitChoices.PIECE,
+            printing_job_id=self.printing_job.id,
+            customer_id=self.customer.id,
+            created_by=self.user,
+        )
+        self.shipment = shipment_models.Shipment.objects.create(
+            merchant=self.merchant,
+            customer=self.customer,
+            shipment_date="2026-01-14",
+            created_by=self.user,
+        )
+        self.sales_item_shipped = shipment_models.SalesItem.objects.create(
+            merchant=self.merchant,
+            name="已出货销售品",
+            quantity=Decimal("30.00"),
+            unit=shipment_models.UnitChoices.METER,
+            printing_job_id=self.printing_job.id,
+            customer_id=self.customer.id,
+            shipment=self.shipment,
+            created_by=self.user,
+        )
+        shipment_models.SalesItem.objects.create(
+            merchant=self.merchant,
+            name="其它客户销售品",
+            quantity=Decimal("40.00"),
+            unit=shipment_models.UnitChoices.METER,
+            customer_id=self.other_customer.id,
+            created_by=self.user,
+        )
+
+        self.client.force_authenticate(user=self.user)
+
+    def test_get_sales_items_by_customer_exclude_shipped(self):
+        resp = self.client.get(
+            f"/api/v1/shipment/sales-items/by-customer/{self.customer.id}/"
+        )
+
+        self.assertEqual(resp.status_code, status.HTTP_200_OK)
+        data = resp.json()
+
+        self.assertEqual(data["count"], 2)
+        self.assertEqual(len(data["results"]), 2)
+        item = next(it for it in data["results"] if it["id"] == self.sales_item1.id)
+        self.assertEqual(item["customer_id"], self.customer.id)
+        self.assertEqual(item["customer_name"], self.customer.name)
+        self.assertEqual(item["printing_job_id"], self.printing_job.id)
+        self.assertEqual(item["printing_order_id"], self.printing_order.id)
+        self.assertEqual(item["position"], "A1-01")
+
+    def test_get_sales_items_by_customer_include_shipped_and_paginate(self):
+        resp = self.client.get(
+            f"/api/v1/shipment/sales-items/by-customer/{self.customer.id}/"
+            "?include_already_has_shipment=true&limit=2&offset=1"
+        )
+
+        self.assertEqual(resp.status_code, status.HTTP_200_OK)
+        data = resp.json()
+
+        self.assertEqual(data["count"], 3)
+        self.assertEqual(len(data["results"]), 2)
+        result_ids = [item["id"] for item in data["results"]]
+        self.assertEqual(result_ids, [self.sales_item2.id, self.sales_item_shipped.id])
+
+    def test_get_sales_items_by_customer_not_found(self):
+        resp = self.client.get("/api/v1/shipment/sales-items/by-customer/99999/")
+
+        self.assertEqual(resp.status_code, status.HTTP_404_NOT_FOUND)
+
+    def test_get_sales_items_by_customer_other_merchant_404(self):
+        resp = self.client.get(
+            f"/api/v1/shipment/sales-items/by-customer/{self.foreign_customer.id}/"
+        )
+
+        self.assertEqual(resp.status_code, status.HTTP_404_NOT_FOUND)
+
+
 class ShipmentCreateAPITestCase(TestCase):
    """测试创建出货单 API"""