"""Conftest.

This file gets picked up when running py.test tests:
http://pytest.org/latest/writing_plugins.html#conftest
"""

import pytest
from snowflake import connector

from dim_refresh_etl.conf import config


@pytest.fixture(autouse=True)
def mock_snowflake_connect(mocker):
    """Mock Snowflake connect method."""
    return mocker.patch.object(connector, 'connect')


@pytest.fixture
def sf_config_mock(mocker):
    """Mock Snowflake default config."""
    config_mock = {
        'account': 'test_acc',
        'db': 'test_db',
        'host': 'test_host',
        'password': 'test_pass',
        'port': 10,
        'role': 'test_role',
        'schema': 'test_schema',
        'user': 'test_user',
        'warehouse': 'test_wh'
    }
    return mocker.patch.object(config, 'SF_CONFIG', config_mock)


@pytest.fixture
def aws_config():
    """Mock AWS default config."""
    return {
        'access_key': 'test_access_key',
        'access_secret': 'test_access_secret'}