python mock 脚本

巴黎的灯光下

rt，mock的工具有很多，但是我觉得还是不如自己写脚本mock随意，方便。

1. #! usr/bin/python #coding=utf-8
   
2. import BaseHTTPServer  
   
3. import urlparse  
   
4. import time
   
5. from SocketServer import ThreadingMixIn
   
6. import threading
   
7. import os
   
8. 
   
9. class WebRequestHandler(BaseHTTPServer.BaseHTTPRequestHandler):  
   
10. 
    
11.     def do_POST(self):
    
12. 
    
13.         print 'post message'
    
14.         parsed_path = urlparse.urlparse(self.path)
    
15.         paramstr = parsed_path.query
    
16.         path = parsed_path.path
    
17.         print paramstr
    
18.         print path
    
19.         for param in paramstr.split('&'):
    
20.             print param
    
21. 
    
22.         length = self.headers.getheader('content-length');
    
23.         nbytes = int(length)
    
24.         data = self.rfile.read(nbytes)
    
25.         cur_thread = threading.currentThread()
    
26.         print 'Thread:%s\tdata:%s' % (cur_thread.getName(), data)
    
27. 
    
28.         message_parts = [ 'just a test']  
    
29.         message = '\r\n'.join(message_parts)  
    
30.         self.send_response(200)  
    
31.         self.end_headers()  
    
32.         self.wfile.write(message)  
    
33. 
    
34.     def do_GET(self):
    
35. 
    
36.         print 'get message'
    
37.         parsed_path = urlparse.urlparse(self.path)
    
38.         paramstr = parsed_path.query
    
39.         path = parsed_path.path
    
40.         print paramstr
    
41.         print path
    
42.         for param in paramstr.split('&'):
    
43.             print param
    
44. 
    
45.         if path == self.getShopLevelURL:
    
46.             buf = self.getShopLevelResponseBody
    
47.         else:
    
48.             buf = 'it works'
    
49.             self.send_header("Welcome", "Contect")
    
50. 
    
51.         self.protocal_version = 'HTTP/1.1'   
    
52.         self.send_response(200)  
    
53.         self.end_headers()  
    
54.         self.wfile.write(buf)
    
55. 
    
56.     '''获取店铺评级接口'''
    
57.     getShopLevelURL = '/abtest'
    
58.     getShopLevelResponseBody = '''
    
59.         {
    
60.             "status": 1,
    
61.             "info": "success",
    
62.             "data": {
    
63.                 "input": {
    
64.                     "phone_number": "15168329206",
    
65.                     "cookie": null,
    
66.                     "id": "1234567",
    
67.                     "session_id": null,
    
68.                     "active_id": null
    
69.                 },
    
70.                 "abt_type": "b",
    
71.                 "abt_name": "abt_001"
    
72.             }
    
73.         }'''
    
74. 
    
75. class ThreadingHttpServer(ThreadingMixIn, BaseHTTPServer.HTTPServer):
    
76.     pass
    
77. 
    
78. if __name__ == '__main__':
    
79.     '''server = BaseHTTPServer.HTTPServer(('0.0.0.0',18360), WebRequestHandler) '''
    
80.     server = ThreadingHttpServer(('0.0.0.0',18360), WebRequestHandler)
    
81.     ip, port = server.server_address
    
82.     '''Start a thread with the server -- that thread will then start one '''
    
83.     '''more thread for each request'''
    
84.     server_thread = threading.Thread(target=server.serve_forever)
    
85.     '''Exit the server thread when the main thread terminates'''
    
86.     server_thread.setDaemon(True)
    
87.     server_thread.start()
    
88. 
    
89.     print "Server loop running in thread:", server_thread.getName()
    
90.     while True:
    
91.        pass

复制代码

八戒你干嘛

太直接了。。。先说明一下遇到什么痛点，为何市面上的 mock 都不合适，然后自己为了解决这个痛点采用什么方案，最后再贴代码。这样大伙比较容易理解。

悠悠小仙仙

我感觉不错，用什么工具不重要，思路正确就可以，这段代码活很清楚，自己用的惯就好。

乐哈哈yoyo

我建议改用flask实现，代码量少的很