@@ -114,10 +114,10 @@ def test_successful_fetch_all_environments(
114114 mock_iam_client .return_value = mock_iam_client_instance
115115
116116
117- mock_client_instance .get_environment_list .return_value = (
117+ mock_client_instance .get_environments .return_value = (
118118 HttpStatus .OK , None , mock_environments
119119 )
120- mock_client_instance .get_kafka_cluster_list .return_value = (
120+ mock_client_instance .get_kafka_clusters .return_value = (
121121 HttpStatus .OK , None , [mock_kafka_clusters [0 ]]
122122 )
123123 mock_iam_client_instance .create_api_key .return_value = (
@@ -149,7 +149,7 @@ def test_environment_list_failure(
149149 mock_client_instance = Mock ()
150150 mock_environment_client .return_value = mock_client_instance
151151
152- mock_client_instance .get_environment_list .return_value = (
152+ mock_client_instance .get_environments .return_value = (
153153 HttpStatus .INTERNAL_SERVER_ERROR , "API Error" , []
154154 )
155155
@@ -172,10 +172,10 @@ def test_kafka_cluster_list_failure(
172172 mock_client_instance = Mock ()
173173 mock_environment_client .return_value = mock_client_instance
174174
175- mock_client_instance .get_environment_list .return_value = (
175+ mock_client_instance .get_environments .return_value = (
176176 HttpStatus .OK , None , mock_environments
177177 )
178- mock_client_instance .get_kafka_cluster_list .return_value = (
178+ mock_client_instance .get_kafka_clusters .return_value = (
179179 HttpStatus .INTERNAL_SERVER_ERROR , "Cluster API Error" , []
180180 )
181181
@@ -204,10 +204,10 @@ def test_environment_filter(
204204 mock_iam_client_instance = Mock ()
205205 mock_iam_client .return_value = mock_iam_client_instance
206206
207- mock_client_instance .get_environment_list .return_value = (
207+ mock_client_instance .get_environments .return_value = (
208208 HttpStatus .OK , None , mock_environments
209209 )
210- mock_client_instance .get_kafka_cluster_list .return_value = (
210+ mock_client_instance .get_kafka_clusters .return_value = (
211211 HttpStatus .OK , None , [mock_kafka_clusters [0 ]]
212212 )
213213 mock_iam_client_instance .create_api_key .return_value = (
@@ -223,7 +223,7 @@ def test_environment_filter(
223223
224224 # Assert - only one environment should be processed
225225 assert len (result ) == 1
226- mock_client_instance .get_kafka_cluster_list .assert_called_once_with (
226+ mock_client_instance .get_kafka_clusters .assert_called_once_with (
227227 environment_id = "env-1"
228228 )
229229
@@ -244,10 +244,10 @@ def test_kafka_cluster_filter(
244244 mock_iam_client_instance = Mock ()
245245 mock_iam_client .return_value = mock_iam_client_instance
246246
247- mock_client_instance .get_environment_list .return_value = (
247+ mock_client_instance .get_environments .return_value = (
248248 HttpStatus .OK , None , [mock_environments [0 ]]
249249 )
250- mock_client_instance .get_kafka_cluster_list .return_value = (
250+ mock_client_instance .get_kafka_clusters .return_value = (
251251 HttpStatus .OK , None , mock_kafka_clusters
252252 )
253253 mock_iam_client_instance .create_api_key .return_value = (
@@ -281,10 +281,10 @@ def test_forbidden_access(
281281 mock_iam_client_instance = Mock ()
282282 mock_iam_client .return_value = mock_iam_client_instance
283283
284- mock_client_instance .get_environment_list .return_value = (
284+ mock_client_instance .get_environments .return_value = (
285285 HttpStatus .OK , None , [mock_environments [0 ]]
286286 )
287- mock_client_instance .get_kafka_cluster_list .return_value = (
287+ mock_client_instance .get_kafka_clusters .return_value = (
288288 HttpStatus .OK , None , [mock_kafka_clusters [0 ]]
289289 )
290290 mock_iam_client_instance .create_api_key .return_value = (
@@ -315,10 +315,10 @@ def test_api_key_creation_failure_returns_empty(
315315 mock_iam_client_instance = Mock ()
316316 mock_iam_client .return_value = mock_iam_client_instance
317317
318- mock_client_instance .get_environment_list .return_value = (
318+ mock_client_instance .get_environments .return_value = (
319319 HttpStatus .OK , None , [mock_environments [0 ]]
320320 )
321- mock_client_instance .get_kafka_cluster_list .return_value = (
321+ mock_client_instance .get_kafka_clusters .return_value = (
322322 HttpStatus .OK , None , mock_kafka_clusters
323323 )
324324
@@ -360,10 +360,10 @@ def test_cleanup_deletion_failure(
360360 mock_iam_client_instance = Mock ()
361361 mock_iam_client .return_value = mock_iam_client_instance
362362
363- mock_client_instance .get_environment_list .return_value = (
363+ mock_client_instance .get_environments .return_value = (
364364 HttpStatus .OK , None , [mock_environments [0 ]]
365365 )
366- mock_client_instance .get_kafka_cluster_list .return_value = (
366+ mock_client_instance .get_kafka_clusters .return_value = (
367367 HttpStatus .OK , None , mock_kafka_clusters
368368 )
369369
@@ -402,10 +402,10 @@ def test_multiple_environment_ids_filter(
402402 mock_iam_client_instance = Mock ()
403403 mock_iam_client .return_value = mock_iam_client_instance
404404
405- mock_client_instance .get_environment_list .return_value = (
405+ mock_client_instance .get_environments .return_value = (
406406 HttpStatus .OK , None , mock_environments
407407 )
408- mock_client_instance .get_kafka_cluster_list .return_value = (
408+ mock_client_instance .get_kafka_clusters .return_value = (
409409 HttpStatus .OK , None , [mock_kafka_clusters [0 ]]
410410 )
411411 mock_iam_client_instance .create_api_key .return_value = (
@@ -433,11 +433,11 @@ def test_no_kafka_credentials_found(
433433 mock_client_instance = Mock ()
434434 mock_environment_client .return_value = mock_client_instance
435435
436- mock_client_instance .get_environment_list .return_value = (
436+ mock_client_instance .get_environments .return_value = (
437437 HttpStatus .OK , None , mock_environments
438438 )
439439 # No Kafka clusters in environments
440- mock_client_instance .get_kafka_cluster_list .return_value = (
440+ mock_client_instance .get_kafka_clusters .return_value = (
441441 HttpStatus .OK , None , []
442442 )
443443
0 commit comments