"""Tests for the Snowflake Connector."""

from unittest.mock import MagicMock, patch

from gridgen import config
from gridgen.connectors import snowflake_connector


@patch("snowflake.connector.connect")
def test_get_connection(mock_obj: MagicMock) -> None:
    """Test get_connection."""
    snowflake_connector.get_connection()
    mock_obj.assert_called_with(
        user=config.SNOWFLAKE_CONFIG["user"],
        account=config.SNOWFLAKE_CONFIG["account"],
        database=config.SNOWFLAKE_CONFIG["database"],
        schema=config.SNOWFLAKE_CONFIG["schema"],
        warehouse=config.SNOWFLAKE_CONFIG["warehouse"],
        role=config.SNOWFLAKE_CONFIG["role"],
        private_key=config.SNOWFLAKE_CONFIG["private_key"],
    )