Add main running file and fetch information from database

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):
         follow_orders.assert_called()
         log_stage.assert_called_with("process_sell_all__2_all_buy_end")
 
+    def test_reset_all(self):
+        portfolio.BalanceStore.all = { "foo": "bar" }
+        portfolio.ReportStore.logs.append("hey")
+        portfolio.TradeStore.all.append("bouh")
+
+        helper.reset_all()
+
+        self.assertEqual(0, len(portfolio.BalanceStore.all))
+        self.assertEqual(0, len(portfolio.ReportStore.logs))
+        self.assertEqual(0, len(portfolio.TradeStore.all))
 
 @unittest.skipUnless("unit" in limits, "Unit skipped")
 class TradeStoreTest(WebMockTestCase):