]>
Commit | Line | Data |
---|---|---|
e6adf6ce SW |
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 | ||
aa060d9b | 12 | from images import FakeImageStore |
94027099 | 13 | from glowfic import makeChunk, Thread |
62043b2b | 14 | from texify import PandocTexifier |
e6adf6ce SW |
15 | |
16 | ||
17 | class TestSplit(unittest.TestCase): | |
18 | ||
19 | def testSplit1(self) -> None: | |
94027099 | 20 | t = Thread(BeautifulSoup(b''' |
e6adf6ce SW |
21 | <html><body><div class="post-container post-post"> |
22 | The "post" | |
94027099 SW |
23 | </div></body></html>''', 'html.parser')) |
24 | self.assertEqual([list(t.stripped_strings) for t in t.chunkDOMs()], | |
e6adf6ce SW |
25 | [['The "post"']]) |
26 | ||
27 | def testSplit2(self) -> None: | |
94027099 | 28 | t = Thread(BeautifulSoup(b''' |
e6adf6ce SW |
29 | <html><body> |
30 | <div class="post-container post-post">The "post"</div> | |
31 | <div class="flat-post-replies"> | |
32 | <div class="post-container post-reply">The "reply"</div> | |
33 | </div> | |
94027099 SW |
34 | </body></html>''', 'html.parser')) |
35 | self.assertEqual([list(t.stripped_strings) for t in t.chunkDOMs()], | |
e6adf6ce SW |
36 | [['The "post"'], ['The "reply"']]) |
37 | ||
38 | def testSplit3(self) -> None: | |
94027099 | 39 | t = Thread(BeautifulSoup(b''' |
e6adf6ce SW |
40 | <html><body> |
41 | <div class="post-container post-post">The "post"</div> | |
42 | <div class="flat-post-replies"> | |
43 | <div class="post-container post-reply">1st reply</div> | |
44 | <div class="post-container post-reply">2nd reply</div> | |
45 | </div> | |
94027099 SW |
46 | </body></html>''', 'html.parser')) |
47 | self.assertEqual([list(t.stripped_strings) for t in t.chunkDOMs()], | |
e6adf6ce SW |
48 | [['The "post"'], ['1st reply'], ['2nd reply']]) |
49 | ||
50 | ||
aa060d9b SW |
51 | class TestMakeChunk(unittest.TestCase): |
52 | ||
53 | def testEmptyContent(self) -> None: | |
54 | with open('testdata/empty-content.html', 'rb') as f: | |
94027099 SW |
55 | t = Thread(BeautifulSoup(f, 'html.parser')) |
56 | c = makeChunk(next(iter(t.chunkDOMs())), FakeImageStore()) | |
aa060d9b SW |
57 | self.assertEqual( |
58 | c.icon, | |
59 | 'stored:https://d1anwqy6ci9o1i.cloudfront.net/' + | |
60 | 'users%2F366%2Ficons%2Fxqmypqvflgdy28aorw9ml_shock.png') | |
37c47bc2 SW |
61 | assert c.character |
62 | assert c.screen_name | |
63 | assert c.author | |
64 | self.assertEqual(list(c.character.stripped_strings), ['Keltham']) | |
65 | self.assertEqual( | |
66 | list(c.screen_name.stripped_strings), ['lawful chaotic']) | |
67 | self.assertEqual(list(c.author.stripped_strings), ['Iarwain']) | |
aa060d9b SW |
68 | self.assertEqual(str(c.content), |
69 | '<div class="post-content"><p></p></div>') | |
70 | ||
62043b2b SW |
71 | self.assertEqual( |
72 | PandocTexifier("pandoc").texify(c.character), b'{Keltham}\n') | |
73 | ||
aa060d9b | 74 | |
21e82200 SW |
75 | class TestThread(unittest.TestCase): |
76 | ||
77 | def testTitle(self) -> None: | |
78 | t = Thread(BeautifulSoup(b''' | |
79 | <html><body> | |
80 | <div class="content-header"> | |
81 | <span id="post-title"> | |
82 | <a href="/posts/1234">Teh Story!</a> | |
83 | </span> | |
84 | </div> | |
85 | <div class="post-container post-post">The "post"</div> | |
86 | </body></html>''', 'html.parser')) | |
87 | self.assertEqual(t.title(), 'Teh Story!') | |
88 | ||
89 | ||
e6adf6ce SW |
90 | if __name__ == '__main__': |
91 | unittest.main() |