From 332154e7263d01ceea9665fb51d4d24cd8b54470 Mon Sep 17 00:00:00 2001
From: Aymeric Augustin <aymeric.augustin@m4x.org>
Date: Sun, 14 Dec 2014 21:22:57 +0100
Subject: [PATCH] Added basic tests for template backends.

---
 tests/requirements/base.txt                   |  1 +
 tests/template_backends/__init__.py           |  0
 .../forbidden/template_backends/hello.html    |  1 +
 .../jinja2/template_backends/csrf.html        |  1 +
 .../jinja2/template_backends/hello.html       |  1 +
 .../template_backends/csrf.html               |  1 +
 .../template_backends/hello.html              |  1 +
 .../templates/template_backends/csrf.html     |  1 +
 .../templates/template_backends/hello.html    |  1 +
 tests/template_backends/test_django.py        |  9 +++
 tests/template_backends/test_dummy.py         | 70 +++++++++++++++++++
 tests/template_backends/test_jinja2.py        | 26 +++++++
 12 files changed, 113 insertions(+)
 create mode 100644 tests/template_backends/__init__.py
 create mode 100644 tests/template_backends/forbidden/template_backends/hello.html
 create mode 100644 tests/template_backends/jinja2/template_backends/csrf.html
 create mode 100644 tests/template_backends/jinja2/template_backends/hello.html
 create mode 100644 tests/template_backends/template_strings/template_backends/csrf.html
 create mode 100644 tests/template_backends/template_strings/template_backends/hello.html
 create mode 100644 tests/template_backends/templates/template_backends/csrf.html
 create mode 100644 tests/template_backends/templates/template_backends/hello.html
 create mode 100644 tests/template_backends/test_django.py
 create mode 100644 tests/template_backends/test_dummy.py
 create mode 100644 tests/template_backends/test_jinja2.py

diff --git a/tests/requirements/base.txt b/tests/requirements/base.txt
index c38e8695aab..3ffb042ae97 100644
--- a/tests/requirements/base.txt
+++ b/tests/requirements/base.txt
@@ -1,5 +1,6 @@
 bcrypt
 docutils
+jinja2
 # move to py2.txt when dropping Python 3.2
 mock
 numpy
