"""Tests for datawarehouse module."""

from unittest.mock import Mock
from unittest.mock import patch

from garcon_contrib.snowflake import garcon_snowflake as snowflake

from flows import config
from flows import datawarehouse


@patch('flows.datawarehouse.snowflake.execute_with_py_conn')
def test_execute(snowflake_mock):
    """Test execute function."""
    sql = Mock()
    sf_config = {
        'user': 'user',
        'password': 'password',
        'account': 'account',
        'role': 'role',
        'warehouse': 'warehouse',
        'db': 'db',
        'schema': 'schema'
    }
    with patch.dict(config.SNOWFLAKE_CREDENTIALS, sf_config):
        datawarehouse.execute(sql)

    snowflake_mock.assert_called_once_with(
        sql, snowflake.FetchEnum.ALL,
        sf_config=config.SNOWFLAKE_CREDENTIALS,
        result_key='results')