]> git.scottworley.com Git - paperdoorknob/blame - paperdoorknob_test.py
Change default layout: below → beside
[paperdoorknob] / paperdoorknob_test.py
CommitLineData
b25a2f90
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
38621839 8from abc import ABC, abstractmethod
b25a2f90 9import unittest
36ae1d5f 10import io
4aa87092 11import re
07f9b178 12import subprocess
23f31879 13
b25a2f90 14import paperdoorknob
23f31879 15
41b11505 16from testing.fakeserver import FakeGlowficServer
38621839 17from fetch import DirectFetcher, FakeFetcher, Fetcher
37221176 18from glowfic import ContentOnlyLayout, BesideIconLayout
4b7d905e 19from images import FakeImageStore
23f31879 20from spec import Spec
f1dec720 21from texify import DirectTexifier, PandocTexifier, VerifyingTexifier
b25a2f90 22
b25a2f90
SW
23TIMEOUT = 8
24
25
38621839
SW
26class BaseTestProcess(ABC):
27
28 @abstractmethod
29 def url(self) -> str:
30 raise NotImplementedError()
31
32 @abstractmethod
33 def fetcher(self) -> Fetcher:
34 raise NotImplementedError()
b25a2f90 35
36ae1d5f 36 def testProcess(self) -> None:
38621839
SW
37 buf = io.BytesIO()
38 spec = Spec(
39 self.url(),
40 self.fetcher(),
4b7d905e 41 FakeImageStore(),
929db576 42 lambda x: x,
4640c55a 43 lambda x: None,
131deef1 44 lambda x: x,
d2a41ff4 45 ContentOnlyLayout(PandocTexifier('pandoc')),
e10b5b6f 46 'margin=20mm',
c594c8bf
SW
47 buf,
48 lambda _: None)
38621839 49 paperdoorknob.process(spec)
4aa87092
SW
50 assert re.match(br'''\\documentclass{article}
51(\\usepackage{[a-z]+}\n)+\\usepackage\[margin=20mm\]{geometry}
07f9b178
SW
52\\begin{document}
53This is glowfic
85bcacb0 54
4aa87092 55You \\emph{sure}\?
85bcacb0 56
07f9b178 57Pretty sure.
23dabdf5 58
07f9b178 59\\end{document}
4aa87092 60''', buf.getvalue())
07f9b178 61
f1dec720
SW
62 def testDirectTexifier(self) -> None:
63 texifier = VerifyingTexifier(
64 PandocTexifier('pandoc'), DirectTexifier())
38621839 65 buf = io.BytesIO()
8be20b9d
SW
66 spec = Spec(
67 self.url(),
68 self.fetcher(),
4b7d905e 69 FakeImageStore(),
8be20b9d 70 lambda x: x,
4640c55a 71 lambda x: None,
131deef1 72 lambda x: x,
d2a41ff4 73 ContentOnlyLayout(texifier),
e10b5b6f 74 None,
c594c8bf
SW
75 buf,
76 lambda _: None)
38621839 77 paperdoorknob.process(spec)
f1dec720 78
07f9b178 79 def testPDF(self) -> None:
38621839
SW
80 with open("test.tex", 'wb') as out:
81 spec = Spec(
82 self.url(),
83 self.fetcher(),
4b7d905e 84 FakeImageStore(),
929db576 85 lambda x: x,
4640c55a 86 lambda x: None,
131deef1 87 lambda x: x,
37221176 88 BesideIconLayout(PandocTexifier('pandoc'), 20),
e10b5b6f 89 None,
c594c8bf
SW
90 out,
91 lambda _: None)
38621839 92 paperdoorknob.process(spec)
23f31879
SW
93 subprocess.run(['pdflatex', 'test.tex'],
94 stdin=subprocess.DEVNULL, check=True)
36ae1d5f 95
b25a2f90 96
38621839
SW
97class TestProcessFromWebserver(BaseTestProcess, unittest.TestCase):
98
99 def setUp(self) -> None:
100 self._fetcher = self.enterContext(DirectFetcher(TIMEOUT))
101 self._server = self.enterContext(FakeGlowficServer())
102 self._port = self._server.port()
103
104 def url(self) -> str:
105 return f"http://localhost:{self._port}"
106
107 def fetcher(self) -> Fetcher:
108 return self._fetcher
109
110
111class TestProcessFromFakeFetcher(BaseTestProcess, unittest.TestCase):
112
113 def url(self) -> str:
114 return 'fic'
115
116 def fetcher(self) -> Fetcher:
117 with open('testdata/this-is-glowfic.html', 'rb') as f:
1452f8d3 118 return FakeFetcher({'fic?view=flat': f.read(9999)})
38621839
SW
119
120
b25a2f90
SW
121if __name__ == '__main__':
122 unittest.main()