From 19d5aea7f617391026c652825bc4588dc9c7f19b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Malo=20Jaffr=C3=A9?=
 <8469951+ghjklw@users.noreply.github.com>
Date: Mon, 24 Feb 2025 15:19:45 +0000
Subject: [PATCH 1/2] Bugfix operator_args override configuration

---
 cosmos/converter.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cosmos/converter.py b/cosmos/converter.py
index 7c917a0225..b0716a01bb 100644
--- a/cosmos/converter.py
+++ b/cosmos/converter.py
@@ -223,7 +223,7 @@ def override_configuration(
     if execution_config.invocation_mode:
         operator_args["invocation_mode"] = execution_config.invocation_mode
 
-    if execution_config in (ExecutionMode.LOCAL, ExecutionMode.VIRTUALENV):
+    if execution_config.execution_mode in (ExecutionMode.LOCAL, ExecutionMode.VIRTUALENV):
         if "install_deps" not in operator_args:
             operator_args["install_deps"] = project_config.install_dbt_deps
 

From 8e50a4c13869ddb7998fed33a4487e82b6f2239c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Malo=20Jaffr=C3=A9?=
 <8469951+ghjklw@users.noreply.github.com>
Date: Tue, 25 Feb 2025 00:22:19 +0000
Subject: [PATCH 2/2] Unit tests install_dbt_deps

---
 tests/test_converter.py | 57 +++++++++++++++++++++++++++++++++++++++--
 1 file changed, 55 insertions(+), 2 deletions(-)

diff --git a/tests/test_converter.py b/tests/test_converter.py
index 6b7d9181ff..8e1b724e03 100644
--- a/tests/test_converter.py
+++ b/tests/test_converter.py
@@ -96,9 +96,11 @@ def test_validate_initial_user_config_expects_profile(execution_mode):
     assert validate_initial_user_config(execution_config, profile_config, project_config, None, {}) is None
 
 
-@pytest.mark.parametrize("operator_args", [{"env": {"key": "value"}}, {"vars": {"key": "value"}}])
+@pytest.mark.parametrize(
+    "operator_args", [{"env": {"key": "value"}}, {"vars": {"key": "value"}}, {"install_deps": {"key": "value"}}]
+)
 def test_validate_user_config_operator_args_deprecated(operator_args):
-    """Deprecating warnings should be raised when using operator_args with "vars" or "env"."""
+    """Deprecating warnings should be raised when using operator_args with "vars", "env" or "install_deps"."""
     project_config = ProjectConfig()
     execution_config = ExecutionConfig()
     render_config = RenderConfig()
@@ -709,6 +711,57 @@ def test_validate_converter_fetches_project_name_from_render_config(
     assert mock_build_airflow_graph.call_args.kwargs["dbt_project_name"] == "project1"
 
 
+@pytest.mark.parametrize(
+    "execution_mode,operator_args,install_dbt_deps,expected",
+    [
+        (ExecutionMode.KUBERNETES, {}, False, None),
+        (ExecutionMode.LOCAL, {}, False, False),
+        (ExecutionMode.VIRTUALENV, {}, False, False),
+        (ExecutionMode.LOCAL, {}, True, True),
+        (ExecutionMode.VIRTUALENV, {}, True, True),
+        (ExecutionMode.KUBERNETES, {"install_deps": True}, False, True),
+        (ExecutionMode.LOCAL, {"install_deps": True}, False, True),
+        (ExecutionMode.VIRTUALENV, {"install_deps": True}, False, True),
+    ],
+)
+@patch("cosmos.config.ProjectConfig.validate_project")
+@patch("cosmos.converter.validate_initial_user_config")
+@patch("cosmos.converter.DbtGraph")
+@patch("cosmos.converter.build_airflow_graph")
+def test_project_config_install_dbt_deps_overrides_operator_args(
+    mock_build_airflow_graph,
+    mock_user_config,
+    mock_dbt_graph,
+    mock_validate_project,
+    execution_mode,
+    operator_args,
+    install_dbt_deps,
+    expected,
+):
+    """Tests that the value project_config.install_dbt_deps is used to define operator_args["install_deps"] if
+    execution mode is ExecutionMode.LOCAL or ExecutionMode.VIRTUALENV and operator_args["install_deps"] is not
+    already defined.
+    """
+    project_config = ProjectConfig(project_name="fake-project", dbt_project_path="/some/project/path")
+    project_config.install_dbt_deps = install_dbt_deps
+    execution_config = ExecutionConfig(execution_mode=execution_mode)
+    render_config = MagicMock()
+    profile_config = MagicMock()
+    with DAG("test-id", start_date=datetime(2022, 1, 1)) as dag:
+        DbtToAirflowConverter(
+            dag=dag,
+            nodes=nodes,
+            project_config=project_config,
+            profile_config=profile_config,
+            execution_config=execution_config,
+            render_config=render_config,
+            operator_args=operator_args,
+        )
+    _, kwargs = mock_build_airflow_graph.call_args
+
+    assert kwargs["task_args"].get("install_deps", None) == expected
+
+
 @pytest.mark.parametrize("invocation_mode", [None, InvocationMode.SUBPROCESS, InvocationMode.DBT_RUNNER])
 @patch("cosmos.config.ProjectConfig.validate_project")
 @patch("cosmos.converter.validate_initial_user_config")