26 lines
1.2 KiB
Python
26 lines
1.2 KiB
Python
from unittest import TestCase
|
|
from bs4 import BeautifulSoup
|
|
from html_to_django.formatters import for_wrap_formatter
|
|
|
|
|
|
class ForWrapTests(TestCase):
|
|
def test_basic(self) -> None:
|
|
soup = BeautifulSoup('<div dj-for-wrap="item in list"></div>', "html.parser")
|
|
for_wrap_formatter.format(soup)
|
|
expected_result = "{% for item in list %}<div></div>{% endfor %}"
|
|
self.assertEqual(expected_result, str(soup))
|
|
|
|
def test_with_content(self) -> None:
|
|
soup = BeautifulSoup('<div dj-for-wrap="item in list">Test</div>', "html.parser")
|
|
for_wrap_formatter.format(soup)
|
|
expected_result = "{% for item in list %}<div>Test</div>{% endfor %}"
|
|
self.assertEqual(expected_result, str(soup))
|
|
|
|
def test_with_for_wrap_inside(self) -> None:
|
|
soup = BeautifulSoup('<div dj-for-wrap="item in list"><div dj-for-wrap="subitem in sublist">Test</div></div>',
|
|
"html.parser")
|
|
for_wrap_formatter.format(soup)
|
|
expected_result = ("{% for item in list %}<div>{% for subitem in sublist %}<div>Test</div>{% endfor %}"
|
|
"</div>{% endfor %}")
|
|
self.assertEqual(expected_result, str(soup))
|