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__ = ["unittest", "WebMockTestCase", "mock", "D",
- "StringIO"]
+__all__ = ["limits", "unittest", "WebMockTestCase", "mock", "D",
+ "StringIO", "tz"]
+
+limits = ["acceptance", "unit"]
+for test_type in limits:
+ if "--no{}".format(test_type) in sys.argv:
+ sys.argv.remove("--no{}".format(test_type))
+ limits.remove(test_type)
+ if "--only{}".format(test_type) in sys.argv:
+ sys.argv.remove("--only{}".format(test_type))
+ 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()