aboutsummaryrefslogtreecommitdiff
path: root/test.py
diff options
context:
space:
mode:
Diffstat (limited to 'test.py')
-rw-r--r--test.py10
1 files changed, 10 insertions, 0 deletions
diff --git a/test.py b/test.py
index 34b2fe3..fc331c3 100644
--- a/test.py
+++ b/test.py
@@ -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")
1080class TradeStoreTest(WebMockTestCase): 1090class TradeStoreTest(WebMockTestCase):