diff --git a/tests/tests/test_utils.py b/tests/tests/test_utils.py index d65f23cd..615ecc67 100644 --- a/tests/tests/test_utils.py +++ b/tests/tests/test_utils.py @@ -37,6 +37,14 @@ def test_parent_template_from_variable(self): ], ) + +class TestGetTemplateDirs(SimpleTestCase): + def get_relative_template_dirs(self): + """Make paths relative with a predefined root so we can use them in assertions.""" + base = os.path.dirname(settings.BASE_DIR) + dirs = get_template_dirs() + return ['/'.join(str(d).replace(base, 'dpl').split('/')[-4:-1]) for d in dirs] + @override_settings(TEMPLATES=[ { 'BACKEND': 'django.template.backends.django.DjangoTemplates', @@ -44,8 +52,7 @@ def test_parent_template_from_variable(self): }, ]) def test_get_template_dirs_app_dirs(self): - template_dirs = ['/'.join(d.replace(os.path.dirname(settings.BASE_DIR), 'dpl').split('/')[-4:-1]) for d in get_template_dirs()] - self.assertListEqual(template_dirs, [ + self.assertListEqual(self.get_relative_template_dirs(), [ 'django/contrib/auth', 'dpl/pattern_library', 'dpl/tests', @@ -65,8 +72,7 @@ def test_get_template_dirs_app_dirs(self): }, ]) def test_get_template_dirs_list_dirs(self): - template_dirs = ['/'.join(d.replace(os.path.dirname(settings.BASE_DIR), 'dpl').split('/')[-4:-1]) for d in get_template_dirs()] - self.assertListEqual(template_dirs, [ + self.assertListEqual(self.get_relative_template_dirs(), [ 'dpl/tests/test_one', 'dpl/tests/test_two', 'django/contrib/auth',