diff --git a/test_see_models.py b/test_see_models.py index 4ee96e7..5331b25 100644 --- a/test_see_models.py +++ b/test_see_models.py @@ -60,6 +60,14 @@ DEFAULT_HOST = "10.101.60.25" DEFAULT_PORT = 6184 BASE_PATH = "/see" +_DEFAULT_ARGS = argparse.Namespace( + host=DEFAULT_HOST, + port=DEFAULT_PORT, + asserts=None, + verbose=False, + no_color=False +) + TEXT_RULES = [ # (keyword, text) — первое совпадение побеждает ("fio", "Иванов Иван Иванович"), @@ -167,6 +175,8 @@ class SeeAPI: class TestSeeModels(unittest.TestCase): @classmethod def setUpClass(cls): + if not hasattr(cls, 'args'): + cls.args = _DEFAULT_ARGS cls.api = SeeAPI(cls.args.host, cls.args.port) cls.asserts = load_asserts(cls.args.asserts) data = cls.api.get_models() diff --git a/test_smc_models.py b/test_smc_models.py index 41d0990..bcce8db 100644 --- a/test_smc_models.py +++ b/test_smc_models.py @@ -55,6 +55,14 @@ DEFAULT_HOST = "10.101.60.27" DEFAULT_PORT = 6181 BASE_PATH = "/smc" +_DEFAULT_ARGS = argparse.Namespace( + host=DEFAULT_HOST, + port=DEFAULT_PORT, + asserts=None, + verbose=False, + no_color=False +) + TEXT_RULES = [ ("филиал", "родники"), ("offices", "родники"), @@ -112,6 +120,8 @@ class SmcAPI: class TestSmcModels(unittest.TestCase): @classmethod def setUpClass(cls): + if not hasattr(cls, 'args'): + cls.args = _DEFAULT_ARGS cls.api = SmcAPI(cls.args.host, cls.args.port) cls.asserts = load_asserts(cls.args.asserts) data = cls.api.get_models()