diff --git a/tests/operators/test_kubernetes.py b/tests/operators/test_kubernetes.py index d0bbb565d6..be8624942a 100644 --- a/tests/operators/test_kubernetes.py +++ b/tests/operators/test_kubernetes.py @@ -226,13 +226,13 @@ 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" +def test_created_pod(): ls_kwargs = {"env_vars": {"FOO": "BAR"}} ls_kwargs.update(base_kwargs) ls_operator = DbtLSKubernetesOperator(**ls_kwargs) + ls_operator.hook = MagicMock() + ls_operator.hook.is_in_cluster = False + ls_operator.hook._get_namespace.return_value.to_dict.return_value = "foo" ls_operator.build_kube_args(context={}, cmd_flags=MagicMock()) pod_obj = ls_operator.build_pod_request_obj() expected_result = {