]> git.scottworley.com Git - paperdoorknob/blob - paperdoorknob_test.py
fafd1c37284c80486f0859aa33c39e0e2728153d
[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 threading.Thread(target=web_server.serve_forever).start()
40 self._stop_server = web_server.shutdown
41
42 def tearDown(self) -> None:
43 self._stop_server()
44
45 def testFetch(self) -> None:
46 paperdoorknob.fetch(f"http://localhost:{self._port}", TIMEOUT)
47
48 def testFetchErrors(self) -> None:
49 with self.assertRaises(requests.HTTPError):
50 paperdoorknob.fetch(
51 f"http://localhost:{self._port}/not_found", TIMEOUT)
52 with self.assertRaises(requests.HTTPError):
53 paperdoorknob.fetch(
54 f"http://localhost:{self._port}/server_error", TIMEOUT)
55
56
57 if __name__ == '__main__':
58 unittest.main()