"""Test Setup."""

from decimal import Decimal

from dotenv import load_dotenv
import pytest

load_dotenv()


@pytest.fixture
def mock_event():
    """Mock an event."""
    return {
        'abacus_event_id': 1,
        'target_type': 'accounting_run',
        'target_id': '1337',
        'event_name': 'commit-royalties',
        'timestamp': '1997-07-16T19:20:30.45+01:00',
        'user_type': 'royalties',
        'user_id': '1'
    }


@pytest.fixture
def mock_sales():
    """Mock results from snowflake query."""
    return [
        {
            'PAYEE_ID': 1,
            'CONTRACT_ID': 1,
            'CURRENCY_CODE': 'USD',
            'AMOUNT': Decimal('44.444444444444444444')
        },
        {
            'PAYEE_ID': 1,
            'CONTRACT_ID': 2,
            'CURRENCY_CODE': 'USD',
            'AMOUNT': Decimal('0.000003300000000000')
        },
        {
            'PAYEE_ID': 1,
            'CONTRACT_ID': 3,
            'CURRENCY_CODE': 'USD',
            'AMOUNT': Decimal('111111.111111111111110000')
        },
        {
            'PAYEE_ID': 1,
            'CONTRACT_ID': 4,
            'CURRENCY_CODE': 'USD',
            'AMOUNT': Decimal('111111.139000000000000000')
        }
    ]


@pytest.fixture
def mock_payee_ledger_entry():
    """Mock payee sales ledger entry."""
    return {
        'payee_id': 1,
        'abacus_event_id': 1,
        'ledger_type': 'credit',
        'currency_code': 'USD',
        'currency_amount': Decimal('191.02'),
        'previous_balance': Decimal('0'),
        'current_balance': Decimal('191.02'),
        'note': None
    }


@pytest.fixture
def mock_results():
    """Mock results."""
    return [
        {
            'currency_code': 'USD',
            'abacus_event_id': 1,
            'ledger_type': 'credit',
            'model_type': 'deposit',
            'remaining_amount': Decimal('0.004444444444444444'),
            'rounded_amount': Decimal('44.44'),
            'contract_id': 1,
            'payee_id': 1
        },
        {
            'currency_code': 'USD',
            'abacus_event_id': 1,
            'ledger_type': 'credit',
            'model_type': 'deposit',
            'remaining_amount': Decimal('0.000003300000000000'),
            'rounded_amount': Decimal('0.00'),
            'contract_id': 2,
            'payee_id': 1
        },
        {
            'currency_code': 'USD',
            'abacus_event_id': 1,
            'ledger_type': 'credit',
            'model_type': 'deposit',
            'remaining_amount': Decimal('0.001111111111110000'),
            'rounded_amount': Decimal('111111.11'),
            'contract_id': 3,
            'payee_id': 1
        },
        {
            'currency_code': 'USD',
            'abacus_event_id': 1,
            'ledger_type': 'credit',
            'model_type': 'deposit',
            'remaining_amount': Decimal('-0.001000000000000000'),
            'rounded_amount': Decimal('111111.14'),
            'contract_id': 4,
            'payee_id': 1
        },
        {
            'currency_amount': Decimal('222266.69'),
            'currency_code': 'USD',
            'abacus_event_id': 1,
            'ledger_type': 'credit',
            'model_type': 'payee',
            'payee_id': 1
        },
        {
            'currency_amount': Decimal('-222266.694558855556'),
            'currency_code': 'USD',
            'abacus_event_id': 1,
            'ledger_type': 'debit',
            'model_type': 'general'
        }
    ]


@pytest.fixture
def mock_payee_list():
    """Mock list of distinct payee ids."""
    return [
        [
            {'PAYEE_ID': 1},
            {'PAYEE_ID': 2},
            {'PAYEE_ID': 4},
        ]
    ]