diff --git a/tests/template_backends/__init__.py b/tests/template_backends/__init__.py
new file mode 100644
index 00000000000..e69de29bb2d
diff --git a/tests/template_backends/forbidden/template_backends/hello.html b/tests/template_backends/forbidden/template_backends/hello.html
new file mode 100644
index 00000000000..14240a43ee9
--- /dev/null
+++ b/tests/template_backends/forbidden/template_backends/hello.html
@@ -0,0 +1 @@
+Hu ho.
diff --git a/tests/template_backends/jinja2/template_backends/csrf.html b/tests/template_backends/jinja2/template_backends/csrf.html
new file mode 100644
index 00000000000..081577fe3ae
--- /dev/null
+++ b/tests/template_backends/jinja2/template_backends/csrf.html
@@ -0,0 +1 @@
+{{ csrf_input }}
diff --git a/tests/template_backends/jinja2/template_backends/hello.html b/tests/template_backends/jinja2/template_backends/hello.html
new file mode 100644
index 00000000000..4ce626e9be1
--- /dev/null
+++ b/tests/template_backends/jinja2/template_backends/hello.html
@@ -0,0 +1 @@
+Hello {{ name }}!
diff --git a/tests/template_backends/template_strings/template_backends/csrf.html b/tests/template_backends/template_strings/template_backends/csrf.html
new file mode 100644
index 00000000000..b1c144dcb7d
--- /dev/null
+++ b/tests/template_backends/template_strings/template_backends/csrf.html
@@ -0,0 +1 @@
+$csrf_input
diff --git a/tests/template_backends/template_strings/template_backends/hello.html b/tests/template_backends/template_strings/template_backends/hello.html
new file mode 100644
index 00000000000..7a1b4446c73
--- /dev/null
+++ b/tests/template_backends/template_strings/template_backends/hello.html
@@ -0,0 +1 @@
+Hello $name!
diff --git a/tests/template_backends/templates/template_backends/csrf.html b/tests/template_backends/templates/template_backends/csrf.html
new file mode 100644
index 00000000000..a8367c4f081
--- /dev/null
+++ b/tests/template_backends/templates/template_backends/csrf.html
@@ -0,0 +1 @@
+{% csrf_token %}
diff --git a/tests/template_backends/templates/template_backends/hello.html b/tests/template_backends/templates/template_backends/hello.html
new file mode 100644
index 00000000000..4ce626e9be1
--- /dev/null
+++ b/tests/template_backends/templates/template_backends/hello.html
@@ -0,0 +1 @@
+Hello {{ name }}!
diff --git a/tests/template_backends/test_django.py b/tests/template_backends/test_django.py
new file mode 100644
index 00000000000..5c5070cb188
--- /dev/null
+++ b/tests/template_backends/test_django.py
@@ -0,0 +1,9 @@
+from django.template.backends.django import DjangoTemplates
+
+from .test_dummy import TemplateStringsTests
+
+
+class DjangoTemplatesTests(TemplateStringsTests):
+
+    engine_class = DjangoTemplates
+    backend_name = 'django'
diff --git a/tests/template_backends/test_dummy.py b/tests/template_backends/test_dummy.py
new file mode 100644
index 00000000000..e79dbbe02a7
--- /dev/null
+++ b/tests/template_backends/test_dummy.py
@@ -0,0 +1,70 @@
+# coding: utf-8
+
+from __future__ import unicode_literals
+
+from django.http import HttpRequest
+from django.middleware.csrf import CsrfViewMiddleware, get_token
+from django.template import TemplateDoesNotExist
+from django.template.backends.dummy import TemplateStrings
+from django.test import SimpleTestCase
+
+
+class TemplateStringsTests(SimpleTestCase):
+
+    engine_class = TemplateStrings
+    backend_name = 'dummy'
+    options = {}
+
+    @classmethod
+    def setUpClass(cls):
+        params = {
+            'DIRS': [],
+            'APP_DIRS': True,
+            'NAME': cls.backend_name,
+            'OPTIONS': cls.options,
+        }
+        cls.engine = cls.engine_class(params)
+
+    def test_from_string(self):
+        template = self.engine.from_string("Hello!\n")
+        content = template.render()
+        self.assertEqual(content, "Hello!\n")
+
+    def test_get_template(self):
+        template = self.engine.get_template('template_backends/hello.html')
+        content = template.render({'name': 'world'})
+        self.assertEqual(content, "Hello world!\n")
+
+    def test_get_template_non_existing(self):
+        with self.assertRaises(TemplateDoesNotExist):
+            self.engine.get_template('template_backends/non_existing.html')
+
+    def test_html_escaping(self):
+        template = self.engine.get_template('template_backends/hello.html')
+        context = {'name': '<script>alert("XSS!");</script>'}
+        content = template.render(context)
+
+        self.assertIn('&lt;script&gt;', content)
+        self.assertNotIn('<script>', content)
+
+    def test_csrf_token(self):
+        request = HttpRequest()
+        CsrfViewMiddleware().process_view(request, lambda r: None, (), {})
+
+        template = self.engine.get_template('template_backends/csrf.html')
+        content = template.render(request=request)
+
+        expected = (
+            '<input type="hidden" name="csrfmiddlewaretoken" '
+            'value="{}" />'.format(get_token(request)))
+
+        self.assertHTMLEqual(content, expected)
+
+    def test_no_directory_traversal(self):
+        with self.assertRaises(TemplateDoesNotExist):
+            self.engine.get_template('../forbidden/template_backends/hello.html')
+
+    def test_non_ascii_characters(self):
+        template = self.engine.get_template('template_backends/hello.html')
+        content = template.render({'name': 'Jérôme'})
+        self.assertEqual(content, "Hello Jérôme!\n")
diff --git a/tests/template_backends/test_jinja2.py b/tests/template_backends/test_jinja2.py
new file mode 100644
index 00000000000..8c0b8fc8f4e
--- /dev/null
+++ b/tests/template_backends/test_jinja2.py
@@ -0,0 +1,26 @@
+import sys
+
+from unittest import skipIf
+
+# Jinja2 doesn't run on Python 3.2 because it uses u-prefixed unicode strings.
+if sys.version_info[:2] == (2, 7) or sys.version_info[:2] >= (3, 3):
+    try:
+        import jinja2
+    except ImportError:
+        jinja2 = None
+        Jinja2 = None
+    else:
+        from django.template.backends.jinja2 import Jinja2
+else:
+    jinja2 = None
+    Jinja2 = None
+
+from .test_dummy import TemplateStringsTests
+
+
+@skipIf(jinja2 is None, "this test requires jinja2")
+class Jinja2Tests(TemplateStringsTests):
+
+    engine_class = Jinja2
+    backend_name = 'jinja2'
+    options = {'keep_trailing_newline': True}