feat: [SNOW-1890085] dbt deploy: provide main file param

Author: sfc-gh-jwilkowski

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

@@ -53,11 +53,12 @@ def deploy(
         force: bool,
     ) -> SnowflakeCursor:
         # TODO: what to do with force?
-        if not path.joinpath("dbt_project.yml").exists():
+        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.")
 
         if dbt_version is None:
-            with path.joinpath("dbt_project.yml").open() as fd:
+            with dbt_project_path.open() as fd:
                 dbt_project_config = yaml.safe_load(fd)
                 try:
                     dbt_version = dbt_project_config["version"]
@@ -77,11 +78,18 @@ def deploy(
             cli_console.step(f"Copied {len(results)} files")
 
         with cli_console.phase("Creating DBT project"):
-            query = f"CREATE OR REPLACE DBT PROJECT {name} FROM {stage_name} DBT_VERSION='{dbt_version}' DBT_ADAPTER_VERSION='{dbt_adapter_version}'"
+            staged_dbt_project_path = self._get_dbt_project_stage_path(stage_name)
+            query = f"""CREATE OR REPLACE 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)
 
     def execute(self, dbt_command: str, name: str, *dbt_cli_args):
         query = f"EXECUTE DBT PROJECT {name} {dbt_command}"
         if dbt_cli_args:
             query += " " + " ".join([arg for arg in dbt_cli_args])
         return self.execute_query(query)
+
+    @staticmethod
+    def _get_dbt_project_stage_path(stage_name):
+        return "/".join([stage_name, "dbt_project.yml"])

tests/dbt/test_dbt_commands.py

@@ -73,7 +73,9 @@ 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 FROM @MockDatabase.MockSchema.dbt_TEST_PIPELINE_stage DBT_VERSION='1.2.3' DBT_ADAPTER_VERSION='3.4.5'"
+            == """CREATE OR REPLACE 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'"""
         )
         stage_fqn = FQN.from_string(f"dbt_TEST_PIPELINE_stage").using_context()
         mock_create.assert_called_once_with(stage_fqn, temporary=True)
@@ -100,7 +102,9 @@ 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 FROM @MockDatabase.MockSchema.dbt_TEST_PIPELINE_stage DBT_VERSION='2.3.4' DBT_ADAPTER_VERSION='3.4.5'"
+            == """CREATE OR REPLACE 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'"""
         )
 
     def test_raises_when_dbt_project_is_not_available(