]> git.scottworley.com Git - paperdoorknob/blob - paperdoorknob_test.py
2bd354419c113d8d6e321bb8395278879ea65dcc
[paperdoorknob] / 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 threading
10 from http.server import BaseHTTPRequestHandler, HTTPServer
11 import paperdoorknob
12
13 TIMEOUT = 8
14
15
16 class FakeGlowficHTTPRequestHandler(BaseHTTPRequestHandler):
17
18 def do_GET(self) -> None:
19 body = b'<html><body>This is glowfic</body></html>'
20 self.send_response(200)
21 self.send_header("Content-type", "text/html")
22 self.send_header("Content-Length", str(len(body)))
23 self.end_headers()
24 self.wfile.write(body)
25
26
27 class TestFetch(unittest.TestCase):
28 def setUp(self) -> None:
29 web_server = HTTPServer(('', 0), FakeGlowficHTTPRequestHandler)
30 self._port = web_server.socket.getsockname()[1]
31 threading.Thread(target=web_server.serve_forever).start()
32 self._stop_server = web_server.shutdown
33
34 def tearDown(self) -> None:
35 self._stop_server()
36
37 def testFetch(self) -> None:
38 paperdoorknob.fetch(f"http://localhost:{self._port}", TIMEOUT)
39
40
41 if __name__ == '__main__':
42 unittest.main()