]> git.scottworley.com Git - paperdoorknob/blob - paperdoorknob_test.py
b1d41d493574103bc5b22696d72454a43c33d07e
[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 requests
12 import paperdoorknob
13
14 TIMEOUT = 8
15
16
17 class FakeGlowficHTTPRequestHandler(BaseHTTPRequestHandler):
18
19 def _response_code(self) -> int:
20 if self.path == "/not_found":
21 return 404
22 if self.path == "/server_error":
23 return 500
24 return 200
25
26 def do_GET(self) -> None:
27 body = b'<html><body>This is glowfic</body></html>'
28 self.send_response(self._response_code())
29 self.send_header("Content-type", "text/html")
30 self.send_header("Content-Length", str(len(body)))
31 self.end_headers()
32 self.wfile.write(body)
33
34
35 class TestFetch(unittest.TestCase):
36 def setUp(self) -> None:
37 web_server = HTTPServer(('', 0), FakeGlowficHTTPRequestHandler)
38 self._port = web_server.socket.getsockname()[1]
39 self._thread = threading.Thread(target=web_server.serve_forever)
40 self._thread.start()
41 self._stop_server = web_server.shutdown
42
43 def tearDown(self) -> None:
44 self._stop_server()
45 self._thread.join()
46
47 def testFetch(self) -> None:
48 paperdoorknob.fetch(f"http://localhost:{self._port}", TIMEOUT)
49
50 def testFetchErrors(self) -> None:
51 with self.assertRaises(requests.HTTPError):
52 paperdoorknob.fetch(
53 f"http://localhost:{self._port}/not_found", TIMEOUT)
54 with self.assertRaises(requests.HTTPError):
55 paperdoorknob.fetch(
56 f"http://localhost:{self._port}/server_error", TIMEOUT)
57
58
59 if __name__ == '__main__':
60 unittest.main()