"""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': 'payment_group_payment', 'target_id': '1', 'event_name': 'payments_posted', 'timestamp': '1997-07-16T19:20:30.45+01:00', 'user_type': 'royalties', 'user_id': '1' } @pytest.fixture def mock_payment_group_payment_payee(): """Mock results from payment group payees.""" return [ { 'note': 'test', 'current_balance': '667686.00', 'payment_group_payment_id': 2, 'percent_difference': '45%', 'payee_id': 123, 'last_payment': '0.00', 'currency_name': 'Canadian Dollar', 'payment_difference': 'null', 'payment_group_payment_payee_id': 17, 'currency_code': 'CAD', 'payee_name': '(SME Sweden) GMG Sweden AB' }, { 'note': 'test', 'current_balance': '789865.00', 'payment_group_payment_id': 2, 'percent_difference': '46%', 'payee_id': 124, 'last_payment': '0.00', 'currency_name': 'Pound Sterling', 'payment_difference': '789.00', 'payment_group_payment_payee_id': 18, 'currency_code': 'GBP', 'payee_name': 'Lifestyle Music Pty Ltd' } ] @pytest.fixture def mock_results(): """Mock results.""" return [ { 'currency_code': 'CAD', 'abacus_event_id': 1, 'ledger_type': 'credit', 'model_type': 'deposit', 'current_balance': Decimal('667686.00'), 'payee_id': 1 }, { 'currency_code': 'GBP', 'abacus_event_id': 1, 'ledger_type': 'credit', 'model_type': 'deposit', 'current_balance': Decimal('789865.00'), 'payee_id': 1 } ] @pytest.fixture def mock_payee_list(): """Mock list of distinct payee ids.""" return [ [ {'payee_id': 1}, {'payee_id': 2}, {'payee_id': 4}, ] ]