"""Test for Snowflake Connector."""

from unittest.mock import MagicMock
from unittest.mock import patch

import pytest

from src.connectors.snowflake import get_snowflake_connector


@patch('src.connectors.snowflake.snowflake.connector')
@patch('src.connectors.snowflake.config')
def test_valid_key(config_mock, snowflake_mock, valid_key):
    """Test handler for valid_message."""
    mock_connector = MagicMock()
    snowflake_mock.connect.return_value = mock_connector
    config_mock.SNOWFLAKE_PRIVATE_KEY = valid_key
    config_mock.SNOWFLAKE_PRIVATE_KEY_PASS = 'unittest'
    config_mock.SNOWFLAKE_ACCOUNT = 'test'

    result = get_snowflake_connector()
    assert result == mock_connector


@patch('src.connectors.snowflake.snowflake.connector')
@patch('src.connectors.snowflake.config')
def test_invalid_key(config_mock, snowflake_mock, valid_key):
    """Test handler for invalid key."""
    mock_connector = MagicMock()
    snowflake_mock.connect.return_value = mock_connector
    config_mock.SNOWFLAKE_PRIVATE_KEY = 'dummy'
    config_mock.SNOWFLAKE_PRIVATE_KEY_PASS = 'unittest'
    config_mock.SNOWFLAKE_ACCOUNT = 'test'
    with pytest.raises(ValueError, match=r'Could not deserialize key data'):
        get_snowflake_connector()