git.scottworley.com Git - paperdoorknob/blob - paperdoorknob_test.py

   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 import io
  10 import subprocess
  11 import paperdoorknob
  12 from testing.fakeserver import FakeGlowficServer
  13 from fetch import DirectFetcher
  14 from texify import PandocTexifier
  15
  16 TIMEOUT = 8
  17
  18
  19 class TestPaperDoorknob(unittest.TestCase):
  20     def setUp(self) -> None:
  21         self._server = self.enterContext(FakeGlowficServer())
  22         self._port = self._server.port()
  23
  24     def testReplies(self) -> None:
  25         with DirectFetcher(TIMEOUT) as f:
  26             replies = paperdoorknob.replies(
  27                 paperdoorknob.clean(
  28                     paperdoorknob.parse(
  29                         f.fetch(f"http://localhost:{self._port}"))))
  30             self.assertEqual([r.text.strip() for r in replies],
  31                              ["This is glowfic", "You sure?", "Pretty sure."])
  32
  33     def testProcess(self) -> None:
  34         texifier = PandocTexifier('pandoc')
  35         with DirectFetcher(TIMEOUT) as f:
  36             buf = io.BytesIO()
  37             paperdoorknob.process(
  38                 f"http://localhost:{self._port}", f, texifier, buf)
  39             self.assertEqual(buf.getvalue(), b'''\\documentclass{article}
  40 \\begin{document}
  41 This is glowfic
  42 You \\emph{sure}?
  43 Pretty sure.
  44 \\end{document}
  45 ''')
  46
  47     def testPDF(self) -> None:
  48         texifier = PandocTexifier('pandoc')
  49         with DirectFetcher(TIMEOUT) as f:
  50             with open("test.tex", 'wb') as out:
  51                 paperdoorknob.process(
  52                     f"http://localhost:{self._port}", f, texifier, out)
  53             subprocess.run(['pdflatex', 'test.tex'],
  54                            stdin=subprocess.DEVNULL, check=True)
  55
  56
  57 if __name__ == '__main__':
  58     unittest.main()