[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 paperdoorknob

TIMEOUT = 8


class FakeGlowficHTTPRequestHandler(BaseHTTPRequestHandler):

    def do_GET(self) -> None:
        body = b'<html><body>This is glowfic</body></html>'
        self.send_response(200)
        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)


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
	11	import paperdoorknob
	12
b25a2f90 SW	13	TIMEOUT = 8
	14
	15
	16	class FakeGlowficHTTPRequestHandler(BaseHTTPRequestHandler):
	17
	18	def do_GET(self) -> None:
	19	body = b'<html><body>This is glowfic</body></html>'
	20	self.send_response(200)
	21	self.send_header("Content-type", "text/html")
	22	self.send_header("Content-Length", str(len(body)))
	23	self.end_headers()
	24	self.wfile.write(body)
	25
	26
	27	class TestFetch(unittest.TestCase):
	28	def setUp(self) -> None:
680f7d10 SW	29	web_server = HTTPServer(('', 0), FakeGlowficHTTPRequestHandler)
680f7d10 SW	30	self._port = web_server.socket.getsockname()[1]
b25a2f90 SW	31	threading.Thread(target=web_server.serve_forever).start()
	32	self._stop_server = web_server.shutdown
	33
	34	def tearDown(self) -> None:
	35	self._stop_server()
	36
	37	def testFetch(self) -> None:
680f7d10	38	paperdoorknob.fetch(f"http://localhost:{self._port}", TIMEOUT)
b25a2f90 SW	39
	40
	41	if __name__ == '__main__':
	42	unittest.main()