]> git.scottworley.com Git - paperdoorknob/blame - paperdoorknob_test.py
fetch: Send User-Agent header
[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
07f9b178 11import subprocess
23f31879 12
b25a2f90 13import paperdoorknob
23f31879 14
41b11505 15from testing.fakeserver import FakeGlowficServer
8be20b9d 16from domfilter import ApplyDOMFilters
38621839 17from fetch import DirectFetcher, FakeFetcher, Fetcher
d2a41ff4 18from glowfic import ContentOnlyLayout, BelowIconLayout
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,
8be20b9d 43 lambda x: ApplyDOMFilters('NoEdit,NoFooter', x),
d2a41ff4 44 ContentOnlyLayout(PandocTexifier('pandoc')),
e10b5b6f 45 'margin=20mm',
38621839
SW
46 buf)
47 paperdoorknob.process(spec)
48 assert buf.getvalue() == b'''\\documentclass{article}
d2a41ff4 49\\usepackage{graphicx}
23dabdf5 50\\usepackage{varwidth}
357f37be 51\\usepackage{wrapstuff}
e10b5b6f 52\\usepackage[margin=20mm]{geometry}
07f9b178
SW
53\\begin{document}
54This is glowfic
85bcacb0 55
07f9b178 56You \\emph{sure}?
85bcacb0 57
07f9b178 58Pretty sure.
23dabdf5 59
07f9b178 60\\end{document}
38621839 61'''
07f9b178 62
f1dec720
SW
63 def testDirectTexifier(self) -> None:
64 texifier = VerifyingTexifier(
65 PandocTexifier('pandoc'), DirectTexifier())
38621839 66 buf = io.BytesIO()
8be20b9d
SW
67 spec = Spec(
68 self.url(),
69 self.fetcher(),
4b7d905e 70 FakeImageStore(),
8be20b9d
SW
71 lambda x: x,
72 lambda x: ApplyDOMFilters('NoEdit,NoFooter', x),
d2a41ff4 73 ContentOnlyLayout(texifier),
e10b5b6f 74 None,
8be20b9d 75 buf)
38621839 76 paperdoorknob.process(spec)
f1dec720 77
07f9b178 78 def testPDF(self) -> None:
38621839
SW
79 with open("test.tex", 'wb') as out:
80 spec = Spec(
81 self.url(),
82 self.fetcher(),
4b7d905e 83 FakeImageStore(),
929db576 84 lambda x: x,
8be20b9d 85 lambda x: ApplyDOMFilters('NoEdit,NoFooter', x),
c62e8d40 86 BelowIconLayout(PandocTexifier('pandoc'), 20),
e10b5b6f 87 None,
38621839
SW
88 out)
89 paperdoorknob.process(spec)
23f31879
SW
90 subprocess.run(['pdflatex', 'test.tex'],
91 stdin=subprocess.DEVNULL, check=True)
36ae1d5f 92
b25a2f90 93
38621839
SW
94class TestProcessFromWebserver(BaseTestProcess, unittest.TestCase):
95
96 def setUp(self) -> None:
97 self._fetcher = self.enterContext(DirectFetcher(TIMEOUT))
98 self._server = self.enterContext(FakeGlowficServer())
99 self._port = self._server.port()
100
101 def url(self) -> str:
102 return f"http://localhost:{self._port}"
103
104 def fetcher(self) -> Fetcher:
105 return self._fetcher
106
107
108class TestProcessFromFakeFetcher(BaseTestProcess, unittest.TestCase):
109
110 def url(self) -> str:
111 return 'fic'
112
113 def fetcher(self) -> Fetcher:
114 with open('testdata/this-is-glowfic.html', 'rb') as f:
1452f8d3 115 return FakeFetcher({'fic?view=flat': f.read(9999)})
38621839
SW
116
117
b25a2f90
SW
118if __name__ == '__main__':
119 unittest.main()