X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=test.py;h=c496e11a916adb2aaeb0a371fb4695ae8b4fb5a7;hb=f320eb8aafbceafbbfca02617db4846b3571e598;hp=f7df9a861ffc46240515f5868d5a508567122313;hpb=df9e4e7f30a3505675bf61f7da19af4453647772;p=perso%2FImmae%2FProjets%2FCryptomonnaies%2FCryptoportfolio%2FTrader.git diff --git a/test.py b/test.py index f7df9a8..c496e11 100644 --- a/test.py +++ b/test.py @@ -220,6 +220,8 @@ class AmountTest(WebMockTestCase): amount4 = portfolio.Amount("ETH", 0.0) self.assertEqual(amount1, amount1 + amount4) + self.assertEqual(amount1, amount1 + 0) + def test__radd(self): amount = portfolio.Amount("XVG", "12.9") @@ -241,6 +243,13 @@ class AmountTest(WebMockTestCase): amount4 = portfolio.Amount("ETH", 0.0) self.assertEqual(amount1, amount1 - amount4) + def test__rsub(self): + amount = portfolio.Amount("ETH", "1.6") + with self.assertRaises(Exception): + 3 - amount + + self.assertEqual(portfolio.Amount("ETH", "-1.6"), -amount) + def test__mul(self): amount = portfolio.Amount("XEM", 11) @@ -724,7 +733,8 @@ class HelperTest(WebMockTestCase): portfolio.TradeStore.all = [trade1, trade2, trade3] balance1 = portfolio.Balance("BTC", { "margin_free": "0" }) balance2 = portfolio.Balance("USDT", { "margin_free": "100" }) - portfolio.BalanceStore.all = {"BTC": balance1, "USDT": balance2} + balance3 = portfolio.Balance("ETC", { "margin_free": "10" }) + portfolio.BalanceStore.all = {"BTC": balance1, "USDT": balance2, "ETC": balance3} market = mock.Mock() @@ -736,6 +746,7 @@ class HelperTest(WebMockTestCase): else: market.transfer_balance.assert_any_call("BTC", 3, "exchange", "margin") market.transfer_balance.assert_any_call("USDT", 50, "margin", "exchange") + market.transfer_balance.assert_any_call("ETC", 10, "margin", "exchange") @mock.patch.object(helper, "prepare_trades") @mock.patch.object(portfolio.TradeStore, "prepare_orders") @@ -757,11 +768,141 @@ class HelperTest(WebMockTestCase): } helper.print_orders(market) prepare_trades.assert_called_with(market, base_currency="BTC", - compute_value="average") + compute_value="average", debug=True) prepare_orders.assert_called_with(compute_value="average") print_all_with_order.assert_called() self.assertRegex(stdout_mock.getvalue(), "Balance") + @mock.patch.object(helper, "prepare_trades") + @mock.patch.object(helper, "follow_orders") + @mock.patch.object(portfolio.TradeStore, "prepare_orders") + @mock.patch.object(portfolio.TradeStore, "print_all_with_order") + @mock.patch.object(portfolio.TradeStore, "run_orders") + @mock.patch('sys.stdout', new_callable=StringIO) + def test_process_sell_needed__1_sell(self, stdout_mock, run_orders, + print_all_with_order, prepare_orders, follow_orders, + prepare_trades): + market = mock.Mock() + portfolio.BalanceStore.all = { + "BTC": portfolio.Balance("BTC", { + "total": "0.65", + "exchange_total":"0.65", + "exchange_free": "0.35", + "exchange_used": "0.30"}), + "ETH": portfolio.Balance("ETH", { + "total": 3, + "exchange_total": 3, + "exchange_free": 3, + "exchange_used": 0}), + } + helper.process_sell_needed__1_sell(market) + prepare_trades.assert_called_with(market, base_currency="BTC", + debug=False) + prepare_orders.assert_called_with(compute_value="average", + only="dispose") + print_all_with_order.assert_called() + run_orders.assert_called() + follow_orders.assert_called() + self.assertRegex(stdout_mock.getvalue(), "Balance") + + @mock.patch.object(helper, "update_trades") + @mock.patch.object(helper, "follow_orders") + @mock.patch.object(helper, "move_balances") + @mock.patch.object(portfolio.TradeStore, "prepare_orders") + @mock.patch.object(portfolio.TradeStore, "print_all_with_order") + @mock.patch.object(portfolio.TradeStore, "run_orders") + @mock.patch('sys.stdout', new_callable=StringIO) + def test_process_sell_needed__2_buy(self, stdout_mock, run_orders, + print_all_with_order, prepare_orders, move_balances, + follow_orders, update_trades): + market = mock.Mock() + portfolio.BalanceStore.all = { + "BTC": portfolio.Balance("BTC", { + "total": "0.65", + "exchange_total":"0.65", + "exchange_free": "0.35", + "exchange_used": "0.30"}), + "ETH": portfolio.Balance("ETH", { + "total": 3, + "exchange_total": 3, + "exchange_free": 3, + "exchange_used": 0}), + } + helper.process_sell_needed__2_buy(market) + update_trades.assert_called_with(market, base_currency="BTC", + debug=False, only="acquire") + prepare_orders.assert_called_with(compute_value="average", + only="acquire") + print_all_with_order.assert_called() + move_balances.assert_called_with(market, debug=False) + run_orders.assert_called() + follow_orders.assert_called() + self.assertRegex(stdout_mock.getvalue(), "Balance") + + @mock.patch.object(helper, "prepare_trades_to_sell_all") + @mock.patch.object(helper, "follow_orders") + @mock.patch.object(portfolio.TradeStore, "prepare_orders") + @mock.patch.object(portfolio.TradeStore, "print_all_with_order") + @mock.patch.object(portfolio.TradeStore, "run_orders") + @mock.patch('sys.stdout', new_callable=StringIO) + def test_process_sell_all__1_sell(self, stdout_mock, run_orders, + print_all_with_order, prepare_orders, follow_orders, + prepare_trades_to_sell_all): + market = mock.Mock() + portfolio.BalanceStore.all = { + "BTC": portfolio.Balance("BTC", { + "total": "0.65", + "exchange_total":"0.65", + "exchange_free": "0.35", + "exchange_used": "0.30"}), + "ETH": portfolio.Balance("ETH", { + "total": 3, + "exchange_total": 3, + "exchange_free": 3, + "exchange_used": 0}), + } + helper.process_sell_all__1_all_sell(market) + prepare_trades_to_sell_all.assert_called_with(market, base_currency="BTC", + debug=False) + prepare_orders.assert_called_with(compute_value="average") + print_all_with_order.assert_called() + run_orders.assert_called() + follow_orders.assert_called() + self.assertRegex(stdout_mock.getvalue(), "Balance") + + @mock.patch.object(helper, "prepare_trades") + @mock.patch.object(helper, "follow_orders") + @mock.patch.object(helper, "move_balances") + @mock.patch.object(portfolio.TradeStore, "prepare_orders") + @mock.patch.object(portfolio.TradeStore, "print_all_with_order") + @mock.patch.object(portfolio.TradeStore, "run_orders") + @mock.patch('sys.stdout', new_callable=StringIO) + def test_process_sell_all__2_all_buy(self, stdout_mock, run_orders, + print_all_with_order, prepare_orders, move_balances, + follow_orders, prepare_trades): + market = mock.Mock() + portfolio.BalanceStore.all = { + "BTC": portfolio.Balance("BTC", { + "total": "0.65", + "exchange_total":"0.65", + "exchange_free": "0.35", + "exchange_used": "0.30"}), + "ETH": portfolio.Balance("ETH", { + "total": 3, + "exchange_total": 3, + "exchange_free": 3, + "exchange_used": 0}), + } + helper.process_sell_all__2_all_buy(market) + prepare_trades.assert_called_with(market, base_currency="BTC", + debug=False) + prepare_orders.assert_called_with() + print_all_with_order.assert_called() + move_balances.assert_called_with(market, debug=False) + run_orders.assert_called() + follow_orders.assert_called() + self.assertRegex(stdout_mock.getvalue(), "Balance") + @unittest.skipUnless("unit" in limits, "Unit skipped") class TradeStoreTest(WebMockTestCase):