File tree Expand file tree Collapse file tree 2 files changed +15
-1
lines changed Expand file tree Collapse file tree 2 files changed +15
-1
lines changed Original file line number Diff line number Diff line change 19
19
from click import ClickException
20
20
from snowflake .cli ._plugins .stage .manager import StageManager
21
21
from snowflake .cli .api .console import cli_console
22
+ from snowflake .cli .api .feature_flags import FeatureFlag
22
23
from snowflake .cli .api .identifiers import FQN
23
24
from snowflake .cli .api .sql_execution import SqlExecutionMixin
24
25
from snowflake .connector .cursor import SnowflakeCursor
25
26
26
27
27
- class DBTManager (SqlExecutionMixin ):
28
+ class StdoutExecutionMixin (SqlExecutionMixin ):
29
+ def execute_query (self , query , ** kwargs ):
30
+ if FeatureFlag .ENABLE_DBT_POC .is_enabled ():
31
+ from unittest .mock import MagicMock
32
+
33
+ cli_console .message (f"Sending query: { query } " )
34
+ mock_cursor = MagicMock ()
35
+ mock_cursor .description = []
36
+ return mock_cursor
37
+ return super ().execute_query (query , ** kwargs )
38
+
39
+
40
+ class DBTManager (StdoutExecutionMixin ):
28
41
def list (self ) -> SnowflakeCursor : # noqa: A003
29
42
query = "SHOW DBT"
30
43
return self .execute_query (query )
Original file line number Diff line number Diff line change @@ -68,3 +68,4 @@ class FeatureFlag(FeatureFlagMixin):
68
68
)
69
69
ENABLE_SNOWPARK_GLOB_SUPPORT = BooleanFlag ("ENABLE_SNOWPARK_GLOB_SUPPORT" , False )
70
70
ENABLE_SPCS_SERVICE_EVENTS = BooleanFlag ("ENABLE_SPCS_SERVICE_EVENTS" , False )
71
+ ENABLE_DBT_POC = BooleanFlag ("ENABLE_DBT_POC" , False )
You can’t perform that action at this time.
0 commit comments