diff options
Diffstat (limited to 'tests/test_market.py')
-rw-r--r-- | tests/test_market.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/tests/test_market.py b/tests/test_market.py index ab3cd5e..46fad53 100644 --- a/tests/test_market.py +++ b/tests/test_market.py | |||
@@ -1032,6 +1032,15 @@ class ProcessorTest(WebMockTestCase): | |||
1032 | processor.process_step("foo", step, {"foo":"bar"}) | 1032 | processor.process_step("foo", step, {"foo":"bar"}) |
1033 | self.m.balances.fetch_balances.assert_not_called() | 1033 | self.m.balances.fetch_balances.assert_not_called() |
1034 | 1034 | ||
1035 | self.m.reset_mock() | ||
1036 | with mock.patch.object(processor, "run_action") as run_action: | ||
1037 | step = processor.scenarios["print_balances"][0] | ||
1038 | |||
1039 | processor.process_step("foo", step, {"foo":"bar"}) | ||
1040 | self.m.balances.fetch_balances.assert_called_once_with( | ||
1041 | add_portfolio=True, log_tickers=True, | ||
1042 | tag='process_foo__1_print_balances_begin') | ||
1043 | |||
1035 | def test_parse_args(self): | 1044 | def test_parse_args(self): |
1036 | processor = market.Processor(self.m) | 1045 | processor = market.Processor(self.m) |
1037 | 1046 | ||