2424import ssl
2525
2626try :
27- from http .server import HTTPServer , BaseHTTPRequestHandler
27+ from http .server import HTTPServer
2828 from http .server import SimpleHTTPRequestHandler
2929except ImportError :
30- from BaseHTTPServer import HTTPServer , BaseHTTPRequestHandler
30+ from BaseHTTPServer import HTTPServer
3131 from SimpleHTTPServer import SimpleHTTPRequestHandler
3232
3333from threading import Thread
@@ -101,7 +101,8 @@ def _handle_request(handler):
101101 authorization = handler .headers .get ('Authorization' )
102102 if authorization != "Bearer %s" % self .TOKEN :
103103 handler .send_response (401 )
104- handler .wfile .write ('' )
104+ handler .end_headers ()
105+ handler .wfile .write ('401 - Unauthorized' .encode ('utf-8' ))
105106 else :
106107 sleep (delay )
107108 handler .send_response (code )
@@ -116,7 +117,7 @@ def _handle_request(handler):
116117 ] = _handle_request
117118
118119 def set_json_response (self , path , code , body ):
119- def _handle_request (handler ):
120+ def _handle_request (handler ):
120121 handler .send_response (code )
121122 handler .send_header ('Content-Type' , 'application/json' )
122123 handler .end_headers ()
@@ -128,10 +129,13 @@ def _handle_request(handler):
128129
129130 def start_server (self , host = 'localhost' ):
130131 self ._httpd = HTTPServer ((self .host (), self .port ()), TestHandler )
131- self ._httpd .socket = ssl .wrap_socket (
132- self ._httpd .socket ,
133- keyfile = self .__absolute_path ('%s.key' % host ),
132+ context = ssl .SSLContext (ssl .PROTOCOL_TLS_SERVER )
133+ context .load_cert_chain (
134134 certfile = self .__absolute_path ('%s.crt' % host ),
135+ keyfile = self .__absolute_path ('%s.key' % host )
136+ )
137+ self ._httpd .socket = context .wrap_socket (
138+ self ._httpd .socket ,
135139 server_side = True
136140 )
137141 # Path handler for username/password authentication service:
@@ -168,7 +172,7 @@ def port(self):
168172 try :
169173 server = HTTPServer (
170174 (self .host (), port ),
171- BaseHTTPRequestHandler
175+ TestHandler
172176 )
173177 self .PORT = port
174178 break
0 commit comments