diff options
author | Ismaël Bouya <ismael.bouya@normalesup.org> | 2018-02-12 02:08:50 +0100 |
---|---|---|
committer | Ismaël Bouya <ismael.bouya@normalesup.org> | 2018-02-12 02:08:50 +0100 |
commit | 0c79fad318711394874d94672e96db6da1ed9c52 (patch) | |
tree | bf39fa15f5fef160fbb4911f2a67a7aa9236e4cd /test.py | |
parent | 2308a1c4c0d2514dc9fc9acf235f9e5a33299f63 (diff) | |
download | Trader-0c79fad318711394874d94672e96db6da1ed9c52.tar.gz Trader-0c79fad318711394874d94672e96db6da1ed9c52.tar.zst Trader-0c79fad318711394874d94672e96db6da1ed9c52.zip |
Fix move_balance not moving currencies absent from trades
Diffstat (limited to 'test.py')
-rw-r--r-- | test.py | 4 |
1 files changed, 3 insertions, 1 deletions
@@ -724,7 +724,8 @@ class HelperTest(WebMockTestCase): | |||
724 | portfolio.TradeStore.all = [trade1, trade2, trade3] | 724 | portfolio.TradeStore.all = [trade1, trade2, trade3] |
725 | balance1 = portfolio.Balance("BTC", { "margin_free": "0" }) | 725 | balance1 = portfolio.Balance("BTC", { "margin_free": "0" }) |
726 | balance2 = portfolio.Balance("USDT", { "margin_free": "100" }) | 726 | balance2 = portfolio.Balance("USDT", { "margin_free": "100" }) |
727 | portfolio.BalanceStore.all = {"BTC": balance1, "USDT": balance2} | 727 | balance3 = portfolio.Balance("ETC", { "margin_free": "10" }) |
728 | portfolio.BalanceStore.all = {"BTC": balance1, "USDT": balance2, "ETC": balance3} | ||
728 | 729 | ||
729 | market = mock.Mock() | 730 | market = mock.Mock() |
730 | 731 | ||
@@ -736,6 +737,7 @@ class HelperTest(WebMockTestCase): | |||
736 | else: | 737 | else: |
737 | market.transfer_balance.assert_any_call("BTC", 3, "exchange", "margin") | 738 | market.transfer_balance.assert_any_call("BTC", 3, "exchange", "margin") |
738 | market.transfer_balance.assert_any_call("USDT", 50, "margin", "exchange") | 739 | market.transfer_balance.assert_any_call("USDT", 50, "margin", "exchange") |
740 | market.transfer_balance.assert_any_call("ETC", 10, "margin", "exchange") | ||
739 | 741 | ||
740 | @mock.patch.object(helper, "prepare_trades") | 742 | @mock.patch.object(helper, "prepare_trades") |
741 | @mock.patch.object(portfolio.TradeStore, "prepare_orders") | 743 | @mock.patch.object(portfolio.TradeStore, "prepare_orders") |