Skip to content

Commit 1123e46

Browse files
feat: [SNOW-1966187] remove StdoutExecutionMixin
1 parent c3cb841 commit 1123e46

File tree

1 file changed

+1
-14
lines changed

1 file changed

+1
-14
lines changed

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

Lines changed: 1 addition & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -21,25 +21,12 @@
2121
from click import ClickException
2222
from snowflake.cli._plugins.stage.manager import StageManager
2323
from snowflake.cli.api.console import cli_console
24-
from snowflake.cli.api.feature_flags import FeatureFlag
2524
from snowflake.cli.api.identifiers import FQN
2625
from snowflake.cli.api.sql_execution import SqlExecutionMixin
2726
from snowflake.connector.cursor import SnowflakeCursor
2827

2928

30-
class StdoutExecutionMixin(SqlExecutionMixin):
31-
def execute_query(self, query, **kwargs):
32-
if FeatureFlag.ENABLE_DBT_POC.is_enabled():
33-
from unittest.mock import MagicMock
34-
35-
cli_console.message(f"Sending query: {query}")
36-
mock_cursor = MagicMock()
37-
mock_cursor.description = []
38-
return mock_cursor
39-
return super().execute_query(query, **kwargs)
40-
41-
42-
class DBTManager(StdoutExecutionMixin):
29+
class DBTManager(SqlExecutionMixin):
4330
def list(self) -> SnowflakeCursor: # noqa: A003
4431
query = "SHOW DBT PROJECTS"
4532
return self.execute_query(query)

0 commit comments

Comments
 (0)