2020from unittest import mock
2121from dapr .ext .workflow .dapr_workflow_client import DaprWorkflowClient
2222from durabletask import client
23+ import durabletask .internal .orchestrator_service_pb2 as pb
2324
2425mock_schedule_result = 'workflow001'
2526mock_raise_event_result = 'event001'
2627mock_terminate_result = 'terminate001'
2728mock_suspend_result = 'suspend001'
2829mock_resume_result = 'resume001'
29- mockInstanceId = 'instance001'
30+ mock_purge_result = 'purge001'
31+ mock_instance_id = 'instance001'
3032
3133
3234class FakeTaskHubGrpcClient :
33- def schedule_new_orchestration (self , workflow , input , instance_id , start_at ):
35+ def schedule_new_orchestration (
36+ self ,
37+ workflow ,
38+ input ,
39+ instance_id ,
40+ start_at ,
41+ reuse_id_policy : Union [pb .OrchestrationIdReusePolicy , None ] = None ,
42+ ):
3443 return mock_schedule_result
3544
3645 def get_orchestration_state (self , instance_id , fetch_payloads ):
@@ -49,7 +58,9 @@ def raise_orchestration_event(
4958 ):
5059 return mock_raise_event_result
5160
52- def terminate_orchestration (self , instance_id : str , * , output : Union [Any , None ] = None ):
61+ def terminate_orchestration (
62+ self , instance_id : str , * , output : Union [Any , None ] = None , recursive : bool = True
63+ ):
5364 return mock_terminate_result
5465
5566 def suspend_orchestration (self , instance_id : str ):
@@ -58,6 +69,9 @@ def suspend_orchestration(self, instance_id: str):
5869 def resume_orchestration (self , instance_id : str ):
5970 return mock_resume_result
6071
72+ def purge_orchestration (self , instance_id : str , recursive : bool = True ):
73+ return mock_purge_result
74+
6175 def _inner_get_orchestration_state (self , instance_id , state : client .OrchestrationStatus ):
6276 return client .OrchestrationState (
6377 instance_id = instance_id ,
@@ -87,35 +101,38 @@ def test_client_functions(self):
87101 assert actual_schedule_result == mock_schedule_result
88102
89103 actual_get_result = wfClient .get_workflow_state (
90- instance_id = mockInstanceId , fetch_payloads = True
104+ instance_id = mock_instance_id , fetch_payloads = True
91105 )
92106 assert actual_get_result .runtime_status .name == 'PENDING'
93- assert actual_get_result .instance_id == mockInstanceId
107+ assert actual_get_result .instance_id == mock_instance_id
94108
95109 actual_wait_start_result = wfClient .wait_for_workflow_start (
96- instance_id = mockInstanceId , timeout_in_seconds = 30
110+ instance_id = mock_instance_id , timeout_in_seconds = 30
97111 )
98112 assert actual_wait_start_result .runtime_status .name == 'RUNNING'
99- assert actual_wait_start_result .instance_id == mockInstanceId
113+ assert actual_wait_start_result .instance_id == mock_instance_id
100114
101115 actual_wait_completion_result = wfClient .wait_for_workflow_completion (
102- instance_id = mockInstanceId , timeout_in_seconds = 30
116+ instance_id = mock_instance_id , timeout_in_seconds = 30
103117 )
104118 assert actual_wait_completion_result .runtime_status .name == 'COMPLETED'
105- assert actual_wait_completion_result .instance_id == mockInstanceId
119+ assert actual_wait_completion_result .instance_id == mock_instance_id
106120
107121 actual_raise_event_result = wfClient .raise_workflow_event (
108- instance_id = mockInstanceId , event_name = 'test_event' , data = 'test_data'
122+ instance_id = mock_instance_id , event_name = 'test_event' , data = 'test_data'
109123 )
110124 assert actual_raise_event_result == mock_raise_event_result
111125
112126 actual_terminate_result = wfClient .terminate_workflow (
113- instance_id = mockInstanceId , output = 'test_output'
127+ instance_id = mock_instance_id , output = 'test_output'
114128 )
115129 assert actual_terminate_result == mock_terminate_result
116130
117- actual_suspend_result = wfClient .pause_workflow (instance_id = mockInstanceId )
131+ actual_suspend_result = wfClient .pause_workflow (instance_id = mock_instance_id )
118132 assert actual_suspend_result == mock_suspend_result
119133
120- actual_resume_result = wfClient .resume_workflow (instance_id = mockInstanceId )
134+ actual_resume_result = wfClient .resume_workflow (instance_id = mock_instance_id )
121135 assert actual_resume_result == mock_resume_result
136+
137+ actual_purge_result = wfClient .purge_workflow (instance_id = mock_instance_id )
138+ assert actual_purge_result == mock_purge_result
![dependabot[bot]](https://avatars.githubusercontent.com/in/29110?v=4&size=40){kind=avatar}
0 commit comments