@property
def pending(self):
- return list(filter(lambda t: not t.is_fullfiled, self.all))
+ return list(filter(lambda t: t.pending, self.all))
def compute_trades(self, values_in_base, new_repartition, only=None):
computed_trades = []
orders.append(trade.prepare_order(compute_value=compute_value))
self.market.report.log_orders(orders, only, compute_value)
+ def close_trades(self):
+ for trade in self.all:
+ trade.close()
+
def print_all_with_order(self, ind=""):
for trade in self.all:
trade.print_with_order(ind=ind)
trade_mock2.prepare_order.return_value = 2
trade_mock3.prepare_order.return_value = 3
- trade_mock1.is_fullfiled = False
- trade_mock2.is_fullfiled = False
- trade_mock3.is_fullfiled = True
+ trade_mock1.pending = True
+ trade_mock2.pending = True
+ trade_mock3.pending = False
trade_store.all.append(trade_mock1)
trade_store.all.append(trade_mock2)
order_mock2.get_status.assert_called()
order_mock3.get_status.assert_called()
+ def test_close_trades(self):
+ trade_mock1 = mock.Mock()
+ trade_mock2 = mock.Mock()
+ trade_mock3 = mock.Mock()
+
+ trade_store = market.TradeStore(self.m)
+
+ trade_store.all.append(trade_mock1)
+ trade_store.all.append(trade_mock2)
+ trade_store.all.append(trade_mock3)
+
+ trade_store.close_trades()
+
+ trade_mock1.close.assert_called_once_with()
+ trade_mock2.close.assert_called_once_with()
+ trade_mock3.close.assert_called_once_with()
+
def test_pending(self):
trade_mock1 = mock.Mock()
- trade_mock1.is_fullfiled = False
+ trade_mock1.pending = True
trade_mock2 = mock.Mock()
- trade_mock2.is_fullfiled = False
+ trade_mock2.pending = True
trade_mock3 = mock.Mock()
- trade_mock3.is_fullfiled = True
+ trade_mock3.pending = False
trade_store = market.TradeStore(self.m)
self.assertEqual("\t\tMouvement 1", str(calls[3][1][0]))
self.assertEqual("\t\tMouvement 2", str(calls[4][1][0]))
+ def test_close(self):
+ value_from = portfolio.Amount("BTC", "0.5")
+ value_from.linked_to = portfolio.Amount("ETH", "10.0")
+ value_to = portfolio.Amount("BTC", "1.0")
+ trade = portfolio.Trade(value_from, value_to, "ETH", self.m)
+
+ trade.close()
+
+ self.assertEqual(True, trade.closed)
+
+ def test_pending(self):
+ value_from = portfolio.Amount("BTC", "0.5")
+ value_from.linked_to = portfolio.Amount("ETH", "10.0")
+ value_to = portfolio.Amount("BTC", "1.0")
+ trade = portfolio.Trade(value_from, value_to, "ETH", self.m)
+
+ trade.closed = True
+ self.assertEqual(False, trade.pending)
+
+ trade.closed = False
+ self.assertEqual(True, trade.pending)
+
+ order1 = mock.Mock()
+ order1.filled_amount.return_value = portfolio.Amount("BTC", "0.5")
+ trade.orders.append(order1)
+ self.assertEqual(False, trade.pending)
+
def test__repr(self):
value_from = portfolio.Amount("BTC", "0.5")
value_from.linked_to = portfolio.Amount("ETH", "10.0")