Move environment variable checks inside the test case constructor

ivknv · Jul 10, 2021 · 2c56a2e · 2c56a2e
1 parent b18201f
commit 2c56a2e
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/tests/resources_test.py b/tests/resources_test.py
@@ -15,27 +15,25 @@
 yadisk_async.settings.DEFAULT_N_RETRIES = 50
 yadisk_async.settings.DEFAULT_UPLOAD_N_RETRIES = 50
 
-if not os.environ.get("PYTHON_YADISK_APP_TOKEN"):
-    raise ValueError("Environment variable PYTHON_YADISK_APP_TOKEN must be set")
-
-if not os.environ.get("PYTHON_YADISK_TEST_ROOT"):
-    raise ValueError("Environment variable PYTHON_YADISK_TEST_ROOT must be set")
-
-
 def async_test(f):
     def wrapper(*args, **kwargs):
         loop = asyncio.get_event_loop()
         loop.run_until_complete(f(*args, **kwargs))
 
     return wrapper
 
-
 class ResourcesTestCase(TestCase):
     def __init__(self, *args, **kwargs):
         self.yadisk = None
 
         TestCase.__init__(self, *args, **kwargs)
 
+        if not os.environ.get("PYTHON_YADISK_APP_TOKEN"):
+            raise ValueError("Environment variable PYTHON_YADISK_APP_TOKEN must be set")
+
+        if not os.environ.get("PYTHON_YADISK_TEST_ROOT"):
+            raise ValueError("Environment variable PYTHON_YADISK_TEST_ROOT must be set")
+
         self.yadisk = yadisk_async.YaDisk(os.environ.get("PYTHON_YADISK_APP_ID"),
                                           os.environ.get("PYTHON_YADISK_APP_SECRET"),
                                           os.environ.get("PYTHON_YADISK_APP_TOKEN"))