]>
Commit | Line | Data |
---|---|---|
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 | from io import BytesIO | |
8 | from sys import stderr | |
9 | import unittest | |
10 | ||
11 | from typing import Optional | |
12 | ||
13 | from fetch import FakeFetcher | |
14 | from images import FakeImageStore | |
15 | from glowfic import makeChunk, Thread | |
16 | from spec import Spec | |
17 | from texify import PandocTexifier | |
18 | ||
19 | ||
20 | def spec_for_testing(html: bytes, outbuf: Optional[BytesIO] = None) -> Spec: | |
21 | return Spec('https://fake/test', | |
22 | FakeFetcher({'https://fake/test': html, | |
23 | 'https://fake/test?view=flat': html}), | |
24 | FakeImageStore(), | |
25 | lambda x: x, | |
26 | lambda x: None, | |
27 | PandocTexifier('pandoc'), | |
28 | lambda x: x, | |
29 | 20, | |
30 | b'', | |
31 | None, | |
32 | stderr.buffer if outbuf is None else outbuf, | |
33 | lambda x: None) | |
34 | ||
35 | ||
36 | class TestSplit(unittest.TestCase): | |
37 | ||
38 | def testSplit1(self) -> None: | |
39 | t = Thread(spec_for_testing(b''' | |
40 | <html><body><div class="post-container post-post"> | |
41 | The "post" | |
42 | </div></body></html>''')) | |
43 | self.assertEqual([list(t.stripped_strings) for t in t.chunkDOMs()], | |
44 | [['The "post"']]) | |
45 | ||
46 | def testSplit2(self) -> None: | |
47 | t = Thread(spec_for_testing(b''' | |
48 | <html><body> | |
49 | <div class="post-container post-post">The "post"</div> | |
50 | <div class="flat-post-replies"> | |
51 | <div class="post-container post-reply">The "reply"</div> | |
52 | </div> | |
53 | </body></html>''')) | |
54 | self.assertEqual([list(t.stripped_strings) for t in t.chunkDOMs()], | |
55 | [['The "post"'], ['The "reply"']]) | |
56 | ||
57 | def testSplit3(self) -> None: | |
58 | t = Thread(spec_for_testing(b''' | |
59 | <html><body> | |
60 | <div class="post-container post-post">The "post"</div> | |
61 | <div class="flat-post-replies"> | |
62 | <div class="post-container post-reply">1st reply</div> | |
63 | <div class="post-container post-reply">2nd reply</div> | |
64 | </div> | |
65 | </body></html>''')) | |
66 | self.assertEqual([list(t.stripped_strings) for t in t.chunkDOMs()], | |
67 | [['The "post"'], ['1st reply'], ['2nd reply']]) | |
68 | ||
69 | ||
70 | class TestMakeChunk(unittest.TestCase): | |
71 | ||
72 | def testEmptyContent(self) -> None: | |
73 | with open('testdata/empty-content.html', 'rb') as f: | |
74 | t = Thread(spec_for_testing(f.read((9999)))) | |
75 | c = makeChunk(next(iter(t.chunkDOMs())), FakeImageStore()) | |
76 | self.assertEqual( | |
77 | c.icon, | |
78 | 'stored:https://d1anwqy6ci9o1i.cloudfront.net/' + | |
79 | 'users%2F366%2Ficons%2Fxqmypqvflgdy28aorw9ml_shock.png') | |
80 | assert c.character | |
81 | assert c.screen_name | |
82 | assert c.author | |
83 | self.assertEqual(list(c.character.stripped_strings), ['Keltham']) | |
84 | self.assertEqual( | |
85 | list(c.screen_name.stripped_strings), ['lawful chaotic']) | |
86 | self.assertEqual(list(c.author.stripped_strings), ['Iarwain']) | |
87 | self.assertEqual(str(c.content), | |
88 | '<div class="post-content"><p></p></div>') | |
89 | ||
90 | self.assertEqual( | |
91 | PandocTexifier("pandoc").texify(c.character), b'{Keltham}\n') | |
92 | ||
93 | def testMinimal(self) -> None: | |
94 | t = Thread(spec_for_testing(b''' | |
95 | <html><body> | |
96 | <div class="post-container post-post"> | |
97 | <div class="post-content">Just content</div> | |
98 | </div> | |
99 | </body></html>''')) | |
100 | c = makeChunk(next(iter(t.chunkDOMs())), FakeImageStore()) | |
101 | self.assertIsNone(c.icon) | |
102 | self.assertIsNone(c.character) | |
103 | self.assertIsNone(c.screen_name) | |
104 | self.assertIsNone(c.author) | |
105 | self.assertEqual(str(c.content), | |
106 | '<div class="post-content">Just content</div>') | |
107 | ||
108 | ||
109 | class TestThread(unittest.TestCase): | |
110 | ||
111 | def testTitle(self) -> None: | |
112 | t = Thread(spec_for_testing(b''' | |
113 | <html><body> | |
114 | <div class="content-header"> | |
115 | <span id="post-title"> | |
116 | <a href="/posts/1234">Teh Story!</a> | |
117 | </span> | |
118 | </div> | |
119 | <div class="post-container post-post">The "post"</div> | |
120 | </body></html>''')) | |
121 | self.assertEqual(t.title(), 'Teh Story!') | |
122 | ||
123 | def testNextThreadRelative(self) -> None: | |
124 | t = Thread(spec_for_testing(b''' | |
125 | <html><body> | |
126 | <div class="post-navheader"> | |
127 | <a class="view-button-link" href="/posts/4567"><div class="view-button">Next Post »</div> | |
128 | </a><a class="view-button-link" href="/posts/4321"><div class="view-button float-none">« Previous Post</div> | |
129 | </a> | |
130 | </div> | |
131 | <div class="post-container post-post">The "post"</div> | |
132 | </body></html>''')) | |
133 | self.assertEqual(t.next_thread(), 'https://fake/posts/4567') | |
134 | ||
135 | def testNextThreadAbsolute(self) -> None: | |
136 | t = Thread(spec_for_testing(b''' | |
137 | <html><body> | |
138 | <div class="post-navheader"> | |
139 | <a class="view-button-link" href="https://elsewhere/posts/4567"><div class="view-button">Next Post »</div> | |
140 | </a><a class="view-button-link" href="https://elsewhere/posts/4321"><div class="view-button float-none">« Previous Post</div> | |
141 | </a> | |
142 | </div> | |
143 | <div class="post-container post-post">The "post"</div> | |
144 | </body></html>''')) | |
145 | self.assertEqual(t.next_thread(), 'https://elsewhere/posts/4567') | |
146 | ||
147 | ||
148 | class TestEmit(unittest.TestCase): | |
149 | ||
150 | def testEmit(self) -> None: | |
151 | buf = BytesIO() | |
152 | Thread(spec_for_testing(b''' | |
153 | <html><body> | |
154 | <div class="post-container post-post"> | |
155 | <div class="post-content">A</div> | |
156 | </div> | |
157 | <div class="flat-post-replies"> | |
158 | <div class="post-container post-reply"> | |
159 | <div class="post-content">B</div> | |
160 | </div> | |
161 | </div> | |
162 | </body></html>''', buf)).emit() | |
163 | self.assertEqual(buf.getvalue(), rb'''\glowhead{}{}{}{}A | |
164 | \glowhead{}{}{}{}B | |
165 | ''') | |
166 | ||
167 | ||
168 | if __name__ == '__main__': | |
169 | unittest.main() |