Don’t raise when some market is disabled

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

diff --git a/tests/helper.py b/tests/helper.py
index b85bf3ac58d8967edf0c7d691e3de2870b3f1374..a2f8a2243c5e6966a96ccda794741291970d201a 100644 (file)
--- a/tests/helper.py
+++ b/tests/helper.py
@@ -4,7 +4,7 @@ from decimal import Decimal as D
 from unittest import mock
 import requests_mock
 from io import StringIO
-import portfolio, market, main, store
+import portfolio, market, main, store, dbs
 
 __all__ = ["limits", "unittest", "WebMockTestCase", "mock", "D",
         "StringIO"]
@@ -45,9 +45,12 @@ class WebMockTestCase(unittest.TestCase):
                     worker_tag="",
                     worker_notify=None,
                     worker_started=False,
+                    poll_started_at=None,
                     callback=None),
                 mock.patch.multiple(portfolio.Computation,
                     computations=portfolio.Computation.computations),
+                mock.patch.multiple(dbs,
+                    redis=None, psql=None)
                 ]
         for patcher in self.patchers:
             patcher.start()