File tree Expand file tree Collapse file tree 1 file changed +1
-14
lines changed
src/snowflake/cli/_plugins/dbt Expand file tree Collapse file tree 1 file changed +1
-14
lines changed Original file line number Diff line number Diff line change 21
21
from click import ClickException
22
22
from snowflake .cli ._plugins .stage .manager import StageManager
23
23
from snowflake .cli .api .console import cli_console
24
- from snowflake .cli .api .feature_flags import FeatureFlag
25
24
from snowflake .cli .api .identifiers import FQN
26
25
from snowflake .cli .api .sql_execution import SqlExecutionMixin
27
26
from snowflake .connector .cursor import SnowflakeCursor
28
27
29
28
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 ):
43
30
def list (self ) -> SnowflakeCursor : # noqa: A003
44
31
query = "SHOW DBT PROJECTS"
45
32
return self .execute_query (query )
You can’t perform that action at this time.
0 commit comments