]>
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 | ||
8 | from abc import ABC, abstractmethod | |
9 | import unittest | |
10 | import io | |
11 | import subprocess | |
12 | ||
13 | import paperdoorknob | |
14 | ||
15 | from testing.fakeserver import FakeGlowficServer | |
16 | from fetch import DirectFetcher, FakeFetcher, Fetcher | |
17 | from spec import Spec | |
18 | from texify import DirectTexifier, PandocTexifier, VerifyingTexifier | |
19 | ||
20 | TIMEOUT = 8 | |
21 | ||
22 | ||
23 | class BaseTestProcess(ABC): | |
24 | ||
25 | @abstractmethod | |
26 | def url(self) -> str: | |
27 | raise NotImplementedError() | |
28 | ||
29 | @abstractmethod | |
30 | def fetcher(self) -> Fetcher: | |
31 | raise NotImplementedError() | |
32 | ||
33 | def testReplies(self) -> None: | |
34 | replies = paperdoorknob.replies( | |
35 | paperdoorknob.clean( | |
36 | paperdoorknob.parse( | |
37 | self.fetcher().fetch( | |
38 | self.url())))) | |
39 | assert [r.text.strip() for r in replies] == [ | |
40 | "This is glowfic", | |
41 | "You sure?", | |
42 | "Pretty sure."] | |
43 | ||
44 | def testProcess(self) -> None: | |
45 | buf = io.BytesIO() | |
46 | spec = Spec( | |
47 | self.url(), | |
48 | self.fetcher(), | |
49 | lambda x: x, | |
50 | PandocTexifier('pandoc'), | |
51 | buf) | |
52 | paperdoorknob.process(spec) | |
53 | assert buf.getvalue() == b'''\\documentclass{article} | |
54 | \\begin{document} | |
55 | This is glowfic | |
56 | You \\emph{sure}? | |
57 | Pretty sure. | |
58 | \\end{document} | |
59 | ''' | |
60 | ||
61 | def testDirectTexifier(self) -> None: | |
62 | texifier = VerifyingTexifier( | |
63 | PandocTexifier('pandoc'), DirectTexifier()) | |
64 | buf = io.BytesIO() | |
65 | spec = Spec(self.url(), self.fetcher(), lambda x: x, texifier, buf) | |
66 | paperdoorknob.process(spec) | |
67 | ||
68 | def testPDF(self) -> None: | |
69 | with open("test.tex", 'wb') as out: | |
70 | spec = Spec( | |
71 | self.url(), | |
72 | self.fetcher(), | |
73 | lambda x: x, | |
74 | PandocTexifier('pandoc'), | |
75 | out) | |
76 | paperdoorknob.process(spec) | |
77 | subprocess.run(['pdflatex', 'test.tex'], | |
78 | stdin=subprocess.DEVNULL, check=True) | |
79 | ||
80 | ||
81 | class TestProcessFromWebserver(BaseTestProcess, unittest.TestCase): | |
82 | ||
83 | def setUp(self) -> None: | |
84 | self._fetcher = self.enterContext(DirectFetcher(TIMEOUT)) | |
85 | self._server = self.enterContext(FakeGlowficServer()) | |
86 | self._port = self._server.port() | |
87 | ||
88 | def url(self) -> str: | |
89 | return f"http://localhost:{self._port}" | |
90 | ||
91 | def fetcher(self) -> Fetcher: | |
92 | return self._fetcher | |
93 | ||
94 | ||
95 | class TestProcessFromFakeFetcher(BaseTestProcess, unittest.TestCase): | |
96 | ||
97 | def url(self) -> str: | |
98 | return 'fic' | |
99 | ||
100 | def fetcher(self) -> Fetcher: | |
101 | with open('testdata/this-is-glowfic.html', 'rb') as f: | |
102 | return FakeFetcher({'fic': f.read(9999)}) | |
103 | ||
104 | ||
105 | if __name__ == '__main__': | |
106 | unittest.main() |