forked from dbt-labs/dbt-snowflake
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* set snowflake connector log level on env variable * set snowflake connector log level on env variable * set snowflake connector log level on env variable * add boto packages * add logging about stting up logging * merge paw/type-fix * add changie * fix event import * add unit tests * add unit tests
- Loading branch information
1 parent
8569cf2
commit 64f32af
Showing
3 changed files
with
39 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,6 @@ | ||
kind: Under the Hood | ||
body: allow for adding snowflake-python-collector logs to dbt output | ||
time: 2023-09-25T12:01:44.778426-07:00 | ||
custom: | ||
Author: colin-rogers-dbt | ||
Issue: "768" |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,25 @@ | ||
import os | ||
from importlib import reload | ||
from unittest.mock import Mock | ||
import dbt.adapters.snowflake.connections as connections | ||
import dbt.events | ||
|
||
|
||
def test_connections_sets_logs_in_response_to_env_var(monkeypatch): | ||
"""Test that setting the DBT_SNOWFLAKE_CONNECTOR_DEBUG_LOGGING environment variable happens on import""" | ||
log_mock = Mock() | ||
monkeypatch.setattr(dbt.events, "AdapterLogger", Mock(return_value=log_mock)) | ||
monkeypatch.setattr(os, "environ", {"DBT_SNOWFLAKE_CONNECTOR_DEBUG_LOGGING": "true"}) | ||
reload(connections) | ||
|
||
assert log_mock.debug.call_count == 3 | ||
assert log_mock.set_adapter_dependency_log_level.call_count == 3 | ||
|
||
|
||
def test_connections_does_not_set_logs_in_response_to_env_var(monkeypatch): | ||
log_mock = Mock() | ||
monkeypatch.setattr(dbt.events, "AdapterLogger", Mock(return_value=log_mock)) | ||
reload(connections) | ||
|
||
assert log_mock.debug.call_count == 0 | ||
assert log_mock.set_adapter_dependency_log_level.call_count == 0 |