django/tests/template_tests/syntax_tests/test_lorem.py

43 lines
1.6 KiB
Python

from django.template.base import TemplateSyntaxError
from django.test import SimpleTestCase
from django.utils.lorem_ipsum import COMMON_P, WORDS
from ..utils import setup
class LoremTagTests(SimpleTestCase):
@setup({'lorem1': '{% lorem 3 w %}'})
def test_lorem1(self):
output = self.engine.render_to_string('lorem1')
self.assertEqual(output, 'lorem ipsum dolor')
@setup({'lorem_random': '{% lorem 3 w random %}'})
def test_lorem_random(self):
output = self.engine.render_to_string('lorem_random')
words = output.split(' ')
self.assertEqual(len(words), 3)
for word in words:
self.assertIn(word, WORDS)
@setup({'lorem_default': '{% lorem %}'})
def test_lorem_default(self):
output = self.engine.render_to_string('lorem_default')
self.assertEqual(output, COMMON_P)
@setup({'lorem_syntax_error': '{% lorem 1 2 3 4 %}'})
def test_lorem_syntax(self):
msg = "Incorrect format for 'lorem' tag"
with self.assertRaisesMessage(TemplateSyntaxError, msg):
self.engine.render_to_string('lorem_syntax_error')
@setup({'lorem_multiple_paragraphs': '{% lorem 2 p %}'})
def test_lorem_multiple_paragraphs(self):
output = self.engine.render_to_string('lorem_multiple_paragraphs')
self.assertEqual(output.count('<p>'), 2)
@setup({'lorem_incorrect_count': '{% lorem two p %}'})
def test_lorem_incorrect_count(self):
output = self.engine.render_to_string('lorem_incorrect_count')
self.assertEqual(output.count('<p>'), 1)