X-Git-Url: http://git.scottworley.com/paperdoorknob/blobdiff_plain/e6adf6ced68d667429975110f2d1a1bd9c8d79b6..70adfbff16123a62d6b25f8ead23789d831d1d70:/glowfic_test.py diff --git a/glowfic_test.py b/glowfic_test.py index d847333..64da164 100644 --- a/glowfic_test.py +++ b/glowfic_test.py @@ -9,42 +9,83 @@ import unittest from bs4 import BeautifulSoup -from glowfic import chunkDOMs +from images import FakeImageStore +from glowfic import makeChunk, Thread +from texify import PandocTexifier class TestSplit(unittest.TestCase): def testSplit1(self) -> None: - soup = BeautifulSoup(b''' + t = Thread(BeautifulSoup(b'''
The "post" -
''', 'html.parser') - self.assertEqual([list(t.stripped_strings) for t in chunkDOMs(soup)], + ''', 'html.parser')) + self.assertEqual([list(t.stripped_strings) for t in t.chunkDOMs()], [['The "post"']]) def testSplit2(self) -> None: - soup = BeautifulSoup(b''' + t = Thread(BeautifulSoup(b'''
The "post"
The "reply"
- ''', 'html.parser') - self.assertEqual([list(t.stripped_strings) for t in chunkDOMs(soup)], + ''', 'html.parser')) + self.assertEqual([list(t.stripped_strings) for t in t.chunkDOMs()], [['The "post"'], ['The "reply"']]) def testSplit3(self) -> None: - soup = BeautifulSoup(b''' + t = Thread(BeautifulSoup(b'''
The "post"
1st reply
2nd reply
- ''', 'html.parser') - self.assertEqual([list(t.stripped_strings) for t in chunkDOMs(soup)], + ''', 'html.parser')) + self.assertEqual([list(t.stripped_strings) for t in t.chunkDOMs()], [['The "post"'], ['1st reply'], ['2nd reply']]) +class TestMakeChunk(unittest.TestCase): + + def testEmptyContent(self) -> None: + with open('testdata/empty-content.html', 'rb') as f: + t = Thread(BeautifulSoup(f, 'html.parser')) + c = makeChunk(next(iter(t.chunkDOMs())), FakeImageStore()) + self.assertEqual( + c.icon, + 'stored:https://d1anwqy6ci9o1i.cloudfront.net/' + + 'users%2F366%2Ficons%2Fxqmypqvflgdy28aorw9ml_shock.png') + assert c.character + assert c.screen_name + assert c.author + self.assertEqual(list(c.character.stripped_strings), ['Keltham']) + self.assertEqual( + list(c.screen_name.stripped_strings), ['lawful chaotic']) + self.assertEqual(list(c.author.stripped_strings), ['Iarwain']) + self.assertEqual(str(c.content), + '

') + + self.assertEqual( + PandocTexifier("pandoc").texify(c.character), b'{Keltham}\n') + + +class TestThread(unittest.TestCase): + + def testTitle(self) -> None: + t = Thread(BeautifulSoup(b''' + +
+ + Teh Story! + +
+
The "post"
+ ''', 'html.parser')) + self.assertEqual(t.title(), 'Teh Story!') + + if __name__ == '__main__': unittest.main()