feat: purchase_order approve and calcel

api_v1/tests.py

@@ -1,3 +1,4 @@
+import copy
 import shutil
 import tempfile
 from pathlib import Path
@@ -19,6 +20,8 @@ from basic_info.models import (
     WareHouse,
     WareHouseModeEnum,
 )
+from business import models as business_models
+from stock import models as stock_models
 from api_v1 import tasks
 
 
@@ -197,13 +200,22 @@ class PurchaseOrderAPITestCase(TestCase):
             ],
         }
 
+    def _create_purchase_order(self, payload):
+        body = copy.deepcopy(payload)
+        with patch('business.services.create_purchase_order_stock_entries.delay'):
+            response = self.client.post('/api/v1/purchase-orders/', body, format='json')
+        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
+        return response.data['id']
+
     def test_create_purchase_order_success_strict(self):
         with patch('business.services.create_purchase_order_stock_entries.delay') as mock_delay:
             response = self.client.post('/api/v1/purchase-orders/', self.strict_payload, format='json')
 
         self.assertEqual(response.status_code, status.HTTP_201_CREATED)
         self.assertIn('id', response.data)
-        mock_delay.assert_called_once()
+        self.assertEqual(response.data['status'], 1)
+        self.assertIn('等待审批', response.data['message'])
+        mock_delay.assert_not_called()
 
     def test_create_purchase_order_invalid_supplier(self):
         payload = {**self.strict_payload, 'supplier': 999}
@@ -220,7 +232,8 @@ class PurchaseOrderAPITestCase(TestCase):
         with patch('business.services.create_purchase_order_stock_entries.delay') as mock_delay:
             response = self.client.post('/api/v1/purchase-orders/', self.relaxed_payload, format='json')
         self.assertEqual(response.status_code, status.HTTP_201_CREATED)
-        mock_delay.assert_called_once()
+        self.assertEqual(response.data['status'], 1)
+        mock_delay.assert_not_called()
 
     def test_mode_mismatch_raises(self):
         payload = {**self.relaxed_payload}
@@ -229,6 +242,42 @@ class PurchaseOrderAPITestCase(TestCase):
         self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
         self.assertIn('严进模式', response.data['error'])
 
+    def test_review_purchase_order_requires_action(self):
+        order_id = self._create_purchase_order(self.strict_payload)
+        response = self.client.post(f'/api/v1/purchase-orders/{order_id}/review/', {}, format='json')
+        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
+        self.assertIn('action', response.data)
+
+    def test_review_purchase_order_approve_success(self):
+        order_id = self._create_purchase_order(self.strict_payload)
+        response = self.client.post(
+            f'/api/v1/purchase-orders/{order_id}/review/',
+            {'action': 'approve'},
+            format='json',
+        )
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertEqual(response.data['status'], business_models.PurchaseOrderStatusEnum.APPROVED)
+        order = business_models.PurchaseOrder.objects.get(id=order_id)
+        self.assertEqual(order.status, business_models.PurchaseOrderStatusEnum.APPROVED)
+
+    def test_review_purchase_order_cancel_blocked_after_stock_exists(self):
+        order_id = self._create_purchase_order(self.relaxed_payload)
+        stock_models.StockChangeRecord.objects.create(
+            merchant=self.merchant,
+            type=stock_models.StockChangeTypeEnum.ADD,
+            warehouse=self.warehouse_relaxed,
+            source_type=stock_models.StockChangeSourceEnum.PURCHASE,
+            source_id=order_id,
+        )
+        response = self.client.post(
+            f'/api/v1/purchase-orders/{order_id}/review/',
+            {'action': 'cancel'},
+            format='json',
+        )
+        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
+        order = business_models.PurchaseOrder.objects.get(id=order_id)
+        self.assertEqual(order.status, business_models.PurchaseOrderStatusEnum.PENDING)
+
 
 @override_settings(
     CELERY_TASK_ALWAYS_EAGER=True,