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

from integration_scripts import db_utils


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

    db_utils.run_query(session=mock_session, sql=mock_sql)

    assert mock_session.execute.call_count == 1


def test_get_single_snowflake_field(mocker):
    """Test get_single_snowflake_field() function."""
    return_value = [{'a': None, 'b': None, 'c': None}]

    mock_call = mocker.patch.object(
        db_utils.snowflake_utils, 'run_query', return_value=return_value)

    result = db_utils.get_single_snowflake_field(
        field='test_field', table='test_table')

    assert result == [{'a': 'a', 'b': 'b', 'c': 'c'}]
    assert mock_call.call_count == 1