""" 测试 PlateOrder 和 PrintingJob 的 has_started 属性 """ from django.test import TestCase from django.contrib.auth import get_user_model from printing import models as printing_models from basic_info import models as basic_info_models from stateflow import models as stateflow_models from stateflow import services as stateflow_services User = get_user_model() class HasStartedTestCase(TestCase): """测试 has_started 属性的各种情况""" def setUp(self): """设置测试数据""" self.user = User.objects.create_user(username='testuser', password='testpass') # 创建商户 self.merchant = basic_info_models.Merchant.objects.create( name='测试商户', type=basic_info_models.MerchantTypeEnum.STORE ) # 创建客户 self.customer = basic_info_models.Customer.objects.create( name='测试客户', merchant=self.merchant, mobile='13800138000' ) # 创建状态和流程 self.state1 = stateflow_models.State.objects.create(name='状态1') self.state2 = stateflow_models.State.objects.create(name='状态2') self.state3 = stateflow_models.State.objects.create(name='状态3') self.process = stateflow_models.Process.objects.create(name='测试流程') stateflow_models.ProcessNode.objects.create(process=self.process, state=self.state1, order=0) stateflow_models.ProcessNode.objects.create(process=self.process, state=self.state2, order=1) stateflow_models.ProcessNode.objects.create(process=self.process, state=self.state3, order=2) def test_plate_order_has_started_without_business_object(self): """测试没有 business_object 时 has_started 为 False""" plate_order = printing_models.PlateOrder.objects.create( design_code='TEST001', customer=self.customer ) # 没有 business_object，has_started 应该是 False self.assertFalse(plate_order.has_started) def test_plate_order_has_started_with_business_object_no_records(self): """测试有 business_object 但没有状态流转记录时 has_started 为 False""" # 手动创建 business_object business_object = stateflow_models.BusinessObject.objects.create( name='测试业务对象', process=self.process ) plate_order = printing_models.PlateOrder.objects.create( design_code='TEST002', customer=self.customer, business_object=business_object ) # 有 business_object 但没有状态流转记录，has_started 应该是 False self.assertFalse(plate_order.has_started) # 验证确实没有记录 self.assertEqual(business_object.state_logs.count(), 0) def test_plate_order_has_started_after_advance(self): """测试推进状态后 has_started 为 True""" # 手动创建 business_object business_object = stateflow_models.BusinessObject.objects.create( name='测试业务对象', process=self.process ) plate_order = printing_models.PlateOrder.objects.create( design_code='TEST003', customer=self.customer, business_object=business_object ) # 推进一次 stateflow_services.advance_to_next_state(business_object, self.user) # 现在应该有记录了 self.assertEqual(business_object.state_logs.count(), 1) # has_started 应该是 True self.assertTrue(plate_order.has_started) def test_plate_order_has_started_after_step_back(self): """测试回退后 has_started 变为 False（因为只有已撤销的记录）""" # 手动创建 business_object business_object = stateflow_models.BusinessObject.objects.create( name='测试业务对象', process=self.process ) plate_order = printing_models.PlateOrder.objects.create( design_code='TEST004', customer=self.customer, business_object=business_object ) # 推进一次 stateflow_services.advance_to_next_state(business_object, self.user) self.assertTrue(plate_order.has_started) # 回退 stateflow_services.step_back_one_state(business_object, self.user) # 验证有已撤销的记录 self.assertEqual(business_object.state_logs.count(), 1) self.assertTrue(business_object.state_logs.first().is_cancelled) # has_started 应该变为 False（因为只检查未撤销的记录） self.assertFalse(plate_order.has_started) def test_plate_order_has_started_after_reset(self): """测试重置后 has_started 变为 False（因为记录都被标记为已撤销）""" # 手动创建 business_object business_object = stateflow_models.BusinessObject.objects.create( name='测试业务对象', process=self.process ) plate_order = printing_models.PlateOrder.objects.create( design_code='TEST005', customer=self.customer, business_object=business_object ) # 推进两次 stateflow_services.advance_to_next_state(business_object, self.user) stateflow_services.advance_to_next_state(business_object, self.user) self.assertTrue(plate_order.has_started) # 重置 stateflow_services.reset_business_object_progress(business_object) # 验证记录还在但都被撤销了 self.assertEqual(business_object.state_logs.count(), 2) self.assertTrue(all(log.is_cancelled for log in business_object.state_logs.all())) # has_started 应该变为 False（因为没有未撤销的记录） self.assertFalse(plate_order.has_started) def test_plate_order_serializer_includes_has_started(self): """测试序列化器包含 has_started 字段""" # 手动创建 business_object business_object = stateflow_models.BusinessObject.objects.create( name='测试业务对象', process=self.process ) plate_order = printing_models.PlateOrder.objects.create( design_code='TEST200', customer=self.customer, business_object=business_object ) # 验证 has_started 是一个属性 self.assertFalse(plate_order.has_started) # 推进后 stateflow_services.advance_to_next_state(business_object, self.user) # 重新获取以确保属性被重新计算 plate_order.refresh_from_db() self.assertTrue(plate_order.has_started) def test_printing_order_can_change_process(self): """测试 PrintingOrder.can_change_process() 使用正确的 has_started 逻辑""" # 创建 PrintingOrder printing_order = printing_models.PrintingOrder.objects.create( customer=self.customer, fabric='测试布料', width='150cm' ) # 创建 Product 和 Category category = basic_info_models.ProductCategory.objects.create( name='测试分类', merchant=self.merchant ) product = basic_info_models.Product.objects.create( name='测试产品', merchant=self.merchant, category=category ) # 创建 PrintingJob business_object = stateflow_models.BusinessObject.objects.create( name='测试业务对象', process=self.process ) printing_job = printing_models.PrintingJob.objects.create( printing_order=printing_order, product=product, quantity=100, unit='米', business_object=business_object ) # 未推进时，可以修改流程 self.assertTrue(printing_order.can_change_process()) self.assertFalse(printing_job.has_started) # 推进后，不能修改流程 stateflow_services.advance_to_next_state(business_object, self.user) self.assertFalse(printing_order.can_change_process()) self.assertTrue(printing_job.has_started) # 回退后，又可以修改流程了（因为 has_started 变为 False） stateflow_services.step_back_one_state(business_object, self.user) self.assertTrue(printing_order.can_change_process()) self.assertFalse(printing_job.has_started) def test_has_started_query_performance(self): """测试 has_started 的查询逻辑（排除已撤销记录）""" # 创建多个 PlateOrder business_object1 = stateflow_models.BusinessObject.objects.create( name='测试业务对象1', process=self.process ) business_object2 = stateflow_models.BusinessObject.objects.create( name='测试业务对象2', process=self.process ) plate_order1 = printing_models.PlateOrder.objects.create( design_code='TEST101', customer=self.customer, business_object=business_object1 ) plate_order2 = printing_models.PlateOrder.objects.create( design_code='TEST102', customer=self.customer, business_object=business_object2 ) # 只推进第一个 stateflow_services.advance_to_next_state(business_object1, self.user) # 验证 self.assertTrue(plate_order1.has_started) self.assertFalse(plate_order2.has_started) # 推进第二个 stateflow_services.advance_to_next_state(business_object2, self.user) # 现在都应该是 True self.assertTrue(plate_order1.has_started) self.assertTrue(plate_order2.has_started) # 回退第一个 stateflow_services.step_back_one_state(business_object1, self.user) # 第一个应该变为 False，第二个保持 True self.assertFalse(plate_order1.has_started) self.assertTrue(plate_order2.has_started)