feat: [SNOW-1890085] dbt deploy: use --force to create or replace

Author: sfc-gh-jwilkowski

src/snowflake/cli/_plugins/dbt/manager.py

@@ -52,7 +52,6 @@ def deploy(
         dbt_adapter_version: str,
         force: bool,
     ) -> SnowflakeCursor:
-        # TODO: what to do with force?
         dbt_project_path = path.joinpath("dbt_project.yml")
         if not dbt_project_path.exists():
             raise ClickException(f"dbt_project.yml does not exist in provided path.")
@@ -79,7 +78,7 @@ def deploy(
 
         with cli_console.phase("Creating DBT project"):
             staged_dbt_project_path = self._get_dbt_project_stage_path(stage_name)
-            query = f"""CREATE OR REPLACE DBT PROJECT {name}
+            query = f"""{'CREATE OR REPLACE' if force is True else 'CREATE'} DBT PROJECT {name}
 FROM {stage_name} MAIN_FILE='{staged_dbt_project_path}'
 DBT_VERSION='{dbt_version}' DBT_ADAPTER_VERSION='{dbt_adapter_version}'"""
             return self.execute_query(query)

tests/dbt/test_dbt_commands.py

@@ -73,7 +73,7 @@ def test_deploys_project_from_source(
         assert result.exit_code == 0, result.output
         assert (
             mock_connect.mocked_ctx.get_query()
-            == """CREATE OR REPLACE DBT PROJECT TEST_PIPELINE
+            == """CREATE DBT PROJECT TEST_PIPELINE
 FROM @MockDatabase.MockSchema.dbt_TEST_PIPELINE_stage MAIN_FILE='@MockDatabase.MockSchema.dbt_TEST_PIPELINE_stage/dbt_project.yml'
 DBT_VERSION='1.2.3' DBT_ADAPTER_VERSION='3.4.5'"""
         )
@@ -102,11 +102,32 @@ def test_dbt_version_from_option_has_precedence_over_file(
         assert result.exit_code == 0, result.output
         assert (
             mock_connect.mocked_ctx.get_query()
-            == """CREATE OR REPLACE DBT PROJECT TEST_PIPELINE
+            == """CREATE DBT PROJECT TEST_PIPELINE
 FROM @MockDatabase.MockSchema.dbt_TEST_PIPELINE_stage MAIN_FILE='@MockDatabase.MockSchema.dbt_TEST_PIPELINE_stage/dbt_project.yml'
 DBT_VERSION='2.3.4' DBT_ADAPTER_VERSION='3.4.5'"""
         )
 
+    @mock.patch("snowflake.cli._plugins.dbt.manager.StageManager.put_recursive")
+    @mock.patch("snowflake.cli._plugins.dbt.manager.StageManager.create")
+    def test_force_flag_uses_create_or_replace(
+        self, _mock_create, _mock_put_recursive, mock_connect, runner, dbt_project_path
+    ):
+        result = runner.invoke(
+            [
+                "dbt",
+                "deploy",
+                "TEST_PIPELINE",
+                f"--source={dbt_project_path}",
+                "--force",
+                "--dbt-adapter-version=3.4.5",
+            ]
+        )
+
+        assert result.exit_code == 0, result.output
+        assert mock_connect.mocked_ctx.get_query().startswith(
+            "CREATE OR REPLACE DBT PROJECT"
+        )
+
     def test_raises_when_dbt_project_is_not_available(
         self, dbt_project_path, mock_connect, runner
     ):