"""Unit tests for Snowflake SQL executor."""

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

import pytest
from snowflake import connector

from feed_ingestion.flows.amazon_digital_services.snowflake_executor \
    import AmazonDSTheOrchard
from tests.conftest import SubstringMatcher


@pytest.fixture
def mock_executor(sf_config_mock, monkeypatch):
    """Yield executor context."""
    connect_mock = MagicMock()
    monkeypatch.setattr(connector, 'connect', connect_mock)
    executor = AmazonDSTheOrchard(sf_config_mock)
    with patch.object(executor, 'execute', wraps=executor.execute) as \
            executor.ex_mock:
        yield executor


def test_load_staging_fact_table(mock_executor):
    """Test load_staging_fact_table method."""
    mock_executor.load_staging_fact_table('2017-01-01')

    assert mock_executor.ex_mock.call_count == 1
    mock_executor.ex_mock.assert_any_call(
        SubstringMatcher(
            containing=[
                'staging_fact_analytics_amazon_digital_services'
                '_theorchard_20170101',
                'test_db.test_schema.staging_raw_amazon_v2']),
        params={'reportdate': '2017-01-01', 'licensor': 'theorchard',
                'storeid': 187, 'feedid': 16})


def test_load_fact_error_data(mock_executor):
    """Test load_fact_error_data method."""
    mock_executor.load_fact_error_data('2017-01-01')

    assert mock_executor.ex_mock.call_count == 1
    mock_executor.ex_mock.assert_any_call(
        SubstringMatcher(
            containing=[
                'test_db.test_schema.staging_fact_analytics_'
                'amazon_digital_services_theorchard_20170101',
                'test_db.test_schema.staging_raw_amazon_v2',
                'test_db.test_schema.fact_analytics_error']),
        params={'reportdate': '2017-01-01', 'licensor': 'theorchard',
                'storeid': 187, 'feedid': 16})