diff options
| -rw-r--r-- | test_ledger_process.py | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/test_ledger_process.py b/test_ledger_process.py index 3fe54dc..cf5e64c 100644 --- a/test_ledger_process.py +++ b/test_ledger_process.py @@ -66,6 +66,39 @@ class TestLedgerProcess(unittest.TestCase): with self.assertRaises(TradeNotFound): self.lp.process_ledger([deposit]) + def test_withdraw_before_trade(self): + """Test withdrawing before the trade occurred.""" + + withdrawal = LedgerAction( + type="withdrawal", + asset="BTC", + # already includes removed fee + amount=Decimal("-0.098"), + fee=Decimal("0.001"), + timestamp="2024-04-17 12:00:00", + refid="67890", + ) + + eur_trade = LedgerAction( + type="trade", + asset="EUR", + amount=Decimal("-500.00"), + fee=Decimal("2.00"), + timestamp="2025-04-17 10:00:00", + refid="12345", + ) + crypto_trade = LedgerAction( + type="trade", + asset="BTC", + amount=Decimal("0.1"), + fee=Decimal("0.001"), + timestamp="2025-04-17 10:00:00", + refid="12345", + ) + + with self.assertRaises(TradeNotFound): + self.lp.process_ledger([eur_trade, withdrawal, crypto_trade]) + def test_withdraw_afterbalance(self): """Test withdrawing and subsequently checking balance.""" @@ -158,5 +191,6 @@ class TestLedgerProcess(unittest.TestCase): self.lp.external_wallet["BTC"].get_remaining_amount(), Decimal("0.0") ) + if __name__ == "__main__": unittest.main() |