X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=tests%2Fhelper.py;h=930d04a4cc17f86fbe945794a2716834f07f332e;hb=78ded184e2c7cbb0bf62fc5becd4e4dd3dc59cc0;hp=935e0601f1ad9bbd82433ea841fd89474f1ad2c5;hpb=855a1f7036403c1fc10b9ede13b0dd18ec83929d;p=perso%2FImmae%2FProjets%2FCryptomonnaies%2FCryptoportfolio%2FTrader.git

diff --git a/tests/helper.py b/tests/helper.py
index 935e060..930d04a 100644
--- a/tests/helper.py
+++ b/tests/helper.py
@@ -5,9 +5,10 @@ from unittest import mock
 import requests_mock
 from io import StringIO
 import portfolio, market, main, store, dbs
+import datetime
 
 __all__ = ["limits", "unittest", "WebMockTestCase", "mock", "D",
-        "StringIO"]
+        "StringIO", "tz"]
 
 limits = ["acceptance", "unit"]
 for test_type in limits:
@@ -19,6 +20,9 @@ for test_type in limits:
         limits = [test_type]
         break
 
+def tz(hours):
+    return datetime.timezone(datetime.timedelta(hours=hours))
+
 class WebMockTestCase(unittest.TestCase):
     import time
 
@@ -45,6 +49,7 @@ 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),