diff options
Diffstat (limited to 'tests/test_market.py')
-rw-r--r-- | tests/test_market.py | 32 |
1 files changed, 25 insertions, 7 deletions
diff --git a/tests/test_market.py b/tests/test_market.py index 46fad53..37c009b 100644 --- a/tests/test_market.py +++ b/tests/test_market.py | |||
@@ -174,7 +174,7 @@ class MarketTest(WebMockTestCase): | |||
174 | base_currency='BTC', compute_value='average', | 174 | base_currency='BTC', compute_value='average', |
175 | available_balance_only=False, liquidity='medium', | 175 | available_balance_only=False, liquidity='medium', |
176 | only=None, repartition=None) | 176 | only=None, repartition=None) |
177 | m.report.log_balances.assert_called_once_with(tag="tag") | 177 | m.report.log_balances.assert_called_once_with(tag="tag", checkpoint=None) |
178 | 178 | ||
179 | compute_trades.reset_mock() | 179 | compute_trades.reset_mock() |
180 | with self.subTest(available_balance_only=True),\ | 180 | with self.subTest(available_balance_only=True),\ |
@@ -964,7 +964,7 @@ class ProcessorTest(WebMockTestCase): | |||
964 | process_step.reset_mock() | 964 | process_step.reset_mock() |
965 | 965 | ||
966 | processor.process("sell_needed", steps=["before", "after"]) | 966 | processor.process("sell_needed", steps=["before", "after"]) |
967 | self.assertEqual(3, process_step.call_count) | 967 | self.assertEqual(4, process_step.call_count) |
968 | 968 | ||
969 | def test_method_arguments(self): | 969 | def test_method_arguments(self): |
970 | ccxt = mock.Mock(spec=market.ccxt.poloniexE) | 970 | ccxt = mock.Mock(spec=market.ccxt.poloniexE) |
@@ -1002,17 +1002,17 @@ class ProcessorTest(WebMockTestCase): | |||
1002 | processor = market.Processor(self.m) | 1002 | processor = market.Processor(self.m) |
1003 | 1003 | ||
1004 | with mock.patch.object(processor, "run_action") as run_action: | 1004 | with mock.patch.object(processor, "run_action") as run_action: |
1005 | step = processor.scenarios["sell_needed"][1] | 1005 | step = processor.scenarios["sell_needed"][2] |
1006 | 1006 | ||
1007 | processor.process_step("foo", step, {"foo":"bar"}) | 1007 | processor.process_step("foo", step, {"foo":"bar"}) |
1008 | 1008 | ||
1009 | self.m.report.log_stage.assert_has_calls([ | 1009 | self.m.report.log_stage.assert_has_calls([ |
1010 | mock.call("process_foo__1_sell_begin"), | 1010 | mock.call("process_foo__2_sell_begin"), |
1011 | mock.call("process_foo__1_sell_end"), | 1011 | mock.call("process_foo__2_sell_end"), |
1012 | ]) | 1012 | ]) |
1013 | self.m.balances.fetch_balances.assert_has_calls([ | 1013 | self.m.balances.fetch_balances.assert_has_calls([ |
1014 | mock.call(tag="process_foo__1_sell_begin", log_tickers=True), | 1014 | mock.call(tag="process_foo__2_sell_begin"), |
1015 | mock.call(tag="process_foo__1_sell_end", log_tickers=True), | 1015 | mock.call(tag="process_foo__2_sell_end"), |
1016 | ]) | 1016 | ]) |
1017 | 1017 | ||
1018 | self.assertEqual(5, run_action.call_count) | 1018 | self.assertEqual(5, run_action.call_count) |
@@ -1030,6 +1030,24 @@ class ProcessorTest(WebMockTestCase): | |||
1030 | step = processor.scenarios["sell_needed"][0] | 1030 | step = processor.scenarios["sell_needed"][0] |
1031 | 1031 | ||
1032 | processor.process_step("foo", step, {"foo":"bar"}) | 1032 | processor.process_step("foo", step, {"foo":"bar"}) |
1033 | |||
1034 | self.m.report.log_stage.assert_has_calls([ | ||
1035 | mock.call("process_foo__0_print_balances_begin"), | ||
1036 | mock.call("process_foo__0_print_balances_end"), | ||
1037 | ]) | ||
1038 | self.m.balances.fetch_balances.assert_has_calls([ | ||
1039 | mock.call(add_portfolio=True, checkpoint='end', | ||
1040 | log_tickers=True, | ||
1041 | tag='process_foo__0_print_balances_begin') | ||
1042 | ]) | ||
1043 | |||
1044 | self.assertEqual(0, run_action.call_count) | ||
1045 | |||
1046 | self.m.reset_mock() | ||
1047 | with mock.patch.object(processor, "run_action") as run_action: | ||
1048 | step = processor.scenarios["sell_needed"][1] | ||
1049 | |||
1050 | processor.process_step("foo", step, {"foo":"bar"}) | ||
1033 | self.m.balances.fetch_balances.assert_not_called() | 1051 | self.m.balances.fetch_balances.assert_not_called() |
1034 | 1052 | ||
1035 | self.m.reset_mock() | 1053 | self.m.reset_mock() |