X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=tests%2Fhelper.py;h=935e0601f1ad9bbd82433ea841fd89474f1ad2c5;hb=5f721612111af5c56b5757cb2f21da5f2fa388bf;hp=4cf1b41d7d1c5426090ba0938bd19f902d384fff;hpb=c682bdf4a02a45312ef1aadf8aa26136cf308414;p=perso%2FImmae%2FProjets%2FCryptomonnaies%2FCryptoportfolio%2FTrader.git

diff --git a/tests/helper.py b/tests/helper.py
index 4cf1b41..935e060 100644
--- 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"]
@@ -42,11 +42,14 @@ class WebMockTestCase(unittest.TestCase):
                     last_date=store.LockedVar(None),
                     report=mock.Mock(),
                     worker=None,
+                    worker_tag="",
                     worker_notify=None,
                     worker_started=False,
                     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()