clean up

astronomer · Jan 31, 2024 · 155a9d9 · 155a9d9
1 parent 6f82835
commit 155a9d9
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/tests/operators/test_kubernetes.py b/tests/operators/test_kubernetes.py
@@ -226,15 +226,10 @@ def cleanup(pod: str, remote_pod: str):
     test_operator._handle_warnings(context)
 
 
-@patch("airflow.providers.cncf.kubernetes.operators.pod.KubernetesPodOperator.hook")
 def test_created_pod(test_hook):
-    test_hook.is_in_cluster = False
-    test_hook._get_namespace.return_value.to_dict.return_value = "foo"
     ls_kwargs = {"env_vars": {"FOO": "BAR"}}
     ls_kwargs.update(base_kwargs)
     ls_operator = DbtLSKubernetesOperator(**ls_kwargs)
-    from unittest.mock import MagicMock
-
     ls_operator.hook = MagicMock()
     ls_operator.hook.is_in_cluster = False
     ls_operator.hook._get_namespace.return_value.to_dict.return_value = "foo"