Ticket #2016: 0004-Adds-tests-for-object_tag.diff

new file tests/regressiontests/templates/object_tag.py

@@ +1 @@
+object_tag_tests = """
+>>> t = template.Template('{% load custom %}{% get_meaning_of_life as answer %}{{ answer }}')
+>>> t.render(template.Context())
+u"42"
+>>> t = template.Template('{% load custom %}{% get_mascot_with "awesomeness" "magical powers" as pony %}{{ pony }}')
+>>> t.render(template.Context())
+u"This mascot is filled with awesomeness and magical powers"
+>>> t = template.Template('{% load custom %}{% count "e" word as num %}{{ num }}')
+>>> t.render(template.Context({'word': 'beefcake'}))
+u"3"
+"""

tests/regressiontests/templates/templatetags/custom.py

@@ -9 +9 @@
 
 register.filter(trim)
 
+def get_meaning_of_life():
+    return 42
+register.object_tag(get_meaning_of_life)
+
+def get_mascot_with(power_one, power_two):
+    return 'This mascot is filled with %s and %s' % (power_one, power_two)
+register.object_tag(get_mascot_with)
+
+def count(char, word):
+    return word.count(char)
+register.object_tag(count)

tests/regressiontests/templates/tests.py

@@ -24 +24 @@
 from custom import custom_filters
 from parser import filter_parsing, variable_parsing
 from unicode import unicode_tests
+from object_tag import object_tag_tests
 
 try:
     from loaders import *
@@ -38 +39 @@
     'context': context_tests,
     'filter_parsing': filter_parsing,
     'custom_filters': custom_filters,
+    'object_tag': object_tag_tests,
 }
 
 #################################