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

from helpers.mysql_utils import execute_query


@patch('helpers.mysql_utils.mysql_connection')
def test_execute_query(mock_mysql_connection, monkeypatch):
    """Test execute_query."""
    monkeypatch.setattr('config.mysql_connection', {})
    connection = MagicMock()
    mock_mysql_connection.return_value.__enter__.return_value = connection
    execute_query('SELECT 1', args={})
    (connection.cursor.return_value.__enter__.return_value.execute.
        assert_has_calls([call('SELECT 1', args={})]))