116 lines
3.9 KiB
Python
116 lines
3.9 KiB
Python
from django.test import SimpleTestCase
|
|
|
|
from .embed import YouTubeLiteEmbedFinder
|
|
from .models import BasePage
|
|
from .utils import count_words, extract_text, get_page_models, get_table_of_contents
|
|
|
|
|
|
class BasePageTestCase(SimpleTestCase):
|
|
def test_unique_body_classes(self) -> None:
|
|
body_classes = [page.body_class for page in get_page_models()]
|
|
self.assertEqual(len(body_classes), len(set(body_classes)))
|
|
|
|
def test_pages_inherit_base_page(self) -> None:
|
|
for page_model in get_page_models():
|
|
self.assertTrue(
|
|
issubclass(page_model, BasePage),
|
|
f"{page_model} does not inherit from {BasePage}.",
|
|
)
|
|
|
|
|
|
class YouTubeLiteEmbedFinderTestCase(SimpleTestCase):
|
|
def test_finds_video_id(self) -> None:
|
|
self.assertEqual(
|
|
YouTubeLiteEmbedFinder._get_video_id(
|
|
'<iframe width="200" height="113" src="https://www.youtube.com/embed/dQw4w9WgXcQ?feature=oembed" frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" allowfullscreen title=""></iframe>'
|
|
),
|
|
"dQw4w9WgXcQ",
|
|
)
|
|
with self.assertRaises(ValueError):
|
|
YouTubeLiteEmbedFinder._get_video_id("something-else")
|
|
|
|
|
|
class TableOfContentsTestCase(SimpleTestCase):
|
|
def test_creates_table_of_contents(self) -> None:
|
|
toc = get_table_of_contents(
|
|
"""
|
|
<h2>2</h2>
|
|
<h3>2.1</h3>
|
|
<h3>2.2</h3>
|
|
<h4>2.2.1</h4>
|
|
<h3>2.3</h3>
|
|
<h2>3</h2>
|
|
<h3>3.1</h3>
|
|
<h2>4</h2>
|
|
"""
|
|
)
|
|
|
|
self.assertEqual(len(toc), 3)
|
|
self.assertEqual([entry.title for entry in toc], ["2", "3", "4"])
|
|
|
|
first_entry = toc[0]
|
|
self.assertEqual(len(first_entry.children), 3)
|
|
self.assertEqual(
|
|
[entry.title for entry in first_entry.children], ["2.1", "2.2", "2.3"]
|
|
)
|
|
|
|
sub_entry = first_entry.children[1]
|
|
self.assertEqual(len(sub_entry.children), 1)
|
|
self.assertEqual([entry.title for entry in sub_entry.children], ["2.2.1"])
|
|
|
|
second_entry = toc[1]
|
|
self.assertEqual(len(second_entry.children), 1)
|
|
self.assertEqual([entry.title for entry in second_entry.children], ["3.1"])
|
|
|
|
def test_no_headings(self) -> None:
|
|
toc = get_table_of_contents("<p>I'm no heading</p>")
|
|
self.assertEqual(toc, [])
|
|
|
|
def test_no_content(self) -> None:
|
|
toc = get_table_of_contents("")
|
|
self.assertEqual(toc, [])
|
|
|
|
def test_non_sequential_headings(self) -> None:
|
|
toc = get_table_of_contents(
|
|
"""
|
|
<h2>2</h2>
|
|
<p>2 content</p>
|
|
<h3>2.1</h3>
|
|
<h3>2.2</h3>
|
|
<p>2.2 content</p>
|
|
<h5>2.2.1</h5>
|
|
<h3>2.3</h3>
|
|
"""
|
|
)
|
|
|
|
self.assertEqual(len(toc), 1)
|
|
|
|
first_entry = toc[0]
|
|
self.assertEqual(len(first_entry.children), 3)
|
|
self.assertEqual(
|
|
[entry.title for entry in first_entry.children], ["2.1", "2.2", "2.3"]
|
|
)
|
|
|
|
sub_entry = first_entry.children[1]
|
|
self.assertEqual(len(sub_entry.children), 1)
|
|
self.assertEqual([entry.title for entry in sub_entry.children], ["2.2.1"])
|
|
|
|
|
|
class ExtractTextTestCase(SimpleTestCase):
|
|
def test_extracts_text(self) -> None:
|
|
self.assertEqual(extract_text("<p><b>Hello</b> there!</p>"), "Hello there!")
|
|
self.assertEqual(
|
|
extract_text("<p>Paragraph 1</p>\n<p>Paragraph 2</p>"),
|
|
"Paragraph 1 Paragraph 2",
|
|
)
|
|
|
|
def test_plain_text(self) -> None:
|
|
self.assertEqual(extract_text("Hello there!"), "Hello there!")
|
|
|
|
|
|
class CountWordsTestCase(SimpleTestCase):
|
|
def test_counts_words(self) -> None:
|
|
self.assertEqual(count_words("a b c"), 3)
|
|
self.assertEqual(count_words("Correct Horse Battery Staple"), 4)
|
|
self.assertEqual(count_words("Hello there! How are you?"), 5)
|