From 2f0939e1353f0a95d858fb45c12bdd6d9dcb84f3 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Isma=C3=ABl=20Bouya?= Date: Mon, 12 Feb 2018 00:41:10 +0100 Subject: [PATCH] Add last tests for helper --- helper.py | 4 +- test.py | 132 +++++++++++++++++++++++++++++++++++++++++++++++++++++- 2 files changed, 133 insertions(+), 3 deletions(-) diff --git a/helper.py b/helper.py index 421e8cd..f1aeaf2 100644 --- a/helper.py +++ b/helper.py @@ -109,7 +109,7 @@ def follow_orders(verbose=True, sleep=None): print("All orders finished") def print_orders(market, base_currency="BTC"): - prepare_trades(market, base_currency=base_currency, compute_value="average") + prepare_trades(market, base_currency=base_currency, compute_value="average", debug=True) TradeStore.prepare_orders(compute_value="average") for currency, balance in BalanceStore.all.items(): print(balance) @@ -127,7 +127,7 @@ def process_sell_needed__1_sell(market, base_currency="BTC", debug=False): TradeStore.run_orders() follow_orders() -def process_sell_needed__2_sell(market, base_currency="BTC", debug=False): +def process_sell_needed__2_buy(market, base_currency="BTC", debug=False): update_trades(market, base_currency=base_currency, debug=debug, only="acquire") TradeStore.prepare_orders(compute_value="average", only="acquire") print("------------------") diff --git a/test.py b/test.py index f7df9a8..26646be 100644 --- a/test.py +++ b/test.py @@ -757,11 +757,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): -- 2.41.0