Skip to content

Commit baadce9

Browse files
potiukpotiuk
authored
Fix mocking in cncf.kubernetes tests after get_connections removal (#41805)
The #41733 removed deprecated get_connections but it was still used in cncf.kubernetes tests.
1 parent 615cddf commit baadce9

File tree

1 file changed

+20
-20
lines changed

1 file changed

+20
-20
lines changed

‎tests/providers/google/cloud/operators/test_kubernetes_engine.py

Lines changed: 20 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -342,7 +342,7 @@ def test_config_file_throws_error(self):
342342

343343
@mock.patch.dict(os.environ, {})
344344
@mock.patch(
345-
"airflow.hooks.base.BaseHook.get_connections",
345+
"airflow.hooks.base.BaseHook.get_connection",
346346
return_value=[Connection(extra=json.dumps({"keyfile_dict": '{"private_key": "r4nd0m_k3y"}'}))],
347347
)
348348
@mock.patch(KUB_OPERATOR_EXEC)
@@ -357,7 +357,7 @@ def test_execute_with_impersonation_service_account(
357357

358358
@mock.patch.dict(os.environ, {})
359359
@mock.patch(
360-
"airflow.hooks.base.BaseHook.get_connections",
360+
"airflow.hooks.base.BaseHook.get_connection",
361361
return_value=[Connection(extra=json.dumps({"keyfile_dict": '{"private_key": "r4nd0m_k3y"}'}))],
362362
)
363363
@mock.patch(KUB_OPERATOR_EXEC)
@@ -615,7 +615,7 @@ def test_execute_non_autoscaled_cluster_check_error(
615615

616616
@mock.patch.dict(os.environ, {})
617617
@mock.patch(
618-
"airflow.hooks.base.BaseHook.get_connections",
618+
"airflow.hooks.base.BaseHook.get_connection",
619619
return_value=[Connection(extra=json.dumps({"keyfile_dict": '{"private_key": "r4nd0m_k3y"}'}))],
620620
)
621621
@mock.patch(TEMP_FILE)
@@ -634,7 +634,7 @@ def test_execute_with_impersonation_service_account(
634634

635635
@mock.patch.dict(os.environ, {})
636636
@mock.patch(
637-
"airflow.hooks.base.BaseHook.get_connections",
637+
"airflow.hooks.base.BaseHook.get_connection",
638638
return_value=[Connection(extra=json.dumps({"keyfile_dict": '{"private_key": "r4nd0m_k3y"}'}))],
639639
)
640640
@mock.patch(TEMP_FILE)
@@ -942,7 +942,7 @@ def test_config_file_throws_error(self):
942942

943943
@mock.patch.dict(os.environ, {})
944944
@mock.patch(
945-
"airflow.hooks.base.BaseHook.get_connections",
945+
"airflow.hooks.base.BaseHook.get_connection",
946946
return_value=[Connection(extra=json.dumps({"keyfile_dict": '{"private_key": "r4nd0m_k3y"}'}))],
947947
)
948948
@mock.patch(KUB_JOB_OPERATOR_EXEC)
@@ -959,7 +959,7 @@ def test_execute_with_impersonation_service_account(
959959

960960
@mock.patch.dict(os.environ, {})
961961
@mock.patch(
962-
"airflow.hooks.base.BaseHook.get_connections",
962+
"airflow.hooks.base.BaseHook.get_connection",
963963
return_value=[Connection(extra=json.dumps({"keyfile_dict": '{"private_key": "r4nd0m_k3y"}'}))],
964964
)
965965
@mock.patch(KUB_JOB_OPERATOR_EXEC)
@@ -1042,7 +1042,7 @@ def test_execute(self, mock_job_hook, mock_hook, fetch_cluster_info_mock, file_m
10421042

10431043
@mock.patch.dict(os.environ, {})
10441044
@mock.patch(
1045-
"airflow.hooks.base.BaseHook.get_connections",
1045+
"airflow.hooks.base.BaseHook.get_connection",
10461046
return_value=[Connection(extra=json.dumps({"keyfile_dict": '{"private_key": "r4nd0m_k3y"}'}))],
10471047
)
10481048
@mock.patch(TEMP_FILE)
@@ -1060,7 +1060,7 @@ def test_execute_with_impersonation_service_account(
10601060

10611061
@mock.patch.dict(os.environ, {})
10621062
@mock.patch(
1063-
"airflow.hooks.base.BaseHook.get_connections",
1063+
"airflow.hooks.base.BaseHook.get_connection",
10641064
return_value=[Connection(extra=json.dumps({"keyfile_dict": '{"private_key": "r4nd0m_k3y"}'}))],
10651065
)
10661066
@mock.patch(TEMP_FILE)
@@ -1141,7 +1141,7 @@ def test_execute(self, mock_hook, fetch_cluster_info_mock, file_mock, exec_mock)
11411141

11421142
@mock.patch.dict(os.environ, {})
11431143
@mock.patch(
1144-
"airflow.hooks.base.BaseHook.get_connections",
1144+
"airflow.hooks.base.BaseHook.get_connection",
11451145
return_value=[Connection(extra=json.dumps({"keyfile_dict": '{"private_key": "r4nd0m_k3y"}'}))],
11461146
)
11471147
@mock.patch(KUB_CREATE_RES_OPERATOR_EXEC)
@@ -1158,7 +1158,7 @@ def test_execute_with_impersonation_service_account(
11581158

11591159
@mock.patch.dict(os.environ, {})
11601160
@mock.patch(
1161-
"airflow.hooks.base.BaseHook.get_connections",
1161+
"airflow.hooks.base.BaseHook.get_connection",
11621162
return_value=[Connection(extra=json.dumps({"keyfile_dict": '{"private_key": "r4nd0m_k3y"}'}))],
11631163
)
11641164
@mock.patch(KUB_CREATE_RES_OPERATOR_EXEC)
@@ -1202,7 +1202,7 @@ def test_execute(self, mock_hook, fetch_cluster_info_mock, file_mock, exec_mock)
12021202

12031203
@mock.patch.dict(os.environ, {})
12041204
@mock.patch(
1205-
"airflow.hooks.base.BaseHook.get_connections",
1205+
"airflow.hooks.base.BaseHook.get_connection",
12061206
return_value=[Connection(extra=json.dumps({"keyfile_dict": '{"private_key": "r4nd0m_k3y"}'}))],
12071207
)
12081208
@mock.patch(KUB_DELETE_RES_OPERATOR_EXEC)
@@ -1219,7 +1219,7 @@ def test_execute_with_impersonation_service_account(
12191219

12201220
@mock.patch.dict(os.environ, {})
12211221
@mock.patch(
1222-
"airflow.hooks.base.BaseHook.get_connections",
1222+
"airflow.hooks.base.BaseHook.get_connection",
12231223
return_value=[Connection(extra=json.dumps({"keyfile_dict": '{"private_key": "r4nd0m_k3y"}'}))],
12241224
)
12251225
@mock.patch(KUB_DELETE_RES_OPERATOR_EXEC)
@@ -1281,7 +1281,7 @@ def test_config_file_throws_error(self):
12811281

12821282
@mock.patch.dict(os.environ, {})
12831283
@mock.patch(
1284-
"airflow.hooks.base.BaseHook.get_connections",
1284+
"airflow.hooks.base.BaseHook.get_connection",
12851285
return_value=[Connection(extra=json.dumps({"keyfile_dict": '{"private_key": "r4nd0m_k3y"}'}))],
12861286
)
12871287
@mock.patch(KUB_JOB_OPERATOR_EXEC)
@@ -1298,7 +1298,7 @@ def test_execute_with_impersonation_service_account(
12981298

12991299
@mock.patch.dict(os.environ, {})
13001300
@mock.patch(
1301-
"airflow.hooks.base.BaseHook.get_connections",
1301+
"airflow.hooks.base.BaseHook.get_connection",
13021302
return_value=[Connection(extra=json.dumps({"keyfile_dict": '{"private_key": "r4nd0m_k3y"}'}))],
13031303
)
13041304
@mock.patch(KUB_JOB_OPERATOR_EXEC)
@@ -1393,7 +1393,7 @@ def test_config_file_throws_error(self):
13931393

13941394
@mock.patch.dict(os.environ, {})
13951395
@mock.patch(
1396-
"airflow.hooks.base.BaseHook.get_connections",
1396+
"airflow.hooks.base.BaseHook.get_connection",
13971397
return_value=[Connection(extra=json.dumps({"keyfile_dict": '{"private_key": "r4nd0m_k3y"}'}))],
13981398
)
13991399
@mock.patch(DEL_KUB_JOB_OPERATOR_EXEC)
@@ -1410,7 +1410,7 @@ def test_execute_with_impersonation_service_account(
14101410

14111411
@mock.patch.dict(os.environ, {})
14121412
@mock.patch(
1413-
"airflow.hooks.base.BaseHook.get_connections",
1413+
"airflow.hooks.base.BaseHook.get_connection",
14141414
return_value=[Connection(extra=json.dumps({"keyfile_dict": '{"private_key": "r4nd0m_k3y"}'}))],
14151415
)
14161416
@mock.patch(DEL_KUB_JOB_OPERATOR_EXEC)
@@ -1499,7 +1499,7 @@ def test_execute(self, mock_job_hook, mock_hook, fetch_cluster_info_mock, file_m
14991499

15001500
@mock.patch.dict(os.environ, {})
15011501
@mock.patch(
1502-
"airflow.hooks.base.BaseHook.get_connections",
1502+
"airflow.hooks.base.BaseHook.get_connection",
15031503
return_value=[Connection(extra=json.dumps({"keyfile_dict": '{"private_key": "r4nd0m_k3y"}'}))],
15041504
)
15051505
@mock.patch(TEMP_FILE)
@@ -1517,7 +1517,7 @@ def test_execute_with_impersonation_service_account(
15171517

15181518
@mock.patch.dict(os.environ, {})
15191519
@mock.patch(
1520-
"airflow.hooks.base.BaseHook.get_connections",
1520+
"airflow.hooks.base.BaseHook.get_connection",
15211521
return_value=[Connection(extra=json.dumps({"keyfile_dict": '{"private_key": "r4nd0m_k3y"}'}))],
15221522
)
15231523
@mock.patch(TEMP_FILE)
@@ -1607,7 +1607,7 @@ def test_execute(self, mock_job_hook, mock_hook, fetch_cluster_info_mock, file_m
16071607

16081608
@mock.patch.dict(os.environ, {})
16091609
@mock.patch(
1610-
"airflow.hooks.base.BaseHook.get_connections",
1610+
"airflow.hooks.base.BaseHook.get_connection",
16111611
return_value=[Connection(extra=json.dumps({"keyfile_dict": '{"private_key": "r4nd0m_k3y"}'}))],
16121612
)
16131613
@mock.patch(TEMP_FILE)
@@ -1625,7 +1625,7 @@ def test_execute_with_impersonation_service_account(
16251625

16261626
@mock.patch.dict(os.environ, {})
16271627
@mock.patch(
1628-
"airflow.hooks.base.BaseHook.get_connections",
1628+
"airflow.hooks.base.BaseHook.get_connection",
16291629
return_value=[Connection(extra=json.dumps({"keyfile_dict": '{"private_key": "r4nd0m_k3y"}'}))],
16301630
)
16311631
@mock.patch(TEMP_FILE)

0 commit comments

Comments
 (0)