projects / paperdoorknob / blame
| Commit | Line | Data |
|---|---|---|
| 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 | ||
| 8 | import unittest | |
| 9 | import threading | |
| 10 | from http.server import BaseHTTPRequestHandler, HTTPServer | |
| 6fdb8f01 | 11 | import requests |
| b25a2f90 SW |
12 | import paperdoorknob |
| 13 | ||
| b25a2f90 SW |
14 | TIMEOUT = 8 |
| 15 | ||
| 16 | ||
| 17 | class FakeGlowficHTTPRequestHandler(BaseHTTPRequestHandler): | |
| 18 | ||
| 6fdb8f01 SW |
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 | ||
| b25a2f90 SW |
26 | def do_GET(self) -> None: |
| 27 | body = b'<html><body>This is glowfic</body></html>' | |
| 6fdb8f01 | 28 | self.send_response(self._response_code()) |
| b25a2f90 SW |
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: | |
| 680f7d10 SW |
37 | web_server = HTTPServer(('', 0), FakeGlowficHTTPRequestHandler) |
| 38 | self._port = web_server.socket.getsockname()[1] | |
| e3ba2056 SW |
39 | self._thread = threading.Thread(target=web_server.serve_forever) |
| 40 | self._thread.start() | |
| b25a2f90 SW |
41 | self._stop_server = web_server.shutdown |
| 42 | ||
| 43 | def tearDown(self) -> None: | |
| 44 | self._stop_server() | |
| e3ba2056 | 45 | self._thread.join() |
| b25a2f90 SW |
46 | |
| 47 | def testFetch(self) -> None: | |
| 680f7d10 | 48 | paperdoorknob.fetch(f"http://localhost:{self._port}", TIMEOUT) |
| b25a2f90 | 49 | |
| 6fdb8f01 SW |
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 | ||
| b25a2f90 SW |
58 | |
| 59 | if __name__ == '__main__': | |
| 60 | unittest.main() |