"""Unit tests for import_processed_dig_sales.

TODO
    * execute_a_query test
    * select_an_error test

"""

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

import pytest

from import_processed_dig_sales import snowflakeSalesImporter


@patch('import_processed_dig_sales.snowflakeSalesImporter.select_an_error')
@patch('import_processed_dig_sales.snowflakeSalesImporter._connect')
@patch('import_processed_dig_sales.snowflakeSalesImporter.execute_a_query')
def test_create(mock_query, mock_conn, mock_error):
    """Test s3_import."""
    mock_conn.return_value = MagicMock()
    mock_query.return_value = True
    mock_error.return_value = True
    importer = snowflakeSalesImporter()
    importer.s3_import()
    assert mock_query.call_count == 21
    assert mock_error.call_count == 2
    mock_conn.assert_called_once()


@pytest.mark.skip('todo')
def test_execute_a_query():
    """Test execute_a_query."""
    assert True


@pytest.mark.skip('todo')
def test_select_an_error():
    """Test select_an_error."""
    assert True