X-Git-Url: http://git.ithinksw.org/philo.git/blobdiff_plain/8d3b1128e7925825332eafc7d269cf34e1af170c..92e8e96c8d6814f3bd93ecd82d8da2ccb43e13c4:/tests.py

diff --git a/tests.py b/tests.py
index 1dc48b7..96ac7b6 100644
--- a/tests.py
+++ b/tests.py
@@ -2,9 +2,109 @@ from django.test import TestCase
 from django import template
 from django.conf import settings
 from django.db import connection
+from django.template import loader
+from django.template.loaders import cached
 from philo.exceptions import AncestorDoesNotExist
 from philo.models import Node, Page, Template
 from philo.contrib.penfield.models import Blog, BlogView, BlogEntry
+import sys, traceback
+
+
+class TemplateTestCase(TestCase):
+	fixtures = ['test_fixtures.json']
+	
+	def test_templates(self):
+		"Tests to make sure that embed behaves with complex includes and extends"
+		template_tests = self.get_template_tests()
+		
+		# Register our custom template loader. Shamelessly cribbed from django core regressiontests.
+		def test_template_loader(template_name, template_dirs=None):
+			"A custom template loader that loads the unit-test templates."
+			try:
+				return (template_tests[template_name][0] , "test:%s" % template_name)
+			except KeyError:
+				raise template.TemplateDoesNotExist, template_name
+		
+		cache_loader = cached.Loader(('test_template_loader',))
+		cache_loader._cached_loaders = (test_template_loader,)
+		
+		old_template_loaders = loader.template_source_loaders
+		loader.template_source_loaders = [cache_loader]
+		
+		# Turn TEMPLATE_DEBUG off, because tests assume that.
+		old_td, settings.TEMPLATE_DEBUG = settings.TEMPLATE_DEBUG, False
+		
+		# Set TEMPLATE_STRING_IF_INVALID to a known string.
+		old_invalid = settings.TEMPLATE_STRING_IF_INVALID
+		expected_invalid_str = 'INVALID'
+		
+		failures = []
+		tests = template_tests.items()
+		tests.sort()
+		# Run tests
+		for name, vals in tests:
+			xx, context, result = vals
+			try:
+				test_template = loader.get_template(name)
+				output = test_template.render(template.Context(context))
+			except Exception:
+				exc_type, exc_value, exc_tb = sys.exc_info()
+				if exc_type != result:
+					tb = '\n'.join(traceback.format_exception(exc_type, exc_value, exc_tb))
+					failures.append("Template test %s -- FAILED. Got %s, exception: %s\n%s" % (name, exc_type, exc_value, tb))
+				continue
+			if output != result:
+				failures.append("Template test %s -- FAILED. Expected %r, got %r" % (name, result, output))
+		
+		# Cleanup
+		settings.TEMPLATE_DEBUG = old_td
+		settings.TEMPLATE_STRING_IF_INVALID = old_invalid
+		loader.template_source_loaders = old_template_loaders
+		
+		self.assertEqual(failures, [], "Tests failed:\n%s\n%s" % ('-'*70, ("\n%s\n" % ('-'*70)).join(failures)))
+	
+	
+	def get_template_tests(self):
+		# SYNTAX --
+		# 'template_name': ('template contents', 'context dict', 'expected string output' or Exception class)
+		blog = Blog.objects.all()[0]
+		return {
+			# EMBED INCLUSION HANDLING
+			
+			'embed01': ('{{ embedded.title|safe }}', {'embedded': blog}, blog.title),
+			'embed02': ('{{ embedded.title|safe }}{{ var1 }}{{ var2 }}', {'embedded': blog}, blog.title),
+			'embed03': ('{{ embedded.title|safe }} is a lie!', {'embedded': blog}, '%s is a lie!' % blog.title),
+			
+			# Simple template structure with embed
+			'simple01': ('{% embed penfield.blog with "embed01" %}{% embed penfield.blog 1 %}Simple{% block one %}{% endblock %}', {'blog': blog}, '%sSimple' % blog.title),
+			'simple02': ('{% extends "simple01" %}', {}, '%sSimple' % blog.title),
+			'simple03': ('{% embed penfield.blog with "embed000" %}', {}, settings.TEMPLATE_STRING_IF_INVALID),
+			'simple04': ('{% embed penfield.blog 1 %}', {}, settings.TEMPLATE_STRING_IF_INVALID),
+			'simple05': ('{% embed penfield.blog with "embed01" %}{% embed blog %}', {'blog': blog}, blog.title),
+			
+			# Kwargs
+			'kwargs01': ('{% embed penfield.blog with "embed02" %}{% embed penfield.blog 1 var1="hi" var2=lo %}', {'lo': 'lo'}, '%shilo' % blog.title),
+			
+			# Filters/variables
+			'filters01': ('{% embed penfield.blog with "embed02" %}{% embed penfield.blog 1 var1=hi|first var2=lo|slice:"3" %}', {'hi': ["These", "words"], 'lo': 'lower'}, '%sTheselow' % blog.title),
+			'filters02': ('{% embed penfield.blog with "embed01" %}{% embed penfield.blog entry %}', {'entry': 1}, blog.title),
+			
+			# Blocky structure
+			'block01': ('{% block one %}Hello{% endblock %}', {}, 'Hello'),
+			'block02': ('{% extends "simple01" %}{% block one %}{% embed penfield.blog 1 %}{% endblock %}', {}, "%sSimple%s" % (blog.title, blog.title)),
+			'block03': ('{% extends "simple01" %}{% embed penfield.blog with "embed03" %}{% block one %}{% embed penfield.blog 1 %}{% endblock %}', {}, "%sSimple%s is a lie!" % (blog.title, blog.title)),
+			
+			# Blocks and includes
+			'block-include01': ('{% extends "simple01" %}{% embed penfield.blog with "embed03" %}{% block one %}{% include "simple01" %}{% embed penfield.blog 1 %}{% endblock %}', {}, "%sSimple%sSimple%s is a lie!" % (blog.title, blog.title, blog.title)),
+			'block-include02': ('{% extends "simple01" %}{% block one %}{% include "simple04" %}{% embed penfield.blog with "embed03" %}{% include "simple04" %}{% embed penfield.blog 1 %}{% endblock %}', {}, "%sSimple%s%s is a lie!%s is a lie!" % (blog.title, blog.title, blog.title, blog.title)),
+			
+			# Tests for more complex situations...
+			'complex01': ('{% block one %}{% endblock %}complex{% block two %}{% endblock %}', {}, 'complex'),
+			'complex02': ('{% extends "complex01" %}', {}, 'complex'),
+			'complex03': ('{% extends "complex02" %}{% embed penfield.blog with "embed01" %}', {}, 'complex'),
+			'complex04': ('{% extends "complex03" %}{% block one %}{% embed penfield.blog 1 %}{% endblock %}', {}, '%scomplex' % blog.title),
+			'complex05': ('{% extends "complex03" %}{% block one %}{% include "simple04" %}{% endblock %}', {}, '%scomplex' % blog.title),
+		}
 
 
 class NodeURLTestCase(TestCase):
@@ -125,4 +225,4 @@ class TreePathTestCase(TestCase):
 		self.assertQueryLimit(1, 'root/second/third', callable=third.get_path)
 		self.assertQueryLimit(1, 'second/third', root, callable=third.get_path)
 		self.assertQueryLimit(1, e, third, callable=second2.get_path)
-		self.assertQueryLimit(1, '? - ?', root, ' - ', 'title', callable=third.get_path)
\ No newline at end of file
+		self.assertQueryLimit(1, '? - ?', root, ' - ', 'title', callable=third.get_path)