-
Notifications
You must be signed in to change notification settings - Fork 21
Expand file tree
/
Copy pathtest_server.py
More file actions
76 lines (55 loc) · 2.09 KB
/
test_server.py
File metadata and controls
76 lines (55 loc) · 2.09 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
import unittest
import sys
import threading
import time
from io import BytesIO
try:
# python 2
from BaseHTTPServer import HTTPServer,BaseHTTPRequestHandler
except:
# python 3
from http.server import HTTPServer,BaseHTTPRequestHandler
import gzip
class TestServer(threading.Thread):
def __init__(self, port, delay = None, handler_func = None):
self.port = port
RequestHandler.delay = delay
RequestHandler.handler_func = [handler_func]
RequestHandler.response_data = '{}'
RequestHandler.response_code = 200
threading.Thread.__init__(self)
self.server = HTTPServer(('localhost', self.port), RequestHandler)
def get_request_data(self):
return RequestHandler.request_data
def set_response_data(self, response_data):
RequestHandler.response_data = response_data
def set_response_code(self, response_code):
RequestHandler.response_code = response_code
def run(self):
self.server.handle_request()
class RequestHandler(BaseHTTPRequestHandler):
delay = None
handler_func = None
request_data = None
response_data = '{}'
response_code = 200
def do_GET(self):
if self.delay:
time.sleep(self.delay)
if RequestHandler.handler_func:
RequestHandler.handler_func[0]()
self.send_response(RequestHandler.response_code)
self.send_header('Content-Type', 'application/json')
self.end_headers()
self.wfile.write(RequestHandler.response_data.encode('utf-8'))
def do_POST(self):
if self.delay:
time.sleep(self.delay)
self.request_url = self.path
content_len = int(self.headers.get('content-length'))
decompressed_data = gzip.GzipFile(fileobj=BytesIO(self.rfile.read(content_len))).read()
RequestHandler.request_data = decompressed_data.decode('utf-8')
self.send_response(RequestHandler.response_code)
self.send_header('Content-Type', 'application/json')
self.end_headers()
self.wfile.write(RequestHandler.response_data.encode('utf-8'))