[paperdoorknob] / paperdoorknob_test.py

# paperdoorknob: Print glowfic
#
# This program is free software: you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the
# Free Software Foundation, version 3.


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(self._response_code())
        self.send_header("Content-type", "text/html")
        self.send_header("Content-Length", str(len(body)))
        self.end_headers()
        self.wfile.write(body)


class TestFetch(unittest.TestCase):
    def setUp(self) -> None:
        web_server = HTTPServer(('', 0), FakeGlowficHTTPRequestHandler)
        self._port = web_server.socket.getsockname()[1]
        threading.Thread(target=web_server.serve_forever).start()
        self._stop_server = web_server.shutdown

    def tearDown(self) -> None:
        self._stop_server()

    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()
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
b25a2f90 SW	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:
b25a2f90 SW	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)
680f7d10 SW	38	self._port = web_server.socket.getsockname()[1]
b25a2f90 SW	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:
680f7d10	46	paperdoorknob.fetch(f"http://localhost:{self._port}", TIMEOUT)
b25a2f90	47
6fdb8f01 SW	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
b25a2f90 SW	56
	57	if __name__ == '__main__':
	58	unittest.main()