diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test_market.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tests/test_market.py b/tests/test_market.py index 0211638..6a9ea76 100644 --- a/tests/test_market.py +++ b/tests/test_market.py | |||
@@ -381,12 +381,16 @@ class MarketTest(WebMockTestCase): | |||
381 | order_mock2.trade = trade_mock | 381 | order_mock2.trade = trade_mock |
382 | 382 | ||
383 | trade_mock.tick_actions_recreate.return_value = "tick1" | 383 | trade_mock.tick_actions_recreate.return_value = "tick1" |
384 | new_order_mock = mock.Mock() | ||
385 | trade_mock.prepare_order.return_value = new_order_mock | ||
384 | 386 | ||
385 | m.follow_orders() | 387 | m.follow_orders() |
386 | 388 | ||
387 | trade_mock.tick_actions_recreate.assert_called_once_with(2) | 389 | trade_mock.tick_actions_recreate.assert_called_once_with(2) |
388 | trade_mock.prepare_order.assert_called_once_with(compute_value="tick1") | 390 | trade_mock.prepare_order.assert_called_once_with(compute_value="tick1") |
389 | m.report.log_error.assert_called_once_with("follow_orders", message=mock.ANY) | 391 | m.report.log_error.assert_called_once_with("follow_orders", message=mock.ANY) |
392 | m.report.log_order.assert_called_with(order_mock2, 2, new_order=new_order_mock) | ||
393 | new_order_mock.run.assert_called_once_with() | ||
390 | 394 | ||
391 | @mock.patch.object(market.BalanceStore, "fetch_balances") | 395 | @mock.patch.object(market.BalanceStore, "fetch_balances") |
392 | def test_move_balance(self, fetch_balances): | 396 | def test_move_balance(self, fetch_balances): |