"""Lambda test snowflake_utils."""
from unittest.mock import MagicMock
from unittest.mock import call
from unittest.mock import patch

from helpers.snowflake_utils import execute_snowflake_query


@patch('helpers.snowflake_utils.snowflake_connect')
def test_execute_snowflake_query(mock_snowflake):
    """Test execute_snowflake_query."""
    connection = MagicMock()
    mock_snowflake.return_value.__enter__.return_value = connection
    execute_snowflake_query('SELECT 1', params={})
    (connection.cursor.return_value.__enter__.return_value.execute.
        assert_has_calls([call('SELECT 1', {})]))


@patch('helpers.snowflake_utils.snowflake_connect')
def test_execute_snowflake_query_with_params(mock_snowflake):
    """Test execute_snowflake_query with params."""
    query = 'SELECT * from table WHERE vendor = %(vendor_id)s'
    connection = MagicMock()
    mock_snowflake.return_value.__enter__.return_value = connection
    execute_snowflake_query(query, params={'vendor': 123})
    (connection.cursor.return_value.__enter__.return_value.execute.
        assert_has_calls([call(query, {'vendor': 123})]))