Add main running file and fetch information from database

[perso/Immae/Projets/Cryptomonnaies/Cryptoportfolio/Trader.git] / test.py

diff --git a/test.py b/test.py
index 34b2fe3c228cf46b70878ea998965270eb273109..fc331c314825261dc30c8f4947e040064083f502 100644 (file)
--- 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):