Python + gevent模块对单个接口进行并发测试 1

本文知识点
利用gevent模块进行并发测试

代码如下

 from gevent import monkey

 monkey.patch_all()

 import requests

 import gevent

 import sys, io

 # 解决console显示乱码的编码问题

 sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8')

 class Douban(object):

     """A class containing interface test method of Douban object"""

     def __init__(self):

         self.host = 'movie.douban.com'

         self.headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:61.0) Gecko/20100101 Firefox/61.0',

         'Referer':'https://movie.douban.com/',

         }

     def get_response(self, url, data):

         resp = requests.post(url=url, data=data, headers=self.headers).content.decode('utf-8')

         return resp

     def test_search_tags_movie(self):

         method = 'search_tags'

         url = 'https://%s/j/%s' % (self.host, method)

         post_data = {

             'type':'movie',

             'source':'index'

         }

         resp = self.get_response(url=url, data=post_data)

         print(resp)

         return resp

 if __name__ == '__main__':

     douban = Douban()

     threads = []

     for i in range(6):

         thread = gevent.spawn(douban.test_search_tags_movie)

         threads.append(thread)

     gevent.joinall(threads)

代码运行结果

参考
http://sdiehl.github.io/gevent-tutorial/