"""Tests for db_utils.py."""
from integration_scripts.connectors import snowflake as snowflake_connector
from integration_scripts.db_utils import run_query


def test_run_query(mocker):
    """Test run_query() function."""
    mock_session = mocker.patch.object(snowflake_connector, '_db_session')
    mock_sql = 'SELECT 1 LIMIT 1;'

    run_query(session=mock_session, sql=mock_sql)

    assert mock_session.execute.call_count == 1