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'''
<html><body><div class="post-container post-post">
The "post"
- </div></body></html>''', 'html.parser')
- self.assertEqual([list(t.stripped_strings) for t in chunkDOMs(soup)],
+ </div></body></html>''', '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'''
<html><body>
<div class="post-container post-post">The "post"</div>
<div class="flat-post-replies">
<div class="post-container post-reply">The "reply"</div>
</div>
- </body></html>''', 'html.parser')
- self.assertEqual([list(t.stripped_strings) for t in chunkDOMs(soup)],
+ </body></html>''', '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'''
<html><body>
<div class="post-container post-post">The "post"</div>
<div class="flat-post-replies">
<div class="post-container post-reply">1st reply</div>
<div class="post-container post-reply">2nd reply</div>
</div>
- </body></html>''', 'html.parser')
- self.assertEqual([list(t.stripped_strings) for t in chunkDOMs(soup)],
+ </body></html>''', '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),
+ '<div class="post-content"><p></p></div>')
+
+ self.assertEqual(
+ PandocTexifier("pandoc").texify(c.character), b'{Keltham}\n')
+
+
+class TestThread(unittest.TestCase):
+
+ def testTitle(self) -> None:
+ t = Thread(BeautifulSoup(b'''
+ <html><body>
+ <div class="content-header">
+ <span id="post-title">
+ <a href="/posts/1234">Teh Story!</a>
+ </span>
+ </div>
+ <div class="post-container post-post">The "post"</div>
+ </body></html>''', 'html.parser'))
+ self.assertEqual(t.title(), 'Teh Story!')
+
+
if __name__ == '__main__':
unittest.main()
projects / paperdoorknob / blobdiff