fear: stock reverse placeholder

stock/tests.py

@@ -1,12 +1,10 @@
 from django.core.exceptions import ValidationError
 from django.test import TestCase
-from django.utils import timezone
 from decimal import Decimal
 from unittest.mock import patch, MagicMock
 import logging
 
-from basic_info.models import Product, WareHouse, Supplier, ProductUnitEnum, ProductCategory, Merchant, MerchantTypeEnum, WareHouseModeEnum
-from business.models import PurchaseOrder
+from basic_info.models import Product, WareHouse, ProductUnitEnum, ProductCategory, Merchant, MerchantTypeEnum, WareHouseModeEnum
 from . import models, services
 
 
@@ -49,14 +47,6 @@ class StockServicesTestCase(TestCase):
             location="备用仓库地址"
         )
         
-        # 创建供应商
-        cls.supplier = Supplier.objects.create(
-            merchant=cls.merchant,
-            name="测试供应商",
-            contact="张三",
-            mobile="13800138000"
-        )
-        
         # 创建产品
         cls.product_fabric_a = Product.objects.create(
             merchant=cls.merchant,
@@ -83,21 +73,14 @@ class StockServicesTestCase(TestCase):
 
     def setUp(self):
         """每个测试方法执行前的设置"""
-        # 创建采购单
-        self.purchase_order = PurchaseOrder.objects.create(
-            merchant=self.merchant,
-            supplier=self.supplier,
-            order_date=timezone.now().date(),
-            total_amount=Decimal('1000.00'),
-            remarks="测试采购单"
-        )
+        self.purchase_source_id = 1001
         
         # 创建库存变动记录（入库）
         self.stock_change_in = models.StockChangeRecord.objects.create(
             merchant=self.merchant,
             type=models.StockChangeTypeEnum.ADD,
             source_type=models.StockChangeSourceEnum.PURCHASE,
-            source_id=self.purchase_order.id,
+            source_id=self.purchase_source_id,
             warehouse=self.warehouse_main,
             is_finished=False
         )
@@ -392,7 +375,7 @@ class CreateStockChangeRecordWithDetailsTestCase(StockServicesTestCase):
             type=models.StockChangeTypeEnum.ADD,
             warehouse_id=self.warehouse_main.id,
             source_type=models.StockChangeSourceEnum.PURCHASE,
-            source_id=self.purchase_order.id,
+            source_id=self.purchase_source_id,
             products=self.products_payload,
         )
 
@@ -418,7 +401,7 @@ class CreateStockChangeRecordWithDetailsTestCase(StockServicesTestCase):
                 type=models.StockChangeTypeEnum.ADD,
                 warehouse_id=self.warehouse_main.id,
                 source_type=models.StockChangeSourceEnum.PURCHASE,
-                source_id=self.purchase_order.id,
+                source_id=self.purchase_source_id,
                 products=self.products_payload,
             )
 
@@ -435,7 +418,7 @@ class CreateStockChangeRecordWithDetailsTestCase(StockServicesTestCase):
                 type=models.StockChangeTypeEnum.ADD,
                 warehouse_id=self.warehouse_main.id,
                 source_type=models.StockChangeSourceEnum.PURCHASE,
-                source_id=self.purchase_order.id,
+                source_id=self.purchase_source_id,
                 products=products,
             )
 
@@ -451,7 +434,7 @@ class CreateStockChangeRecordWithDetailsTestCase(StockServicesTestCase):
                 type=models.StockChangeTypeEnum.ADD,
                 warehouse_id=self.warehouse_main.id,
                 source_type=models.StockChangeSourceEnum.PURCHASE,
-                source_id=self.purchase_order.id,
+                source_id=self.purchase_source_id,
                 products=[{'product': self.product_fabric_a.id, 'quantity': [Decimal('10.00')]}],
             )
 
@@ -468,7 +451,7 @@ class CreateStockChangeRecordWithDetailsTestCase(StockServicesTestCase):
             type=models.StockChangeTypeEnum.ADD,
             warehouse_id=self.warehouse_main.id,
             source_type=models.StockChangeSourceEnum.PURCHASE,
-            source_id=self.purchase_order.id,
+            source_id=self.purchase_source_id,
             products=[{'product': self.product_fabric_a.id, 'quantity': [Decimal('5.00'), Decimal('6.00')]}],
         )
 
@@ -604,7 +587,7 @@ class CreateStockChangeRecordRelaxedTestCase(StockServicesTestCase):
             type=models.StockChangeTypeEnum.ADD,
             warehouse_id=self.warehouse_main.id,
             source_type=models.StockChangeSourceEnum.PURCHASE,
-            source_id=self.purchase_order.id,
+            source_id=self.purchase_source_id,
             products=products,
         )
 
@@ -629,7 +612,7 @@ class CreateStockChangeRecordRelaxedTestCase(StockServicesTestCase):
             type=models.StockChangeTypeEnum.ADD,
             warehouse_id=self.warehouse_main.id,
             source_type=models.StockChangeSourceEnum.PURCHASE,
-            source_id=self.purchase_order.id,
+            source_id=self.purchase_source_id,
             products=products,
         )
 
@@ -654,7 +637,7 @@ class CreateStockChangeRecordRelaxedTestCase(StockServicesTestCase):
                 type=models.StockChangeTypeEnum.ADD,
                 warehouse_id=self.warehouse_main.id,
                 source_type=models.StockChangeSourceEnum.PURCHASE,
-                source_id=self.purchase_order.id,
+                source_id=self.purchase_source_id,
                 products=products,
             )
 
@@ -676,7 +659,7 @@ class CreateStockChangeRecordRelaxedTestCase(StockServicesTestCase):
                 type=models.StockChangeTypeEnum.ADD,
                 warehouse_id=self.warehouse_main.id,
                 source_type=models.StockChangeSourceEnum.PURCHASE,
-                source_id=self.purchase_order.id,
+                source_id=self.purchase_source_id,
                 products=products,
             )
 
@@ -700,7 +683,7 @@ class CreateStockChangeRecordRelaxedTestCase(StockServicesTestCase):
                 type=models.StockChangeTypeEnum.ADD,
                 warehouse_id=self.warehouse_main.id,
                 source_type=models.StockChangeSourceEnum.PURCHASE,
-                source_id=self.purchase_order.id,
+                source_id=self.purchase_source_id,
                 products=products,
             )
 
@@ -724,7 +707,7 @@ class CreateStockChangeRecordRelaxedTestCase(StockServicesTestCase):
                 type=models.StockChangeTypeEnum.ADD,
                 warehouse_id=self.warehouse_main.id,
                 source_type=models.StockChangeSourceEnum.PURCHASE,
-                source_id=self.purchase_order.id,
+                source_id=self.purchase_source_id,
                 products=products,
             )