[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

TEST_PORT = 8080
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(('', TEST_PORT), FakeGlowficHTTPRequestHandler)
        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:{TEST_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
	13	TEST_PORT = 8080
	14	TIMEOUT = 8
	15
	16
	17	class FakeGlowficHTTPRequestHandler(BaseHTTPRequestHandler):
	18
	19	def do_GET(self) -> None:
	20	body = b'<html><body>This is glowfic</body></html>'
	21	self.send_response(200)
	22	self.send_header("Content-type", "text/html")
	23	self.send_header("Content-Length", str(len(body)))
	24	self.end_headers()
	25	self.wfile.write(body)
	26
	27
	28	class TestFetch(unittest.TestCase):
	29	def setUp(self) -> None:
	30	web_server = HTTPServer(('', TEST_PORT), FakeGlowficHTTPRequestHandler)
	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:
	38	paperdoorknob.fetch(f"http://localhost:{TEST_PORT}", TIMEOUT)
	39
	40
	41	if __name__ == '__main__':
	42	unittest.main()