Add multi tenant template tests

tenant_schemas/template_loaders.py

@@ -40,7 +40,11 @@ def __init__(self, engine, loaders):
     def cache_key(template_name, template_dirs):
         if connection.tenant:
             if not template_dirs:
-                template_dirs = settings.MULTITENANT_TEMPLATE_DIRS
+                try:
+                    template_dirs = settings.MULTITENANT_TEMPLATE_DIRS
+                except AttributeError:
+                    raise ImproperlyConfigured('To use %s.%s you must define the MULTITENANT_TEMPLATE_DIRS' %
+                                               (__name__, CachedLoader.__name__))
             return '-'.join([str(connection.tenant.pk), template_name,
                              hashlib.sha1(force_bytes('|'.join(template_dirs))).hexdigest()])
         if template_dirs:

tenant_schemas/tests/template_loader/templates/tenant/hello.html

@@ -0,0 +1 @@
+Hello Tenant! (Django templates)

tenant_schemas/tests/template_loader/test_cached_template_loader.py

@@ -2,6 +2,9 @@
 
 from django.template.loader import get_template
 from django.test import SimpleTestCase, override_settings
+from django.db import connection
+
+from tenant_schemas.tests.models import Tenant
 
 
 @override_settings(
@@ -23,9 +26,18 @@
                 ]
             },
         }
+    ],
+    MULTITENANT_TEMPLATE_DIRS = [
+        os.path.join(os.path.dirname(__file__), "templates")
     ]
 )
 class CachedLoaderTests(SimpleTestCase):
     def test_get_template(self):
         template = get_template("hello.html")
         self.assertEqual(template.render(), "Hello! (Django templates)\n")
+
+    def test_get_tenant_template(self):
+        connection.tenant = Tenant(domain_url="tenant", schema_name="tenant")
+        connection..tenant.save()
+        template = get_template("hello.html")
+        self.assertEqual(template.render(), "Hello Tenant! (Django templates)\n")