"""Unit tests for Ledger Reserve Taken schema.""" from ledger.schemas.ledger_reserve_taken import ( LedgerReserveTakenDetailSchema, LedgerReserveTakenWithAccountContractSchema, ) from tests.utils.factories import LedgerReserveTakenFactory def test_ledger_reserve_taken_detail_schema(mock_event_fixtures): """Test Ledger Taken Reserve detail schema.""" ledger_reserve_taken = LedgerReserveTakenFactory.create() result = LedgerReserveTakenDetailSchema().dump(ledger_reserve_taken) assert result == { 'ledger_reserve_taken_id': 1, 'abacus_event_id': 1, 'accounting_run_id': 1, 'contract_reserve_id': 1, 'statement_period_id': 1, 'currency_code': 'USD', 'gross_sales': '150.00', 'gross_returns': '-50.00', 'net_revenue': '104.00', 'reserve_amount': '-60.00', 'net_revenue_after_reserve': '44.00', } def test_ledger_reserve_taken_by_period_id_schema(): """Test ledger_reserve_taken_by_period_id schema.""" mock_data = { 'ledger_reserve_taken_id': 1, 'abacus_event_id': 1, 'accounting_run_id': 1, 'contract_reserve_id': 1, 'statement_period_id': 1, 'currency_code': 'USD', 'gross_sales': '150.00', 'gross_returns': '-50.00', 'net_revenue': '104.00', 'reserve_amount': '-60.00', 'net_revenue_after_reserve': '44.00', 'account_id': 1, 'contract_id': 1, } result = LedgerReserveTakenWithAccountContractSchema().dump(mock_data) assert result == mock_data