ref(tests): expose TEST_ROOT in test for file based (certs, etc) access

Author: helgi

rootfs/api/tests/__init__.py

@@ -2,6 +2,7 @@
 import random
 import requests_mock
 import time
+from os.path import dirname, realpath
 
 from django.conf import settings
 from django.test.runner import DiscoverRunner
@@ -39,6 +40,9 @@ def fake_responses(request, context):
 adapter.register_uri('GET', url + '/health', text=fake_responses)
 adapter.register_uri('GET', url + '/healthz', text=fake_responses)
 
+# Root of the test directory (for files and such)
+TEST_ROOT = dirname(realpath(__file__))
+
 
 class SilentDjangoTestSuiteRunner(DiscoverRunner):
     """Prevents api log messages from cluttering the console during tests."""

rootfs/api/tests/test_build.py

@@ -3,8 +3,6 @@
 
 Run the tests with "./manage.py test api"
 """
-
-
 import json
 
 from django.contrib.auth.models import User

rootfs/api/tests/test_certificate.py

@@ -1,12 +1,11 @@
-import os
-
 from django.contrib.auth.models import User
 from django.core.cache import cache
 from rest_framework.test import APITestCase
 from rest_framework.authtoken.models import Token
 from django.core.exceptions import SuspiciousOperation
 
 from api.models import App, Certificate
+from api.tests import TEST_ROOT
 
 
 class CertificateTest(APITestCase):
@@ -24,11 +23,10 @@ def setUp(self):
         self.app = App.objects.create(owner=self.user, id='test-app')
         self.domain = 'autotest.example.com'
 
-        path = os.path.dirname(os.path.realpath(__file__))
-        with open('{}/certs/{}.key'.format(path, self.domain)) as f:
+        with open('{}/certs/{}.key'.format(TEST_ROOT, self.domain)) as f:
             self.key = f.read()
 
-        with open('{}/certs/{}.cert'.format(path, self.domain)) as f:
+        with open('{}/certs/{}.cert'.format(TEST_ROOT, self.domain)) as f:
             self.cert = f.read()
 
     def tearDown(self):

rootfs/api/tests/test_certificate_use_case_1.py

@@ -1,11 +1,10 @@
-import os
-
 from django.contrib.auth.models import User
 from django.core.cache import cache
 from rest_framework.test import APITestCase
 from rest_framework.authtoken.models import Token
 
 from api.models import App, Certificate, Domain
+from api.tests import TEST_ROOT
 
 
 class CertificateUseCase1Test(APITestCase):
@@ -27,11 +26,10 @@ def setUp(self):
         self.domain = Domain.objects.create(owner=self.user, app=self.app, domain='foo.com')
         self.name = 'foo-com'  # certificate name
 
-        path = os.path.dirname(os.path.realpath(__file__))
-        with open('{}/certs/{}.key'.format(path, self.domain)) as f:
+        with open('{}/certs/{}.key'.format(TEST_ROOT, self.domain)) as f:
             self.key = f.read()
 
-        with open('{}/certs/{}.cert'.format(path, self.domain)) as f:
+        with open('{}/certs/{}.cert'.format(TEST_ROOT, self.domain)) as f:
             self.cert = f.read()
 
     def tearDown(self):

rootfs/api/tests/test_certificate_use_case_2.py

@@ -1,11 +1,10 @@
-import os
-
 from django.contrib.auth.models import User
 from django.core.cache import cache
 from rest_framework.test import APITestCase
 from rest_framework.authtoken.models import Token
 
 from api.models import App, Certificate, Domain
+from api.tests import TEST_ROOT
 
 
 class CertificateUseCase2Test(APITestCase):
@@ -32,12 +31,11 @@ def setUp(self):
         # only foo.com has a cert
         self.domain = 'foo.com'
 
-        path = os.path.dirname(os.path.realpath(__file__))
         self.certificates = {self.domain: {'name': self.domain.replace('.', '-')}}
-        with open('{}/certs/{}.key'.format(path, self.domain)) as f:
+        with open('{}/certs/{}.key'.format(TEST_ROOT, self.domain)) as f:
             self.certificates[self.domain]['key'] = f.read()
 
-        with open('{}/certs/{}.cert'.format(path, self.domain)) as f:
+        with open('{}/certs/{}.cert'.format(TEST_ROOT, self.domain)) as f:
             self.certificates[self.domain]['cert'] = f.read()
 
         # add expires and fingerprints

rootfs/api/tests/test_certificate_use_case_3.py

@@ -1,11 +1,10 @@
-import os
-
 from django.contrib.auth.models import User
 from django.core.cache import cache
 from rest_framework.test import APITestCase
 from rest_framework.authtoken.models import Token
 
 from api.models import App, Certificate, Domain
+from api.tests import TEST_ROOT
 
 
 class CertificateUseCase3Test(APITestCase):
@@ -31,15 +30,13 @@ def setUp(self):
 
         self.certificates = {}
 
-        path = os.path.dirname(os.path.realpath(__file__))
-
         # load up the certs
         for domain in self.domains:
             self.certificates[domain] = {'name': domain.replace('.', '-')}
-            with open('{}/certs/{}.key'.format(path, domain)) as f:
+            with open('{}/certs/{}.key'.format(TEST_ROOT, domain)) as f:
                 self.certificates[domain]['key'] = f.read()
 
-            with open('{}/certs/{}.cert'.format(path, domain)) as f:
+            with open('{}/certs/{}.cert'.format(TEST_ROOT, domain)) as f:
                 self.certificates[domain]['cert'] = f.read()
 
         # add expires and fingerprints

rootfs/api/tests/test_certificate_use_case_4.py

@@ -1,11 +1,10 @@
-import os
-
 from django.contrib.auth.models import User
 from django.core.cache import cache
 from rest_framework.test import APITestCase
 from rest_framework.authtoken.models import Token
 
 from api.models import App, Certificate, Domain
+from api.tests import TEST_ROOT
 
 
 class CertificateUseCase4Test(APITestCase):
@@ -32,8 +31,6 @@ def setUp(self):
 
         self.certificates = {}
 
-        path = os.path.dirname(os.path.realpath(__file__))
-
         # load up the certs
         for domain in self.domains:
             self.certificates[domain] = {'name': domain.replace('.', '-').replace('*', 'wildcard')}
@@ -42,10 +39,10 @@ def setUp(self):
                 # Cheap hack
                 filename = domain.replace('*', 'www')
 
-            with open('{}/certs/{}.key'.format(path, filename)) as f:
+            with open('{}/certs/{}.key'.format(TEST_ROOT, filename)) as f:
                 self.certificates[domain]['key'] = f.read()
 
-            with open('{}/certs/{}.cert'.format(path, filename)) as f:
+            with open('{}/certs/{}.cert'.format(TEST_ROOT, filename)) as f:
                 self.certificates[domain]['cert'] = f.read()
 
         # add expires, common_name and fingerprints

rootfs/api/tests/test_certificate_use_case_5.py

@@ -1,11 +1,10 @@
-import os
-
 from django.contrib.auth.models import User
 from django.core.cache import cache
 from rest_framework.test import APITestCase
 from rest_framework.authtoken.models import Token
 
 from api.models import App, Certificate, Domain
+from api.tests import TEST_ROOT
 
 
 class CertificateUseCase5Test(APITestCase):
@@ -33,17 +32,15 @@ def setUp(self):
 
         self.certificates = {}
 
-        path = os.path.dirname(os.path.realpath(__file__))
-
         # load up the certs
         for domain in self.domains:
             self.certificates[domain] = {'name': domain.replace('.', '-').replace('*', 'wildcard')}
             filename = domain.replace('*', 'wildcard')
 
-            with open('{}/certs/{}.key'.format(path, filename)) as f:
+            with open('{}/certs/{}.key'.format(TEST_ROOT, filename)) as f:
                 self.certificates[domain]['key'] = f.read()
 
-            with open('{}/certs/{}.cert'.format(path, filename)) as f:
+            with open('{}/certs/{}.cert'.format(TEST_ROOT, filename)) as f:
                 self.certificates[domain]['cert'] = f.read()
 
         # add expires, common_name and fingerprints

rootfs/api/tests/test_pods.py

@@ -3,8 +3,6 @@
 
 Run the tests with "./manage.py test api"
 """
-
-
 import json
 
 from django.contrib.auth.models import User