forked from erp-dev/erp
1061 lines
42 KiB
Python
1061 lines
42 KiB
Python
from django.test import TestCase
|
||
from django.contrib.auth.models import User
|
||
from rest_framework.test import APIClient
|
||
from rest_framework import status
|
||
from basic_info.models import (
|
||
Merchant, MerchantTypeEnum, WareHouse, WarehouseTypeEnum,
|
||
Employee, EmployeeType, EmployeeStatusEnum,
|
||
QuickInput, Product, ProductCategory, Supplier, Customer,
|
||
VehicleType, BankAccount, DeviceInfo, VehicleTransportRecord
|
||
)
|
||
|
||
|
||
class WarehouseAPITestCase(TestCase):
|
||
"""测试仓库 API"""
|
||
|
||
def setUp(self):
|
||
"""设置测试数据"""
|
||
# 创建商户
|
||
self.merchant = Merchant.objects.create(
|
||
name='测试商户',
|
||
type=MerchantTypeEnum.STORE
|
||
)
|
||
|
||
# 创建用户和员工
|
||
self.user = User.objects.create_user(
|
||
username='testuser',
|
||
password='testpass123'
|
||
)
|
||
self.employee = Employee.objects.create(
|
||
merchant=self.merchant,
|
||
sys_user=self.user,
|
||
name='测试员工'
|
||
)
|
||
|
||
# 设置 API 客户端
|
||
self.client = APIClient()
|
||
self.client.force_authenticate(user=self.user)
|
||
|
||
def test_create_warehouse_with_default_type(self):
|
||
"""测试创建仓库时使用默认类型"""
|
||
data = {
|
||
'name': '测试仓库1',
|
||
'location': '测试地址',
|
||
}
|
||
response = self.client.post('/api/backend/warehouses/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
||
self.assertEqual(response.data['type'], WarehouseTypeEnum.WHOLE.value)
|
||
|
||
# 验证数据库
|
||
warehouse = WareHouse.objects.get(id=response.data['id'])
|
||
self.assertEqual(warehouse.type, WarehouseTypeEnum.WHOLE)
|
||
|
||
def test_create_warehouse_with_whole_type(self):
|
||
"""测试创建整仓"""
|
||
data = {
|
||
'name': '整仓测试',
|
||
'type': WarehouseTypeEnum.WHOLE.value,
|
||
}
|
||
response = self.client.post('/api/backend/warehouses/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
||
self.assertEqual(response.data['type'], WarehouseTypeEnum.WHOLE.value)
|
||
|
||
def test_create_warehouse_with_scattered_type(self):
|
||
"""测试创建散仓"""
|
||
data = {
|
||
'name': '散仓测试',
|
||
'type': WarehouseTypeEnum.SCATTERED.value,
|
||
}
|
||
response = self.client.post('/api/backend/warehouses/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
||
self.assertEqual(response.data['type'], WarehouseTypeEnum.SCATTERED.value)
|
||
|
||
def test_list_warehouses_includes_type(self):
|
||
"""测试列表接口返回类型字段"""
|
||
WareHouse.objects.create(
|
||
merchant=self.merchant,
|
||
name='仓库1',
|
||
type=WarehouseTypeEnum.WHOLE
|
||
)
|
||
WareHouse.objects.create(
|
||
merchant=self.merchant,
|
||
name='仓库2',
|
||
type=WarehouseTypeEnum.SCATTERED
|
||
)
|
||
|
||
response = self.client.get('/api/backend/warehouses/')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 2)
|
||
|
||
# 验证返回的数据包含 type 字段
|
||
for warehouse in results:
|
||
self.assertIn('type', warehouse)
|
||
self.assertIn(warehouse['type'], [WarehouseTypeEnum.WHOLE.value, WarehouseTypeEnum.SCATTERED.value])
|
||
|
||
def test_update_warehouse_type(self):
|
||
"""测试更新仓库类型"""
|
||
warehouse = WareHouse.objects.create(
|
||
merchant=self.merchant,
|
||
name='待更新仓库',
|
||
type=WarehouseTypeEnum.WHOLE
|
||
)
|
||
|
||
data = {'type': WarehouseTypeEnum.SCATTERED.value}
|
||
response = self.client.patch(f'/api/backend/warehouses/{warehouse.id}/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
self.assertEqual(response.data['type'], WarehouseTypeEnum.SCATTERED.value)
|
||
|
||
# 验证数据库
|
||
warehouse.refresh_from_db()
|
||
self.assertEqual(warehouse.type, WarehouseTypeEnum.SCATTERED)
|
||
|
||
def test_retrieve_warehouse_includes_type(self):
|
||
"""测试检索单个仓库时包含类型字段"""
|
||
warehouse = WareHouse.objects.create(
|
||
merchant=self.merchant,
|
||
name='单个仓库',
|
||
type=WarehouseTypeEnum.SCATTERED
|
||
)
|
||
|
||
response = self.client.get(f'/api/backend/warehouses/{warehouse.id}/')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
self.assertEqual(response.data['type'], WarehouseTypeEnum.SCATTERED.value)
|
||
self.assertEqual(response.data['name'], '单个仓库')
|
||
|
||
def test_filter_warehouses_by_name(self):
|
||
"""测试通过 name 查询参数过滤仓库（icontains 模糊匹配）"""
|
||
WareHouse.objects.create(merchant=self.merchant, name='主仓库')
|
||
WareHouse.objects.create(merchant=self.merchant, name='备用仓库')
|
||
WareHouse.objects.create(merchant=self.merchant, name='临时仓库')
|
||
|
||
# 测试部分匹配
|
||
response = self.client.get('/api/backend/warehouses/?name=主')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 1)
|
||
self.assertEqual(results[0]['name'], '主仓库')
|
||
|
||
# 测试包含"仓库"的所有记录
|
||
response = self.client.get('/api/backend/warehouses/?name=仓库')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 3)
|
||
|
||
# 测试不匹配
|
||
response = self.client.get('/api/backend/warehouses/?name=不存在')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 0)
|
||
|
||
|
||
class EmployeeTypeAPITestCase(TestCase):
|
||
"""测试员工职位类型 API"""
|
||
|
||
def setUp(self):
|
||
"""设置测试数据"""
|
||
# 创建商户
|
||
self.merchant = Merchant.objects.create(
|
||
name='测试商户',
|
||
type=MerchantTypeEnum.STORE
|
||
)
|
||
|
||
# 创建用户和员工
|
||
self.user = User.objects.create_user(
|
||
username='testuser',
|
||
password='testpass123'
|
||
)
|
||
self.employee = Employee.objects.create(
|
||
merchant=self.merchant,
|
||
sys_user=self.user,
|
||
name='测试员工'
|
||
)
|
||
|
||
# 设置 API 客户端
|
||
self.client = APIClient()
|
||
self.client.force_authenticate(user=self.user)
|
||
|
||
def test_create_employee_type(self):
|
||
"""测试创建职位类型"""
|
||
data = {
|
||
'title': '打纸工',
|
||
'description': '负责打纸工作'
|
||
}
|
||
response = self.client.post('/api/backend/employee-types/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
||
self.assertEqual(response.data['title'], '打纸工')
|
||
self.assertEqual(response.data['description'], '负责打纸工作')
|
||
|
||
def test_list_employee_types(self):
|
||
"""测试列出所有职位类型"""
|
||
EmployeeType.objects.create(merchant=self.merchant, title='打纸工')
|
||
EmployeeType.objects.create(merchant=self.merchant, title='滚筒工')
|
||
|
||
response = self.client.get('/api/backend/employee-types/')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 2)
|
||
|
||
def test_update_employee_type(self):
|
||
"""测试更新职位类型"""
|
||
emp_type = EmployeeType.objects.create(
|
||
merchant=self.merchant,
|
||
title='仓库管理员'
|
||
)
|
||
|
||
data = {'description': '新的描述'}
|
||
response = self.client.patch(f'/api/backend/employee-types/{emp_type.id}/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
self.assertEqual(response.data['description'], '新的描述')
|
||
|
||
def test_delete_employee_type(self):
|
||
"""测试删除职位类型"""
|
||
emp_type = EmployeeType.objects.create(
|
||
merchant=self.merchant,
|
||
title='临时工'
|
||
)
|
||
|
||
response = self.client.delete(f'/api/backend/employee-types/{emp_type.id}/')
|
||
self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
|
||
self.assertFalse(EmployeeType.objects.filter(id=emp_type.id).exists())
|
||
|
||
def test_filter_employee_types_by_title(self):
|
||
"""测试通过 title 查询参数过滤职位类型（icontains 模糊匹配）"""
|
||
EmployeeType.objects.create(merchant=self.merchant, title='打纸工')
|
||
EmployeeType.objects.create(merchant=self.merchant, title='滚筒工')
|
||
EmployeeType.objects.create(merchant=self.merchant, title='仓库管理员')
|
||
|
||
# 测试部分匹配
|
||
response = self.client.get('/api/backend/employee-types/?title=打纸')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 1)
|
||
self.assertEqual(results[0]['title'], '打纸工')
|
||
|
||
# 测试包含"工"的所有记录
|
||
response = self.client.get('/api/backend/employee-types/?title=工')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 2) # 打纸工、滚筒工
|
||
|
||
# 测试不匹配
|
||
response = self.client.get('/api/backend/employee-types/?title=不存在')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 0)
|
||
|
||
|
||
class EmployeeAPITestCase(TestCase):
|
||
"""测试员工 API（包含 job_type 兼容性）"""
|
||
|
||
def setUp(self):
|
||
"""设置测试数据"""
|
||
# 创建商户
|
||
self.merchant = Merchant.objects.create(
|
||
name='测试商户',
|
||
type=MerchantTypeEnum.STORE
|
||
)
|
||
|
||
# 创建用户和员工
|
||
self.user = User.objects.create_user(
|
||
username='testuser',
|
||
password='testpass123'
|
||
)
|
||
self.employee = Employee.objects.create(
|
||
merchant=self.merchant,
|
||
sys_user=self.user,
|
||
name='测试员工'
|
||
)
|
||
|
||
# 创建职位类型
|
||
self.emp_type = EmployeeType.objects.create(
|
||
merchant=self.merchant,
|
||
title='打纸工'
|
||
)
|
||
|
||
# 设置 API 客户端
|
||
self.client = APIClient()
|
||
self.client.force_authenticate(user=self.user)
|
||
|
||
def test_create_employee_with_position(self):
|
||
"""测试创建带职位的员工"""
|
||
data = {
|
||
'name': '张三',
|
||
'position': self.emp_type.id,
|
||
'mobile': '13800138000'
|
||
}
|
||
response = self.client.post('/api/backend/employees/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
||
self.assertEqual(response.data['position'], self.emp_type.id)
|
||
self.assertEqual(response.data['job_type'], '打纸工') # 验证兼容性字段
|
||
|
||
def test_create_employee_without_position(self):
|
||
"""测试创建不带职位的员工"""
|
||
data = {
|
||
'name': '李四',
|
||
'mobile': '13900139000'
|
||
}
|
||
response = self.client.post('/api/backend/employees/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
||
self.assertIsNone(response.data['position'])
|
||
self.assertEqual(response.data['job_type'], '') # 验证返回空字符串
|
||
|
||
def test_list_employees_includes_job_type(self):
|
||
"""测试列表接口包含 job_type 字段"""
|
||
Employee.objects.create(
|
||
merchant=self.merchant,
|
||
name='员工1',
|
||
position=self.emp_type
|
||
)
|
||
Employee.objects.create(
|
||
merchant=self.merchant,
|
||
name='员工2'
|
||
)
|
||
|
||
response = self.client.get('/api/backend/employees/')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
|
||
# 验证 job_type 字段存在
|
||
for emp in results:
|
||
self.assertIn('job_type', emp)
|
||
|
||
def test_update_employee_position(self):
|
||
"""测试更新员工职位"""
|
||
new_type = EmployeeType.objects.create(
|
||
merchant=self.merchant,
|
||
title='滚筒工'
|
||
)
|
||
|
||
employee = Employee.objects.create(
|
||
merchant=self.merchant,
|
||
name='王五',
|
||
position=self.emp_type
|
||
)
|
||
|
||
data = {'position': new_type.id}
|
||
response = self.client.patch(f'/api/backend/employees/{employee.id}/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
self.assertEqual(response.data['position'], new_type.id)
|
||
self.assertEqual(response.data['job_type'], '滚筒工')
|
||
|
||
def test_job_type_is_read_only(self):
|
||
"""测试 job_type 是只读字段"""
|
||
employee = Employee.objects.create(
|
||
merchant=self.merchant,
|
||
name='赵六',
|
||
position=self.emp_type
|
||
)
|
||
|
||
# 尝试直接更新 job_type（应该被忽略）
|
||
data = {'job_type': '其他职位'}
|
||
response = self.client.patch(f'/api/backend/employees/{employee.id}/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
self.assertEqual(response.data['job_type'], '打纸工') # 应该保持不变
|
||
|
||
def test_filter_employees_by_name(self):
|
||
"""测试通过 name 查询参数过滤员工（icontains 模糊匹配）"""
|
||
Employee.objects.create(merchant=self.merchant, name='张三')
|
||
Employee.objects.create(merchant=self.merchant, name='李四')
|
||
Employee.objects.create(merchant=self.merchant, name='王五')
|
||
|
||
# 测试部分匹配
|
||
response = self.client.get('/api/backend/employees/?name=三')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 1)
|
||
self.assertEqual(results[0]['name'], '张三')
|
||
|
||
# 测试不匹配
|
||
response = self.client.get('/api/backend/employees/?name=不存在')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 0)
|
||
|
||
|
||
class UserProfileAPITestCase(TestCase):
|
||
"""测试用户资料 API"""
|
||
|
||
def setUp(self):
|
||
"""设置测试数据"""
|
||
# 创建商户
|
||
self.merchant = Merchant.objects.create(
|
||
name='测试商户',
|
||
type=MerchantTypeEnum.STORE
|
||
)
|
||
|
||
# 创建用户和员工
|
||
self.user = User.objects.create_user(
|
||
username='testuser',
|
||
password='testpass123',
|
||
email='test@example.com'
|
||
)
|
||
self.employee = Employee.objects.create(
|
||
merchant=self.merchant,
|
||
sys_user=self.user,
|
||
name='测试员工'
|
||
)
|
||
|
||
# 创建另一个用户用于测试
|
||
self.user2 = User.objects.create_user(
|
||
username='testuser2',
|
||
password='testpass123',
|
||
email='test2@example.com'
|
||
)
|
||
|
||
# 设置 API 客户端
|
||
self.client = APIClient()
|
||
self.client.force_authenticate(user=self.user)
|
||
|
||
def test_create_user_profile(self):
|
||
"""测试创建用户资料"""
|
||
data = {
|
||
'user': self.user2.id,
|
||
'description': '测试用户资料描述'
|
||
}
|
||
response = self.client.post('/api/backend/user-profiles/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
||
self.assertEqual(response.data['user'], self.user2.id)
|
||
self.assertEqual(response.data['merchant'], self.merchant.id) # 应该由BaseViewSet自动设置
|
||
self.assertEqual(response.data['description'], '测试用户资料描述')
|
||
|
||
def test_list_user_profiles(self):
|
||
"""测试列出所有用户资料"""
|
||
from basic_info.models import UserProfile
|
||
UserProfile.objects.create(
|
||
user=self.user,
|
||
merchant=self.merchant,
|
||
description='用户1资料'
|
||
)
|
||
UserProfile.objects.create(
|
||
user=self.user2,
|
||
merchant=self.merchant,
|
||
description='用户2资料'
|
||
)
|
||
|
||
response = self.client.get('/api/backend/user-profiles/')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 2) # 应该只返回当前商户的用户资料
|
||
|
||
def test_retrieve_user_profile(self):
|
||
"""测试检索单个用户资料"""
|
||
from basic_info.models import UserProfile
|
||
profile = UserProfile.objects.create(
|
||
user=self.user,
|
||
merchant=self.merchant,
|
||
description='测试资料'
|
||
)
|
||
|
||
response = self.client.get(f'/api/backend/user-profiles/{profile.id}/')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
self.assertEqual(response.data['id'], profile.id)
|
||
self.assertEqual(response.data['description'], '测试资料')
|
||
self.assertEqual(response.data['user_detail']['username'], 'testuser') # 检查嵌套用户对象
|
||
|
||
def test_update_user_profile(self):
|
||
"""测试更新用户资料"""
|
||
from basic_info.models import UserProfile
|
||
profile = UserProfile.objects.create(
|
||
user=self.user,
|
||
merchant=self.merchant,
|
||
description='原始描述'
|
||
)
|
||
|
||
data = {'description': '更新后的描述'}
|
||
response = self.client.patch(f'/api/backend/user-profiles/{profile.id}/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
self.assertEqual(response.data['description'], '更新后的描述')
|
||
|
||
# 验证数据库
|
||
profile.refresh_from_db()
|
||
self.assertEqual(profile.description, '更新后的描述')
|
||
|
||
def test_delete_user_profile(self):
|
||
"""测试删除用户资料"""
|
||
from basic_info.models import UserProfile
|
||
profile = UserProfile.objects.create(
|
||
user=self.user,
|
||
merchant=self.merchant,
|
||
description='待删除资料'
|
||
)
|
||
|
||
response = self.client.delete(f'/api/backend/user-profiles/{profile.id}/')
|
||
self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
|
||
self.assertFalse(UserProfile.objects.filter(id=profile.id).exists())
|
||
|
||
def test_user_profile_filter_by_merchant(self):
|
||
"""测试用户资料按商户过滤"""
|
||
# 创建另一个商户
|
||
other_merchant = Merchant.objects.create(
|
||
name='其他商户',
|
||
type=MerchantTypeEnum.FACTORY
|
||
)
|
||
|
||
# 创建另一个商户的员工
|
||
other_user = User.objects.create_user(
|
||
username='otheruser',
|
||
password='testpass123'
|
||
)
|
||
other_employee = Employee.objects.create(
|
||
merchant=other_merchant,
|
||
sys_user=other_user,
|
||
name='其他员工'
|
||
)
|
||
|
||
# 为两个商户分别创建用户资料
|
||
from basic_info.models import UserProfile
|
||
UserProfile.objects.create(
|
||
user=self.user,
|
||
merchant=self.merchant,
|
||
description='本商户用户资料'
|
||
)
|
||
UserProfile.objects.create(
|
||
user=other_user,
|
||
merchant=other_merchant,
|
||
description='其他商户用户资料'
|
||
)
|
||
|
||
# 使用当前用户（属于self.merchant）访问API
|
||
response = self.client.get('/api/backend/user-profiles/')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 1) # 只应该返回当前商户的用户资料
|
||
self.assertEqual(results[0]['description'], '本商户用户资料')
|
||
|
||
|
||
# ==================== 补充缺失的 ViewSet 测试 ====================
|
||
|
||
class QuickInputAPITestCase(TestCase):
|
||
"""测试快捷输入 API（支持 name 和 group 查询参数）"""
|
||
|
||
def setUp(self):
|
||
self.merchant = Merchant.objects.create(
|
||
name='测试商户',
|
||
type=MerchantTypeEnum.STORE
|
||
)
|
||
self.user = User.objects.create_user(
|
||
username='testuser',
|
||
password='testpass123'
|
||
)
|
||
self.employee = Employee.objects.create(
|
||
merchant=self.merchant,
|
||
sys_user=self.user,
|
||
name='测试员工'
|
||
)
|
||
self.client = APIClient()
|
||
self.client.force_authenticate(user=self.user)
|
||
|
||
def test_create_quick_input(self):
|
||
"""测试创建快捷输入"""
|
||
data = {
|
||
'name': '测试输入',
|
||
'value': '测试值',
|
||
'group': '测试组'
|
||
}
|
||
response = self.client.post('/api/backend/quick-inputs/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
||
self.assertEqual(response.data['name'], '测试输入')
|
||
self.assertEqual(response.data['group'], '测试组')
|
||
|
||
def test_filter_quick_inputs_by_name(self):
|
||
"""测试通过 name 查询参数过滤快捷输入"""
|
||
QuickInput.objects.create(name='输入1', value='值1', group='组1')
|
||
QuickInput.objects.create(name='输入2', value='值2', group='组1')
|
||
QuickInput.objects.create(name='输入3', value='值3', group='组2')
|
||
|
||
response = self.client.get('/api/backend/quick-inputs/?name=输入1')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 1)
|
||
self.assertEqual(results[0]['name'], '输入1')
|
||
|
||
def test_filter_quick_inputs_by_group(self):
|
||
"""测试通过 group 查询参数过滤快捷输入"""
|
||
QuickInput.objects.create(name='输入1', value='值1', group='组1')
|
||
QuickInput.objects.create(name='输入2', value='值2', group='组1')
|
||
QuickInput.objects.create(name='输入3', value='值3', group='组2')
|
||
|
||
response = self.client.get('/api/backend/quick-inputs/?group=组1')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 2)
|
||
for item in results:
|
||
self.assertEqual(item['group'], '组1')
|
||
|
||
def test_quick_input_groups_action(self):
|
||
"""测试获取所有分组"""
|
||
QuickInput.objects.create(name='输入1', value='值1', group='组1')
|
||
QuickInput.objects.create(name='输入2', value='值2', group='组1')
|
||
QuickInput.objects.create(name='输入3', value='值3', group='组2')
|
||
|
||
response = self.client.get('/api/backend/quick-inputs/groups/')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
groups = list(response.data)
|
||
self.assertEqual(len(groups), 2)
|
||
self.assertIn('组1', groups)
|
||
self.assertIn('组2', groups)
|
||
|
||
|
||
class ProductAPITestCase(TestCase):
|
||
"""测试产品 API（支持 name 查询参数）"""
|
||
|
||
def setUp(self):
|
||
self.merchant = Merchant.objects.create(
|
||
name='测试商户',
|
||
type=MerchantTypeEnum.STORE
|
||
)
|
||
self.user = User.objects.create_user(
|
||
username='testuser',
|
||
password='testpass123'
|
||
)
|
||
self.employee = Employee.objects.create(
|
||
merchant=self.merchant,
|
||
sys_user=self.user,
|
||
name='测试员工'
|
||
)
|
||
self.category = ProductCategory.objects.create(
|
||
merchant=self.merchant,
|
||
name='测试类别',
|
||
product_prefix='TEST'
|
||
)
|
||
self.client = APIClient()
|
||
self.client.force_authenticate(user=self.user)
|
||
|
||
def test_create_product(self):
|
||
"""测试创建产品"""
|
||
data = {
|
||
'name': '测试产品',
|
||
'category': self.category.id
|
||
}
|
||
response = self.client.post('/api/backend/products/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
||
self.assertEqual(response.data['name'], '测试产品')
|
||
|
||
def test_filter_products_by_name(self):
|
||
"""测试通过 name 查询参数过滤产品"""
|
||
Product.objects.create(merchant=self.merchant, category=self.category, name='产品A')
|
||
Product.objects.create(merchant=self.merchant, category=self.category, name='产品B')
|
||
Product.objects.create(merchant=self.merchant, category=self.category, name='产品C')
|
||
|
||
response = self.client.get('/api/backend/products/?name=产品A')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 1)
|
||
self.assertEqual(results[0]['name'], '产品A')
|
||
|
||
|
||
class ProductCategoryAPITestCase(TestCase):
|
||
"""测试产品类别 API（支持 name 查询参数）"""
|
||
|
||
def setUp(self):
|
||
self.merchant = Merchant.objects.create(
|
||
name='测试商户',
|
||
type=MerchantTypeEnum.STORE
|
||
)
|
||
self.user = User.objects.create_user(
|
||
username='testuser',
|
||
password='testpass123'
|
||
)
|
||
self.employee = Employee.objects.create(
|
||
merchant=self.merchant,
|
||
sys_user=self.user,
|
||
name='测试员工'
|
||
)
|
||
self.client = APIClient()
|
||
self.client.force_authenticate(user=self.user)
|
||
|
||
def test_create_product_category(self):
|
||
"""测试创建产品类别"""
|
||
data = {
|
||
'name': '测试类别',
|
||
'product_prefix': 'TEST'
|
||
}
|
||
response = self.client.post('/api/backend/product-categories/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
||
self.assertEqual(response.data['name'], '测试类别')
|
||
|
||
def test_filter_product_categories_by_name(self):
|
||
"""测试通过 name 查询参数过滤产品类别"""
|
||
ProductCategory.objects.create(merchant=self.merchant, name='类别A', product_prefix='A')
|
||
ProductCategory.objects.create(merchant=self.merchant, name='类别B', product_prefix='B')
|
||
|
||
response = self.client.get('/api/backend/product-categories/?name=类别A')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 1)
|
||
self.assertEqual(results[0]['name'], '类别A')
|
||
|
||
|
||
class SupplierAPITestCase(TestCase):
|
||
"""测试供应商 API（支持 name 查询参数）"""
|
||
|
||
def setUp(self):
|
||
self.merchant = Merchant.objects.create(
|
||
name='测试商户',
|
||
type=MerchantTypeEnum.STORE
|
||
)
|
||
self.user = User.objects.create_user(
|
||
username='testuser',
|
||
password='testpass123'
|
||
)
|
||
self.employee = Employee.objects.create(
|
||
merchant=self.merchant,
|
||
sys_user=self.user,
|
||
name='测试员工'
|
||
)
|
||
self.client = APIClient()
|
||
self.client.force_authenticate(user=self.user)
|
||
|
||
def test_create_supplier(self):
|
||
"""测试创建供应商"""
|
||
data = {
|
||
'name': '测试供应商'
|
||
}
|
||
response = self.client.post('/api/backend/suppliers/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
||
self.assertEqual(response.data['name'], '测试供应商')
|
||
|
||
def test_filter_suppliers_by_name(self):
|
||
"""测试通过 name 查询参数过滤供应商"""
|
||
Supplier.objects.create(merchant=self.merchant, name='供应商A')
|
||
Supplier.objects.create(merchant=self.merchant, name='供应商B')
|
||
|
||
response = self.client.get('/api/backend/suppliers/?name=供应商A')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 1)
|
||
self.assertEqual(results[0]['name'], '供应商A')
|
||
|
||
|
||
class CustomerAPITestCase(TestCase):
|
||
"""测试客户 API（支持 name 和 visible_employee_name 查询参数）"""
|
||
|
||
def setUp(self):
|
||
self.merchant = Merchant.objects.create(
|
||
name='测试商户',
|
||
type=MerchantTypeEnum.STORE
|
||
)
|
||
self.user = User.objects.create_user(
|
||
username='testuser',
|
||
password='testpass123'
|
||
)
|
||
self.employee = Employee.objects.create(
|
||
merchant=self.merchant,
|
||
sys_user=self.user,
|
||
name='测试员工'
|
||
)
|
||
# CustomerViewSet 使用 DjangoModelPermissions，需要添加权限
|
||
from django.contrib.auth.models import Permission
|
||
from django.contrib.contenttypes.models import ContentType
|
||
content_type = ContentType.objects.get_for_model(Customer)
|
||
permissions = Permission.objects.filter(content_type=content_type)
|
||
self.user.user_permissions.set(permissions)
|
||
self.client = APIClient()
|
||
self.client.force_authenticate(user=self.user)
|
||
|
||
def test_create_customer(self):
|
||
"""测试创建客户"""
|
||
data = {
|
||
'name': '测试客户'
|
||
}
|
||
response = self.client.post('/api/backend/customers/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
||
self.assertEqual(response.data['name'], '测试客户')
|
||
customer = Customer.objects.get(id=response.data['id'])
|
||
self.assertEqual(customer.merchant, self.merchant)
|
||
self.assertEqual(customer.created_by, self.employee)
|
||
|
||
def test_create_customer_ignores_explicit_merchant_param(self):
|
||
"""测试创建客户时即使前端显式传 merchant，也以当前用户商户为准"""
|
||
other_merchant = Merchant.objects.create(
|
||
name='其他商户',
|
||
type=MerchantTypeEnum.STORE,
|
||
)
|
||
data = {
|
||
'name': '兼容客户',
|
||
'merchant': other_merchant.id,
|
||
}
|
||
|
||
response = self.client.post('/api/backend/customers/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
||
customer = Customer.objects.get(id=response.data['id'])
|
||
self.assertEqual(customer.merchant, self.merchant)
|
||
self.assertEqual(customer.created_by, self.employee)
|
||
|
||
def test_filter_customers_by_name(self):
|
||
"""测试通过 name 查询参数过滤客户"""
|
||
Customer.objects.create(merchant=self.merchant, name='客户A', created_by=self.employee)
|
||
Customer.objects.create(merchant=self.merchant, name='客户B', created_by=self.employee)
|
||
|
||
response = self.client.get('/api/backend/customers/?name=客户A')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 1)
|
||
self.assertEqual(results[0]['name'], '客户A')
|
||
|
||
def test_filter_customers_by_visible_employee_name(self):
|
||
"""测试通过 visible_employee_name 查询参数过滤客户"""
|
||
visible_employee = Employee.objects.create(
|
||
merchant=self.merchant,
|
||
name='李四业务',
|
||
)
|
||
other_employee = Employee.objects.create(
|
||
merchant=self.merchant,
|
||
name='王五跟单',
|
||
)
|
||
customer_a = Customer.objects.create(
|
||
merchant=self.merchant,
|
||
name='客户A',
|
||
created_by=self.employee,
|
||
)
|
||
customer_b = Customer.objects.create(
|
||
merchant=self.merchant,
|
||
name='客户B',
|
||
created_by=self.employee,
|
||
)
|
||
customer_a.visible_employees.add(visible_employee)
|
||
customer_b.visible_employees.add(other_employee)
|
||
|
||
response = self.client.get('/api/backend/customers/?visible_employee_name=李四')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 1)
|
||
self.assertEqual(results[0]['id'], customer_a.id)
|
||
|
||
def test_filter_customers_by_visible_employee_name_excludes_unbound_customers(self):
|
||
"""测试 visible_employee_name 过滤不会返回没有绑定员工的客户"""
|
||
visible_employee = Employee.objects.create(
|
||
merchant=self.merchant,
|
||
name='李四业务',
|
||
)
|
||
unbound_customer = Customer.objects.create(
|
||
merchant=self.merchant,
|
||
name='未绑定客户',
|
||
created_by=self.employee,
|
||
)
|
||
bound_customer = Customer.objects.create(
|
||
merchant=self.merchant,
|
||
name='已绑定客户',
|
||
created_by=self.employee,
|
||
)
|
||
bound_customer.visible_employees.add(visible_employee)
|
||
|
||
response = self.client.get('/api/backend/customers/?visible_employee_name=李四')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
returned_ids = {item['id'] for item in results}
|
||
self.assertIn(bound_customer.id, returned_ids)
|
||
self.assertNotIn(unbound_customer.id, returned_ids)
|
||
|
||
def test_filter_customers_by_visible_employee_name_returns_distinct_results(self):
|
||
"""测试 visible_employee_name 过滤不会因多名匹配员工导致客户重复"""
|
||
customer = Customer.objects.create(
|
||
merchant=self.merchant,
|
||
name='客户A',
|
||
created_by=self.employee,
|
||
)
|
||
customer.visible_employees.add(
|
||
Employee.objects.create(merchant=self.merchant, name='销售张三'),
|
||
Employee.objects.create(merchant=self.merchant, name='客服张三'),
|
||
)
|
||
|
||
response = self.client.get('/api/backend/customers/?visible_employee_name=张三')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 1)
|
||
self.assertEqual(results[0]['id'], customer.id)
|
||
|
||
|
||
class VehicleTypeAPITestCase(TestCase):
|
||
"""测试车辆类型 API（支持 name 查询参数）"""
|
||
|
||
def setUp(self):
|
||
self.merchant = Merchant.objects.create(
|
||
name='测试商户',
|
||
type=MerchantTypeEnum.STORE
|
||
)
|
||
self.user = User.objects.create_user(
|
||
username='testuser',
|
||
password='testpass123'
|
||
)
|
||
self.employee = Employee.objects.create(
|
||
merchant=self.merchant,
|
||
sys_user=self.user,
|
||
name='测试员工'
|
||
)
|
||
self.client = APIClient()
|
||
self.client.force_authenticate(user=self.user)
|
||
|
||
def test_create_vehicle_type(self):
|
||
"""测试创建车辆类型"""
|
||
data = {
|
||
'name': '测试车辆类型'
|
||
}
|
||
response = self.client.post('/api/backend/vehicle-types/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
||
self.assertEqual(response.data['name'], '测试车辆类型')
|
||
|
||
def test_filter_vehicle_types_by_name(self):
|
||
"""测试通过 name 查询参数过滤车辆类型"""
|
||
VehicleType.objects.create(merchant=self.merchant, name='类型A')
|
||
VehicleType.objects.create(merchant=self.merchant, name='类型B')
|
||
|
||
response = self.client.get('/api/backend/vehicle-types/?name=类型A')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 1)
|
||
self.assertEqual(results[0]['name'], '类型A')
|
||
|
||
|
||
class BankAccountAPITestCase(TestCase):
|
||
"""测试银行账户 API（支持 name 查询参数）"""
|
||
|
||
def setUp(self):
|
||
self.merchant = Merchant.objects.create(
|
||
name='测试商户',
|
||
type=MerchantTypeEnum.STORE
|
||
)
|
||
self.user = User.objects.create_user(
|
||
username='testuser',
|
||
password='testpass123'
|
||
)
|
||
self.employee = Employee.objects.create(
|
||
merchant=self.merchant,
|
||
sys_user=self.user,
|
||
name='测试员工'
|
||
)
|
||
self.client = APIClient()
|
||
self.client.force_authenticate(user=self.user)
|
||
|
||
def test_create_bank_account(self):
|
||
"""测试创建银行账户"""
|
||
data = {
|
||
'name': '测试账户',
|
||
'auto_number': 'ACC001'
|
||
}
|
||
response = self.client.post('/api/backend/bank-accounts/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
||
self.assertEqual(response.data['name'], '测试账户')
|
||
|
||
def test_filter_bank_accounts_by_name(self):
|
||
"""测试通过 name 查询参数过滤银行账户"""
|
||
BankAccount.objects.create(merchant=self.merchant, name='账户A', auto_number='A001')
|
||
BankAccount.objects.create(merchant=self.merchant, name='账户B', auto_number='B001')
|
||
|
||
response = self.client.get('/api/backend/bank-accounts/?name=账户A')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 1)
|
||
self.assertEqual(results[0]['name'], '账户A')
|
||
|
||
|
||
class DeviceInfoAPITestCase(TestCase):
|
||
"""测试设备信息 API（支持 name 查询参数）"""
|
||
|
||
def setUp(self):
|
||
self.merchant = Merchant.objects.create(
|
||
name='测试商户',
|
||
type=MerchantTypeEnum.STORE
|
||
)
|
||
self.user = User.objects.create_user(
|
||
username='testuser',
|
||
password='testpass123'
|
||
)
|
||
self.employee = Employee.objects.create(
|
||
merchant=self.merchant,
|
||
sys_user=self.user,
|
||
name='测试员工'
|
||
)
|
||
self.client = APIClient()
|
||
self.client.force_authenticate(user=self.user)
|
||
|
||
def test_create_device_info(self):
|
||
"""测试创建设备信息"""
|
||
data = {
|
||
'name': '测试设备'
|
||
}
|
||
response = self.client.post('/api/backend/device-info/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
||
self.assertEqual(response.data['name'], '测试设备')
|
||
|
||
def test_filter_device_info_by_name(self):
|
||
"""测试通过 name 查询参数过滤设备信息"""
|
||
DeviceInfo.objects.create(merchant=self.merchant, name='设备A')
|
||
DeviceInfo.objects.create(merchant=self.merchant, name='设备B')
|
||
|
||
response = self.client.get('/api/backend/device-info/?name=设备A')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 1)
|
||
self.assertEqual(results[0]['name'], '设备A')
|
||
|
||
|
||
class VehicleTransportRecordAPITestCase(TestCase):
|
||
"""测试司机车次 API（支持 driver_name 查询参数）"""
|
||
|
||
def setUp(self):
|
||
self.merchant = Merchant.objects.create(
|
||
name='测试商户',
|
||
type=MerchantTypeEnum.STORE
|
||
)
|
||
self.user = User.objects.create_user(
|
||
username='testuser',
|
||
password='testpass123'
|
||
)
|
||
self.employee = Employee.objects.create(
|
||
merchant=self.merchant,
|
||
sys_user=self.user,
|
||
name='测试员工'
|
||
)
|
||
self.vehicle_type = VehicleType.objects.create(
|
||
merchant=self.merchant,
|
||
name='测试车辆类型'
|
||
)
|
||
self.client = APIClient()
|
||
self.client.force_authenticate(user=self.user)
|
||
|
||
def test_create_vehicle_transport_record(self):
|
||
"""测试创建司机车次
|
||
|
||
注意：由于 VehicleTransportRecordSerializer 使用 depth=1 可能影响反序列化，
|
||
如果创建失败，可能是序列化器配置问题，但过滤功能测试仍然有效。
|
||
"""
|
||
from datetime import date
|
||
data = {
|
||
'vehicle_type': self.vehicle_type.id,
|
||
'driver_name': '测试司机',
|
||
'delivery_date': date.today().isoformat()
|
||
}
|
||
response = self.client.post('/api/backend/vehicle-transport-records/', data, format='json')
|
||
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
||
self.assertEqual(response.data['driver_name'], '测试司机')
|
||
self.assertEqual(response.data['vehicle_type'], self.vehicle_type.id)
|
||
|
||
def test_filter_vehicle_transport_records_by_driver_name(self):
|
||
"""测试通过 driver_name 查询参数过滤司机车次"""
|
||
from datetime import date
|
||
VehicleTransportRecord.objects.create(
|
||
merchant=self.merchant,
|
||
vehicle_type=self.vehicle_type,
|
||
driver_name='司机A',
|
||
delivery_date=date.today()
|
||
)
|
||
VehicleTransportRecord.objects.create(
|
||
merchant=self.merchant,
|
||
vehicle_type=self.vehicle_type,
|
||
driver_name='司机B',
|
||
delivery_date=date.today()
|
||
)
|
||
|
||
response = self.client.get('/api/backend/vehicle-transport-records/?driver_name=司机A')
|
||
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||
results = response.data['results'] if isinstance(response.data, dict) else response.data
|
||
self.assertEqual(len(results), 1)
|
||
self.assertEqual(results[0]['driver_name'], '司机A')
|