"""Tests for Snowflake connector."""

from unittest import mock

import pytest
from snowflake.connector import errors

from accounting.connectors import snowflake


@mock.patch('accounting.connectors.snowflake.snowflake.connector.connect')
def test_get_connection(connect_mock):
    """Test get Snowflake connection succeeds."""
    connection = snowflake.get_snowflake_connection()
    assert connection
    connect_mock.assert_called_with(
        account='test_snowflake_account',
        user='test_snowflake_user',
        password='test_snowflake_password',
        database='test_snowflake_db',
        schema='test_snowflake_schema',
        role='test_snowflake_role',
        warehouse='test_snowflake_warehouse'
    )


@mock.patch(
    'accounting.connectors.snowflake.snowflake.connector.connect',
    side_effect=errors.Error)
def test_get_connection_failed(connect_mock):
    """Test get Snowflake connection succeeds."""
    with pytest.raises(errors.Error):
        snowflake.get_snowflake_connection()