"""conftest."""

from pytest import fixture


@fixture
def mock_adjustments_types():
    """Mock adjustment types."""
    return ['Label Earnings', 'Legal Fees', 'Admin Fees']


@fixture
def mock_account_ids():
    """Mock account ids."""
    return ['68952', '72346', '74211']


@fixture
def mock_account_contracts():
    """Mock account contracts."""
    return {'68952': ['533565'], '72346': ['533993'], '74211': ['535978']}


@fixture
def mock_account_payment_entities():
    """Mock account payment entities."""
    return {'68952': '3', '72346': '3', '74211': '1'}


@fixture
def mock_statement_periods():
    """Mock statement periods."""
    return [
        {
            'STATEMENT_YEAR': 2022,
            'STATEMENT_MONTH': 6,
            'STATEMENT_PERIOD_STATUS': 'closed',
        },
        {
            'STATEMENT_YEAR': 2022,
            'STATEMENT_MONTH': 7,
            'STATEMENT_PERIOD_STATUS': 'current',
        },
        {
            'STATEMENT_YEAR': 2022,
            'STATEMENT_MONTH': 8,
            'STATEMENT_PERIOD_STATUS': 'open',
        },
    ]


@fixture
def mock_upcs():
    """Mock UPCs."""
    return [
        {'DISPLAY_UPC': '0028948106431', 'UPCS': '20000000137657'},
        {'DISPLAY_UPC': '0028948107698', 'UPCS': '20000000137678'},
        {'DISPLAY_UPC': '0028948107902', 'UPCS': '20000000137477'},
    ]


@fixture
def mock_contract_product_terms():
    """Mock contract product terms."""
    return [
        {
            'CONTRACT_ID': '533565',
            'TERM_TYPE': 'product',
            'ATTACHMENTS': str(['196626383143', '192562944424']),
        },
        {
            'CONTRACT_ID': '533993',
            'TERM_TYPE': 'product',
            'ATTACHMENTS': str(['20000000028901']),
        },
        {
            'CONTRACT_ID': '535978',
            'TERM_TYPE': 'product',
            'ATTACHMENTS': str(['886445283891', '886444667999', '886444370851']),
        },
        {
            'CONTRACT_ID': '535979',
            'TERM_TYPE': 'product',
            'ATTACHMENTS': str(['20000000137477']),
        },
        {
            'CONTRACT_ID': '535980',
            'TERM_TYPE': 'product',
            'ATTACHMENTS': str(['20000000170953']),
        },
    ]


@fixture
def mock_contract_label_terms():
    """Mock contract label terms."""
    return [
        {
            'CONTRACT_ID': '533565',
            'TERM_TYPE': 'label',
            'ATTACHMENTS': str(['68952']),
        },
        {
            'CONTRACT_ID': '533565',
            'TERM_TYPE': 'label',
            'ATTACHMENTS': str(['68953']),
        },
        {
            'CONTRACT_ID': '533993',
            'TERM_TYPE': 'label',
            'ATTACHMENTS': str(['72346']),
        },
        {
            'CONTRACT_ID': '535978',
            'TERM_TYPE': 'label',
            'ATTACHMENTS': str(['74211']),
        },
        {
            'CONTRACT_ID': '535980',
            'TERM_TYPE': 'label',
            'ATTACHMENTS': str(['72346']),
        },
    ]


@fixture
def mock_upc_labels():
    """Mock UPC labels."""
    return [
        {'VENDOR_ID': '74211', 'UPC': '20000000137103', 'DISPLAY_UPC': '0028948109173'}
    ]


@fixture
def mock_statement_period():
    """Mock statement period."""
    return [{'statement_period_id': 1}]


@fixture
def mock_payment_entity_close_balance_statuses():
    """Mock closing balance statuses for payment entities in statement period."""
    return [
        {
            'statement_period_payment_entity_id': 1,
            'reference_payment_entity_id': 1,
            'statement_period_id': 1,
            'abacus_state_id': 1000,
            'action_name': 'close_balance',
            'action_status': 'complete',
        },
        {
            'statement_period_payment_entity_id': 1,
            'reference_payment_entity_id': 2,
            'statement_period_id': 1,
            'abacus_state_id': 1001,
            'action_name': 'close_balance',
            'action_status': 'init',
        },
    ]


@fixture
def mock_adjustment():
    """Mock adjustment row data."""
    return {
        'account_id': '68952',
        'contract_id': '533565',
        'upc': '196626383143',
        'amount': '-50.8979990',
        'currency': 'USD',
        'activity_year': '2022',
        'activity_month': '8',
        'statement_year': '2022',
        'statement_month': '7',
        'adjustment_type': 'Label Earnings',
        'client_facing_comments': 'comment',
        'distribution_type': 'digital',
        'internal_note': None,
    }


@fixture
def mock_adjustment_special_characters():
    """Mock adjustment row data with special characters."""
    return {
        'account_id': '68952\xa0',
        'contract_id': '533565\u200b',
        'upc': '196626383143   ',
        'amount': '-50.8979990',
        'currency': 'USD',
        'activity_year': '2022',
        'activity_month': '8',
        'statement_year': '2022',
        'statement_month': '7',
        'adjustment_type': 'Label Earnings',
        'client_facing_comments': 'comment',
        'distribution_type': 'digital',
    }


@fixture
def mock_event():
    """Mock abacus event."""
    return {
        'abacus_event_id': 1,
        'target_type': 'statement_period_adjustment_file',
        'target_id': '1',
        'statement_period_id': 123,
        'event_name': 'adjustment_file_upload',
        'event_date': '2021-07-11T11:20:35.45+01:00',
    }


@fixture
def mock_statement_period_adjustment_file_states():
    """Fixture for statement period adjustment file upload abacus state."""
    return [
        {
            'abacus_state_id': 123,
            'action_name': 'upload_file',
            'action_status': 'error',
            'message': None,
        },
        {
            'abacus_state_id': 124,
            'action_name': 'import_file',
            'action_status': 'init',
            'message': None,
        },
    ]