django/tests/template_tests/syntax_tests/test_lorem.py

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)