from unittest import mock
import requests_mock
from io import StringIO
-import portfolio, market, main, store
+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:
limits = [test_type]
break
+def tz(hours):
+ return datetime.timezone(datetime.timedelta(hours=hours))
+
class WebMockTestCase(unittest.TestCase):
import time
last_date=store.LockedVar(None),
report=mock.Mock(),
worker=None,
+ 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()