@@ -113,7 +113,7 @@ def test_happy_path(self, target_state, previous_state):
113
113
for state in [previous_state , target_state ]
114
114
]
115
115
)
116
- cmd_clusters ._wait_for_cluster_state (client , "test-cluster" , target_state , check_timeout = 0.1 )
116
+ cmd_clusters ._wait_for_cluster_state (client , "test-cluster" , target_state , poll_duration = 0.1 )
117
117
assert client .list_call_count == 2
118
118
119
119
@pytest .mark .parametrize ("target_state" , [V1ClusterState .RUNNING , V1ClusterState .DELETED ])
@@ -129,9 +129,7 @@ def test_times_out(self, target_state):
129
129
consume = False ,
130
130
)
131
131
with pytest .raises (click .ClickException ) as e :
132
- cmd_clusters ._wait_for_cluster_state (
133
- client , "test-cluster" , target_state , max_wait_time = 0.4 , check_timeout = 0.2
134
- )
132
+ cmd_clusters ._wait_for_cluster_state (client , "test-cluster" , target_state , timeout = 0.4 , poll_duration = 0.2 )
135
133
assert "Max wait time elapsed" in str (e .value )
136
134
137
135
@mock .patch ("click.echo" )
@@ -165,8 +163,8 @@ def test_echo_state_change_on_desired_running(self, echo: MagicMock):
165
163
client ,
166
164
"test-cluster" ,
167
165
target_state = V1ClusterState .RUNNING ,
168
- max_wait_time = 0.6 ,
169
- check_timeout = 0.1 ,
166
+ timeout = 0.6 ,
167
+ poll_duration = 0.1 ,
170
168
)
171
169
172
170
assert client .list_call_count == 7
@@ -229,8 +227,8 @@ def test_echo_state_change_on_desired_deleted(self, echo: MagicMock):
229
227
client ,
230
228
"test-cluster" ,
231
229
target_state = V1ClusterState .DELETED ,
232
- max_wait_time = 0.4 ,
233
- check_timeout = 0.1 ,
230
+ timeout = 0.4 ,
231
+ poll_duration = 0.1 ,
234
232
)
235
233
236
234
assert client .list_call_count == 4
0 commit comments