django/tests/template_tests/syntax_tests/test_lorem.py

42 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)