diff --git a/backend/models.py b/backend/models.py index f6b3688..4184720 100644 --- a/backend/models.py +++ b/backend/models.py @@ -346,6 +346,11 @@ class Exchange(Base): direction: Mapped[TradeDirection] = mapped_column( Enum(TradeDirection), nullable=False ) + bitcoin_transfer_method: Mapped[BitcoinTransferMethod] = mapped_column( + Enum(BitcoinTransferMethod), + nullable=False, + default=BitcoinTransferMethod.ONCHAIN, + ) eur_amount: Mapped[int] = mapped_column(Integer, nullable=False) # EUR cents sats_amount: Mapped[int] = mapped_column(Integer, nullable=False) # Satoshis diff --git a/backend/tests/test_exchange.py b/backend/tests/test_exchange.py index e1095de..c7ec778 100644 --- a/backend/tests/test_exchange.py +++ b/backend/tests/test_exchange.py @@ -9,7 +9,13 @@ from unittest.mock import patch import pytest -from models import Exchange, ExchangeStatus, PriceHistory, TradeDirection +from models import ( + BitcoinTransferMethod, + Exchange, + ExchangeStatus, + PriceHistory, + TradeDirection, +) from price_fetcher import PAIR_BTC_EUR, SOURCE_BITFINEX @@ -621,6 +627,7 @@ class TestCancelTrade: slot_start=past_time, slot_end=past_time + timedelta(minutes=15), direction=TradeDirection.BUY, + bitcoin_transfer_method=BitcoinTransferMethod.ONCHAIN, eur_amount=10000, sats_amount=500000, market_price_eur=20000.0, @@ -707,6 +714,7 @@ class TestAdminPastTrades: slot_start=past_time, slot_end=past_time + timedelta(minutes=15), direction=TradeDirection.BUY, + bitcoin_transfer_method=BitcoinTransferMethod.ONCHAIN, eur_amount=10000, sats_amount=500000, market_price_eur=20000.0, @@ -742,6 +750,7 @@ class TestAdminCompleteTrade: slot_start=past_time, slot_end=past_time + timedelta(minutes=15), direction=TradeDirection.BUY, + bitcoin_transfer_method=BitcoinTransferMethod.ONCHAIN, eur_amount=10000, sats_amount=500000, market_price_eur=20000.0, @@ -803,6 +812,7 @@ class TestAdminCompleteTrade: slot_start=past_time, slot_end=past_time + timedelta(minutes=15), direction=TradeDirection.BUY, + bitcoin_transfer_method=BitcoinTransferMethod.ONCHAIN, eur_amount=10000, sats_amount=500000, market_price_eur=20000.0, @@ -834,6 +844,7 @@ class TestAdminCompleteTrade: slot_start=past_time, slot_end=past_time + timedelta(minutes=15), direction=TradeDirection.BUY, + bitcoin_transfer_method=BitcoinTransferMethod.ONCHAIN, eur_amount=10000, sats_amount=500000, market_price_eur=20000.0, @@ -869,6 +880,7 @@ class TestAdminNoShowTrade: slot_start=past_time, slot_end=past_time + timedelta(minutes=15), direction=TradeDirection.BUY, + bitcoin_transfer_method=BitcoinTransferMethod.ONCHAIN, eur_amount=10000, sats_amount=500000, market_price_eur=20000.0,