"""Fixtures for the adjustments validation unit tests.""" import pytest from abacus_common_logic.adjustments_validation.types_definitions import ( Adjustment, SnowflakeConfig, ) @pytest.fixture def sf_config_mock() -> SnowflakeConfig: """Return some Snowflake connection params.""" return { 'account': 'test_account', 'role': 'test_role', 'warehouse': 'test_wh', 'db': 'test_db', 'schema': 'test_schema', 'user': 'test_user', 'private_key': 'some_private_key', } @pytest.fixture def adjustments_mock() -> list[Adjustment]: """Return some adjustments to validate.""" return [ { 'account_id': '1', 'contract_id': '11', 'upc': '111122223333', 'amount': '1.23', 'currency': 'USD', 'activity_month': '1', 'activity_year': '2025', 'statement_month': '2', 'statement_year': '2025', 'adjustment_type': 'Physical Storage', 'client_facing_comments': 'Comments', 'distribution_type': 'Physical', 'apply_to_flowthrough_payment': 'N', }, { 'account_id': '2', 'contract_id': '22', 'amount': '4.56', 'currency': 'GBP', 'activity_month': '1', 'activity_year': '2025', 'statement_month': '2', 'statement_year': '2025', 'adjustment_type': 'Settlement', 'client_facing_comments': 'Comments', }, { 'account_id': None, 'contract_id': None, 'amount': None, 'currency': None, 'activity_month': None, 'activity_year': None, 'statement_month': None, 'statement_year': None, 'adjustment_type': None, 'client_facing_comments': None, }, { 'account_id': 'Test', 'contract_id': 'Test', 'amount': 'Test', 'currency': 'Test', 'activity_month': 'Test', 'activity_year': 'Test', 'statement_month': 'Test', 'statement_year': 'Test', 'adjustment_type': 'Test', 'client_facing_comments': 'Test', }, ]