erpnew/stateflow/tests/test_step_back.py

"""
测试单步回退功能
"""
from django.test import TestCase
from django.contrib.auth import get_user_model
from stateflow import models, services

User = get_user_model()


class StepBackTestCase(TestCase):
    """测试单步回退功能"""
    
    def setUp(self):
        # 创建测试用户
        self.user = User.objects.create_user(username='testuser', password='testpass')
        
        # 创建状态
        self.state1 = models.State.objects.create(name='状态1', description='第一个状态')
        self.state2 = models.State.objects.create(name='状态2', description='第二个状态')
        self.state3 = models.State.objects.create(name='状态3', description='第三个状态')
        
        # 创建流程
        self.process = models.Process.objects.create(name='测试流程', description='三个状态的测试流程')
        
        # 创建流程节点
        models.ProcessNode.objects.create(process=self.process, state=self.state1, order=1)
        models.ProcessNode.objects.create(process=self.process, state=self.state2, order=2)
        models.ProcessNode.objects.create(process=self.process, state=self.state3, order=3)
        
        # 创建业务对象
        self.business_object = models.BusinessObject.objects.create(
            name='测试业务对象',
            process=self.process,
            description='测试回退功能'
        )
    
    def test_step_back_from_not_started(self):
        """测试从未开始状态回退（应该返回失败）"""
        success, message = services.step_back_one_state(self.business_object, self.user)
        
        self.assertFalse(success)
        self.assertIn('没有任何状态流转记录', message)
        
        # 验证仍然是未开始状态
        self.assertEqual(services.get_overall_status(self.business_object), 'not_started')
    
    def test_step_back_from_first_state(self):
        """测试从第一个状态回退到未开始"""
        # 推进到第一个状态
        success, _ = services.advance_to_next_state(self.business_object, self.user)
        self.assertTrue(success)
        
        # 验证当前在第一个状态（进行中，准备完成第二个状态）
        current_state = services.get_business_object_current_state(self.business_object)
        self.assertEqual(current_state.id, self.state2.id)
        
        # 回退一步
        success, message = services.step_back_one_state(self.business_object, self.user)
        self.assertTrue(success)
        self.assertIn('状态1', message)
        
        # 验证回到未开始状态
        self.assertEqual(services.get_overall_status(self.business_object), 'not_started')
        
        # 验证状态1的记录已被撤销
        record = models.StateFlowRecord.objects.filter(
            business_object=self.business_object,
            state=self.state1
        ).first()
        self.assertTrue(record.is_cancelled)
        self.assertIsNotNone(record.cancelled_at)
    
    def test_step_back_from_middle_state(self):
        """测试从中间状态回退"""
        # 推进到第二个状态
        services.advance_to_next_state(self.business_object, self.user)
        services.advance_to_next_state(self.business_object, self.user)
        
        # 验证当前在第二个状态（进行中，准备完成第三个状态）
        current_state = services.get_business_object_current_state(self.business_object)
        self.assertEqual(current_state.id, self.state3.id)
        
        # 回退一步
        success, message = services.step_back_one_state(self.business_object, self.user)
        self.assertTrue(success)
        self.assertIn('状态2', message)
        
        # 验证回到第一个状态（进行中，准备完成第二个状态）
        current_state = services.get_business_object_current_state(self.business_object)
        self.assertEqual(current_state.id, self.state2.id)
        
        # 验证状态2的记录已被撤销
        record = models.StateFlowRecord.objects.filter(
            business_object=self.business_object,
            state=self.state2
        ).first()
        self.assertTrue(record.is_cancelled)
    
    def test_step_back_from_completed(self):
        """测试从已完成状态回退"""
        # 推进到完成
        services.advance_to_next_state(self.business_object, self.user)
        services.advance_to_next_state(self.business_object, self.user)
        services.advance_to_next_state(self.business_object, self.user)
        
        # 验证已完成
        self.assertEqual(services.get_overall_status(self.business_object), 'completed')
        
        # 回退一步
        success, message = services.step_back_one_state(self.business_object, self.user)
        self.assertTrue(success)
        self.assertIn('状态3', message)
        
        # 验证回到进行中状态（准备完成第三个状态）
        self.assertEqual(services.get_overall_status(self.business_object), 'in_progress')
        current_state = services.get_business_object_current_state(self.business_object)
        self.assertEqual(current_state.id, self.state3.id)
    
    def test_step_back_multiple_times(self):
        """测试多次回退"""
        # 推进到完成
        services.advance_to_next_state(self.business_object, self.user)
        services.advance_to_next_state(self.business_object, self.user)
        services.advance_to_next_state(self.business_object, self.user)
        
        # 第一次回退
        success, _ = services.step_back_one_state(self.business_object, self.user)
        self.assertTrue(success)
        current = services.get_business_object_current_state(self.business_object)
        self.assertEqual(current.id, self.state3.id)
        
        # 第二次回退
        success, _ = services.step_back_one_state(self.business_object, self.user)
        self.assertTrue(success)
        current = services.get_business_object_current_state(self.business_object)
        self.assertEqual(current.id, self.state2.id)
        
        # 第三次回退
        success, _ = services.step_back_one_state(self.business_object, self.user)
        self.assertTrue(success)
        self.assertEqual(services.get_overall_status(self.business_object), 'not_started')
        
        # 第四次回退（应该失败）
        success, message = services.step_back_one_state(self.business_object, self.user)
        self.assertFalse(success)
        self.assertIn('没有任何状态流转记录', message)
    
    def test_advance_after_step_back(self):
        """测试回退后再前进"""
        # 推进两步
        services.advance_to_next_state(self.business_object, self.user)
        services.advance_to_next_state(self.business_object, self.user)
        
        # 回退一步
        services.step_back_one_state(self.business_object, self.user)
        
        # 再前进
        success, _ = services.advance_to_next_state(self.business_object, self.user)
        self.assertTrue(success)
        
        # 验证当前在第二个状态
        current_state = services.get_business_object_current_state(self.business_object)
        self.assertEqual(current_state.id, self.state3.id)
    
    def test_step_back_preserves_history(self):
        """测试回退操作保留历史记录"""
        # 推进两步
        services.advance_to_next_state(self.business_object, self.user)
        services.advance_to_next_state(self.business_object, self.user)
        
        # 回退一步
        services.step_back_one_state(self.business_object, self.user)
        
        # 验证记录仍然存在，只是被标记为已撤销
        all_records = models.StateFlowRecord.objects.filter(
            business_object=self.business_object
        ).order_by('completed_at')
        
        self.assertEqual(all_records.count(), 2)
        self.assertFalse(all_records[0].is_cancelled)  # 状态1未撤销
        self.assertTrue(all_records[1].is_cancelled)   # 状态2已撤销