From 0ea9c6a03893fb633b8be85136850e3b012f5164 Mon Sep 17 00:00:00 2001
From: Christian Sauer <sauerc@gmail.com>
Date: Wed, 17 Aug 2016 11:08:45 -0400
Subject: [PATCH] tests for user templates

---
 tests/settings.py               | 18 ++++++++++++++++++
 tests/templates/alt/add.html    |  1 +
 tests/templates/alt/change.html |  1 +
 tests/templates/alt/delete.html |  1 +
 tests/tests.py                  | 33 +++++++++++++++++++++++++++++++++
 5 files changed, 54 insertions(+)
 create mode 100644 tests/templates/alt/add.html
 create mode 100644 tests/templates/alt/change.html
 create mode 100644 tests/templates/alt/delete.html

diff --git a/tests/settings.py b/tests/settings.py
index c85df2d..471b8fe 100644
--- a/tests/settings.py
+++ b/tests/settings.py
@@ -1,5 +1,9 @@
+import os
 import django
 
+VERSION = django.VERSION
+SETTINGS_DIR = os.path.dirname(__file__)
+
 DATABASE_ENGINE = 'sqlite3'
 
 DATABASES = {
@@ -25,6 +29,20 @@ MIDDLEWARE_CLASSES = (
     "django.contrib.messages.middleware.MessageMiddleware",
 )
 
+if VERSION[0] == 1 and VERSION[1] < 8:
+    TEMPLATE_DIRS = (
+        os.path.join(SETTINGS_DIR, 'templates'),
+    )
+else:
+    TEMPLATES = [
+        {
+            'BACKEND': 'django.template.backends.django.DjangoTemplates',
+            'APP_DIRS': True,
+            'DIRS': [
+                os.path.join(SETTINGS_DIR, 'templates')
+            ]
+        }
+    ]
 
 ROOT_URLCONF = 'tests.urls'
 
diff --git a/tests/templates/alt/add.html b/tests/templates/alt/add.html
new file mode 100644
index 0000000..37d4839
--- /dev/null
+++ b/tests/templates/alt/add.html
@@ -0,0 +1 @@
+ALTERNATE ADD TEMPLATE
diff --git a/tests/templates/alt/change.html b/tests/templates/alt/change.html
new file mode 100644
index 0000000..bb8650f
--- /dev/null
+++ b/tests/templates/alt/change.html
@@ -0,0 +1 @@
+ALTERNATE CHANGE TEMPLATE
diff --git a/tests/templates/alt/delete.html b/tests/templates/alt/delete.html
new file mode 100644
index 0000000..10bfe9d
--- /dev/null
+++ b/tests/templates/alt/delete.html
@@ -0,0 +1 @@
+ALTERNATE DELETE TEMPLATE
diff --git a/tests/tests.py b/tests/tests.py
index 9c3874c..e8c82f1 100644
--- a/tests/tests.py
+++ b/tests/tests.py
@@ -210,6 +210,39 @@ class AvatarTests(TestCase):
         self.assertIn('<img src="{}"'.format(avatar.avatar_url(100)), result)
         self.assertIn('alt="test" width="100" height="100" />', result)
 
+    def test_default_add_template(self):
+        response = self.client.get('/avatar/add/')
+        self.assertContains(response, 'Upload New Image')
+        self.assertNotContains(response, 'ALTERNATE ADD TEMPLATE')
+
+    @override_settings(AVATAR_ADD_TEMPLATE='alt/add.html')
+    def test_custom_add_template(self):
+        response = self.client.get('/avatar/add/')
+        self.assertNotContains(response, 'Upload New Image')
+        self.assertContains(response, 'ALTERNATE ADD TEMPLATE')
+
+    def test_default_change_template(self):
+        response = self.client.get('/avatar/change/')
+        self.assertContains(response, 'Upload New Image')
+        self.assertNotContains(response, 'ALTERNATE CHANGE TEMPLATE')
+
+    @override_settings(AVATAR_CHANGE_TEMPLATE='alt/change.html')
+    def test_custom_change_template(self):
+        response = self.client.get('/avatar/change/')
+        self.assertNotContains(response, 'Upload New Image')
+        self.assertContains(response, 'ALTERNATE CHANGE TEMPLATE')
+
+    def test_default_delete_template(self):
+        response = self.client.get('/avatar/delete/')
+        self.assertContains(response, 'like to delete.')
+        self.assertNotContains(response, 'ALTERNATE DELETE TEMPLATE')
+
+    @override_settings(AVATAR_DELETE_TEMPLATE='alt/delete.html')
+    def test_custom_delete_template(self):
+        response = self.client.get('/avatar/delete/')
+        self.assertNotContains(response, 'like to delete.')
+        self.assertContains(response, 'ALTERNATE DELETE TEMPLATE')
+    
     # def testAvatarOrder
     # def testReplaceAvatarWhenMaxIsOne
     # def testHashFileName