X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=tests%2Fhelper.py;h=4548b169f430ed0abbd6e78ddc1963171fc5e1c9;hb=a0dcf4e0978331709da164fb0e29ae008b90fc88;hp=4cf1b41d7d1c5426090ba0938bd19f902d384fff;hpb=c682bdf4a02a45312ef1aadf8aa26136cf308414;p=perso%2FImmae%2FProjets%2FCryptomonnaies%2FCryptoportfolio%2FTrader.git diff --git a/tests/helper.py b/tests/helper.py index 4cf1b41..4548b16 100644 --- a/tests/helper.py +++ b/tests/helper.py @@ -6,19 +6,9 @@ import requests_mock from io import StringIO import portfolio, market, main, store -__all__ = ["limits", "unittest", "WebMockTestCase", "mock", "D", +__all__ = ["unittest", "WebMockTestCase", "mock", "D", "StringIO"] -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 - class WebMockTestCase(unittest.TestCase): import time @@ -42,6 +32,7 @@ class WebMockTestCase(unittest.TestCase): last_date=store.LockedVar(None), report=mock.Mock(), worker=None, + worker_tag="", worker_notify=None, worker_started=False, callback=None),