zuowei/erpnew
1
0
Fork 0
forked from erp-dev/erp
Code Pull Requests Activity

feat: added pre sales order and pre purchase order models and crud

Browse Source
This commit is contained in:
colaftc
2026-01-31 23:26:01 +08:00
parent a093174913
commit f643b643b7
12 changed files with 1008 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

120
business/tests/test_pre_purchase_order.py Normal file
View File

@@ -0,0 +1,120 @@
from django.test import TestCase
from django.contrib.auth import get_user_model
from basic_info import models as basic_models
from business import models as business_models
from business import pre_order_services
from .fixtures import create_basic_fixtures
class PrePurchaseOrderServiceTestCase(TestCase):
def setUp(self):
(
self.merchant,
self.supplier,
self.warehouse_strict,
self.warehouse_relaxed,
self.product,
self.operator,
) = create_basic_fixtures()
self.user = get_user_model().objects.create_user(username='purchase-user', password='pass')
self.operator.sys_user = self.user
self.operator.save(update_fields=['sys_user'])
def test_create_pre_purchase_order_success(self):
pre_purchase_order = pre_order_services.create_pre_purchase_order(
merchant=self.merchant,
supplier_id=self.supplier.id,
warehouse_id=self.warehouse_relaxed.id,
created_by=self.user,
operator=self.operator,
kind=business_models.PurchaseOrderKindEnum.WHOLESALE,
items=[
{
'product_id': self.product.id,
'quantity': '12.5',
'unit': '米',
'order_quantity': 8,
'remarks': 'item-remarks',
}
],
remarks='order-remarks',
)
self.assertEqual(pre_purchase_order.merchant_id, self.merchant.id)
self.assertEqual(pre_purchase_order.supplier_id, self.supplier.id)
self.assertEqual(pre_purchase_order.warehouse_id, self.warehouse_relaxed.id)
self.assertEqual(pre_purchase_order.created_by_id, self.user.id)
self.assertEqual(pre_purchase_order.operator_id, self.operator.id)
self.assertEqual(pre_purchase_order.items.count(), 1)
item = pre_purchase_order.items.first()
self.assertEqual(item.product_id, self.product.id)
self.assertEqual(item.unit, '米')
self.assertEqual(item.order_quantity, 8)
def test_create_pre_purchase_order_rejects_cross_merchant_supplier(self):
other_merchant = basic_models.Merchant.objects.create(
name='其他商户',
type=basic_models.MerchantTypeEnum.FACTORY,
)
other_supplier = basic_models.Supplier.objects.create(
merchant=other_merchant,
name='其他供应商',
)
with self.assertRaises(ValueError):
pre_order_services.create_pre_purchase_order(
merchant=self.merchant,
supplier_id=other_supplier.id,
warehouse_id=self.warehouse_relaxed.id,
created_by=self.user,
operator=self.operator,
items=[{'product_id': self.product.id, 'quantity': '1', 'unit': '米'}],
)
def test_update_pre_purchase_order_replaces_items(self):
pre_purchase_order = pre_order_services.create_pre_purchase_order(
merchant=self.merchant,
supplier_id=self.supplier.id,
warehouse_id=self.warehouse_relaxed.id,
created_by=self.user,
operator=self.operator,
items=[{'product_id': self.product.id, 'quantity': '10', 'unit': '米'}],
)
updated = pre_order_services.update_pre_purchase_order(
pre_purchase_order=pre_purchase_order,
warehouse_id=self.warehouse_strict.id,
items=[
{
'product_id': self.product.id,
'quantity': '20',
'unit': '米',
'order_quantity': 3,
}
],
remarks='updated-remarks',
)
self.assertEqual(updated.warehouse_id, self.warehouse_strict.id)
self.assertEqual(updated.remarks, 'updated-remarks')
self.assertEqual(updated.items.count(), 1)
self.assertEqual(str(updated.items.first().quantity), '20.00')
self.assertEqual(updated.items.first().order_quantity, 3)
def test_delete_pre_purchase_order(self):
pre_purchase_order = pre_order_services.create_pre_purchase_order(
merchant=self.merchant,
supplier_id=self.supplier.id,
warehouse_id=self.warehouse_relaxed.id,
created_by=self.user,
operator=self.operator,
items=[{'product_id': self.product.id, 'quantity': '10', 'unit': '米'}],
)
order_id = pre_purchase_order.id
pre_order_services.delete_pre_purchase_order(pre_purchase_order=pre_purchase_order)
self.assertFalse(business_models.PrePurchaseOrder.objects.filter(id=order_id).exists())

20
business/tests/test_pre_sales_order.py
View File

@@ -1,4 +1,5 @@
from django.test import TestCase
from django.contrib.auth import get_user_model
from basic_info import models as basic_models
from business import models as business_models
@@ -19,12 +20,17 @@ class PreSalesOrderServiceTestCase(TestCase):
self.operator,
) = create_sales_fixtures()
self.user = get_user_model().objects.create_user(username='sales-user', password='pass')
self.operator.sys_user = self.user
self.operator.save(update_fields=['sys_user'])
def test_create_pre_sales_order_success(self):
pre_sales_order = pre_order_services.create_pre_sales_order(
merchant=self.merchant,
customer_id=self.customer.id,
warehouse_id=self.warehouse_relaxed.id,
created_by=self.operator,
created_by=self.user,
operator=self.operator,
kind=business_models.SalesOrderKindEnum.WHOLESALE,
items=[
{
@@ -41,7 +47,8 @@ class PreSalesOrderServiceTestCase(TestCase):
self.assertEqual(pre_sales_order.merchant_id, self.merchant.id)
self.assertEqual(pre_sales_order.customer_id, self.customer.id)
self.assertEqual(pre_sales_order.warehouse_id, self.warehouse_relaxed.id)
self.assertEqual(pre_sales_order.created_by_id, self.operator.id)
self.assertEqual(pre_sales_order.created_by_id, self.user.id)
self.assertEqual(pre_sales_order.operator_id, self.operator.id)
self.assertEqual(pre_sales_order.items.count(), 1)
item = pre_sales_order.items.first()
@@ -65,7 +72,8 @@ class PreSalesOrderServiceTestCase(TestCase):
merchant=self.merchant,
customer_id=other_customer.id,
warehouse_id=self.warehouse_relaxed.id,
created_by=self.operator,
created_by=self.user,
operator=self.operator,
items=[{'product_id': self.product.id, 'quantity': '1', 'unit': '米'}],
)
@@ -74,7 +82,8 @@ class PreSalesOrderServiceTestCase(TestCase):
merchant=self.merchant,
customer_id=self.customer.id,
warehouse_id=self.warehouse_relaxed.id,
created_by=self.operator,
created_by=self.user,
operator=self.operator,
items=[{'product_id': self.product.id, 'quantity': '10', 'unit': '米'}],
)
@@ -103,7 +112,8 @@ class PreSalesOrderServiceTestCase(TestCase):
merchant=self.merchant,
customer_id=self.customer.id,
warehouse_id=self.warehouse_relaxed.id,
created_by=self.operator,
created_by=self.user,
operator=self.operator,
items=[{'product_id': self.product.id, 'quantity': '10', 'unit': '米'}],
)
order_id = pre_sales_order.id