diff options
Diffstat (limited to 'test.py')
-rw-r--r-- | test.py | 10 |
1 files changed, 10 insertions, 0 deletions
@@ -1075,6 +1075,16 @@ class HelperTest(WebMockTestCase): | |||
1075 | follow_orders.assert_called() | 1075 | follow_orders.assert_called() |
1076 | log_stage.assert_called_with("process_sell_all__2_all_buy_end") | 1076 | log_stage.assert_called_with("process_sell_all__2_all_buy_end") |
1077 | 1077 | ||
1078 | def test_reset_all(self): | ||
1079 | portfolio.BalanceStore.all = { "foo": "bar" } | ||
1080 | portfolio.ReportStore.logs.append("hey") | ||
1081 | portfolio.TradeStore.all.append("bouh") | ||
1082 | |||
1083 | helper.reset_all() | ||
1084 | |||
1085 | self.assertEqual(0, len(portfolio.BalanceStore.all)) | ||
1086 | self.assertEqual(0, len(portfolio.ReportStore.logs)) | ||
1087 | self.assertEqual(0, len(portfolio.TradeStore.all)) | ||
1078 | 1088 | ||
1079 | @unittest.skipUnless("unit" in limits, "Unit skipped") | 1089 | @unittest.skipUnless("unit" in limits, "Unit skipped") |
1080 | class TradeStoreTest(WebMockTestCase): | 1090 | class TradeStoreTest(WebMockTestCase): |