]> git.scottworley.com Git - paperdoorknob/blob - glowfic_test.py
9b802c180cf3463e659f892948b0c9a64d4f936e
[paperdoorknob] / glowfic_test.py
1 # paperdoorknob: Print glowfic
2 #
3 # This program is free software: you can redistribute it and/or modify it
4 # under the terms of the GNU General Public License as published by the
5 # Free Software Foundation, version 3.
6
7
8 import unittest
9
10 from bs4 import BeautifulSoup
11
12 from images import FakeImageStore
13 from glowfic import chunkDOMs, makeChunk
14
15
16 class TestSplit(unittest.TestCase):
17
18 def testSplit1(self) -> None:
19 soup = BeautifulSoup(b'''
20 <html><body><div class="post-container post-post">
21 The "post"
22 </div></body></html>''', 'html.parser')
23 self.assertEqual([list(t.stripped_strings) for t in chunkDOMs(soup)],
24 [['The "post"']])
25
26 def testSplit2(self) -> None:
27 soup = BeautifulSoup(b'''
28 <html><body>
29 <div class="post-container post-post">The "post"</div>
30 <div class="flat-post-replies">
31 <div class="post-container post-reply">The "reply"</div>
32 </div>
33 </body></html>''', 'html.parser')
34 self.assertEqual([list(t.stripped_strings) for t in chunkDOMs(soup)],
35 [['The "post"'], ['The "reply"']])
36
37 def testSplit3(self) -> None:
38 soup = BeautifulSoup(b'''
39 <html><body>
40 <div class="post-container post-post">The "post"</div>
41 <div class="flat-post-replies">
42 <div class="post-container post-reply">1st reply</div>
43 <div class="post-container post-reply">2nd reply</div>
44 </div>
45 </body></html>''', 'html.parser')
46 self.assertEqual([list(t.stripped_strings) for t in chunkDOMs(soup)],
47 [['The "post"'], ['1st reply'], ['2nd reply']])
48
49
50 class TestMakeChunk(unittest.TestCase):
51
52 def testEmptyContent(self) -> None:
53 with open('testdata/empty-content.html', 'rb') as f:
54 soup = BeautifulSoup(f, 'html.parser')
55 c = makeChunk(next(iter(chunkDOMs(soup))), FakeImageStore())
56 self.assertEqual(
57 c.icon,
58 'stored:https://d1anwqy6ci9o1i.cloudfront.net/' +
59 'users%2F366%2Ficons%2Fxqmypqvflgdy28aorw9ml_shock.png')
60 assert c.character
61 assert c.screen_name
62 assert c.author
63 self.assertEqual(list(c.character.stripped_strings), ['Keltham'])
64 self.assertEqual(
65 list(c.screen_name.stripped_strings), ['lawful chaotic'])
66 self.assertEqual(list(c.author.stripped_strings), ['Iarwain'])
67 self.assertEqual(str(c.content),
68 '<div class="post-content"><p></p></div>')
69
70
71 if __name__ == '__main__':
72 unittest.main()