import unittest
import threading
from http.server import BaseHTTPRequestHandler, HTTPServer
+import requests
import paperdoorknob
TIMEOUT = 8
class FakeGlowficHTTPRequestHandler(BaseHTTPRequestHandler):
+ def _response_code(self) -> int:
+ if self.path == "/not_found":
+ return 404
+ if self.path == "/server_error":
+ return 500
+ return 200
+
def do_GET(self) -> None:
body = b'<html><body>This is glowfic</body></html>'
- self.send_response(200)
+ self.send_response(self._response_code())
self.send_header("Content-type", "text/html")
self.send_header("Content-Length", str(len(body)))
self.end_headers()
def testFetch(self) -> None:
paperdoorknob.fetch(f"http://localhost:{self._port}", TIMEOUT)
+ def testFetchErrors(self) -> None:
+ with self.assertRaises(requests.HTTPError):
+ paperdoorknob.fetch(
+ f"http://localhost:{self._port}/not_found", TIMEOUT)
+ with self.assertRaises(requests.HTTPError):
+ paperdoorknob.fetch(
+ f"http://localhost:{self._port}/server_error", TIMEOUT)
+
if __name__ == '__main__':
unittest.main()