kawaz.core.forms.tests.test_templatetags のソースコード

# ! -*- coding: utf-8 -*-
#
#
#

from django.test import TestCase
from django.template import Template, Context, TemplateSyntaxError
from ..helpers import Bootstrap3HorizontalFormHelper, HorizontalBareFormHelper, InlineBareFormHelper


class GetFormHelperTemplateTagTestCase(TestCase):
    def _render_template(self, type=''):
        t = Template(
            "{{% load form_helper %}}"
            "{{% get_form_helper {} as helper %}}".format(
                "'{}'".format(type) if type else ''
            )
        )
        c = Context()
        r = t.render(c)
        # get_blog_helper は何も描画しない
        self.assertEqual(r.strip(), "")
        return c['helper']

    def test_get_helper_horizontal(self):
        """get_helper horizontalでBootstrap3HorizontalFormHelperを取り出せる"""
        helper = self._render_template(type='horizontal')
        self.assertEqual(type(helper), Bootstrap3HorizontalFormHelper)

    def test_get_helper_horizontal_bare(self):
        """get_helper bareでHorizontalBareFormHelperを取り出せる"""
        helper = self._render_template(type='bare')
        self.assertEqual(type(helper), HorizontalBareFormHelper)

    def test_get_helper_inline_bare(self):
        """get_helper inline_bareでInlineBareFormHelperを取り出せる"""
        helper = self._render_template(type='inline_bare')
        self.assertEqual(type(helper), InlineBareFormHelper)

    def test_get_helper_unknown(self):
        """get_helperに存在しないtypeを渡したとき、TemplateSyntaxErrorが返る"""
        self.assertRaises(TemplateSyntaxError, self._render_template, 'unknown')