Repository navigation
Expand file tree
/
Copy pathtest_server.py
More file actions
70 lines (50 loc) · 1.83 KB
/
Copy pathtest_server.py
File metadata and controls
70 lines (50 loc) · 1.83 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
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]
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 run(self):
self.server.handle_request()
class RequestHandler(BaseHTTPRequestHandler):
delay = None
handler_func = None
request_data = None
response_data = '{}'
def do_GET(self):
if self.delay:
time.sleep(self.delay)
if RequestHandler.handler_func:
RequestHandler.handler_func[0]()
self.send_response(200)
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(200)
self.send_header('Content-Type', 'application/json')
self.end_headers()
self.wfile.write(RequestHandler.response_data.encode('utf-8'))