python自动化脚本实例100条-Python实现自动化压力测试脚本

#coding=utf-8

import sys

import time

import thread

import httplib, urllib

import random

import uuid

import logging

logging.basicConfig(level=logging.DEBUG,

format='%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s',

datefmt='%a, %d %b %Y %H:%M:%S',

filename='测试脚本日志.log',

filemode='w')

def log_uncaught_exceptions(exception_type, exception, tb):

logging.critical(''.join(traceback.format_tb(tb)))

logging.critical('{0}: {1}'.format(exception_type, exception))

sys.excepthook = log_uncaught_exceptions

#网关地址

addr="172.18.2.4"

port=8080

thread_count = 15 #单次并发数量

requst_interval = 10 #请求间隔(秒)

test_count = sys.maxsize #sys.maxsize # 指定测试次数

#字段说明,必须一一对应

#login为空表示使用随机用户名

param_list=[

{"login":"user1","password":"qweqwe12"},

]

now_count = 0

lock_obj = thread.allocate()

def send_http():

global now_count

httpClient = None

try:

for user in user_list:

tmp_user = user["login"]

if tmp_user.strip() =='':

tmp_user = str(uuid.uuid1()) + str(random.random())

print tmp_user

params = urllib.urlencode({"operationData":

[{"login": tmp_user,"password":user["password"]}]})

headers = {"Content-type": "application/x-www-form-urlencoded", "Accept": "text/plain"}

httpClient = httplib.HTTPConnection(addr, port, timeout=5)

httpClient.request("POST", "/simple/spider.task.distribute", params, headers)

response = httpClient.getresponse()

print '发送数据: ' + params

print '返回码: ' + str(response.status)

print '返回数据: ' + response.read()

logging.info('发送数据: ' + params)

logging.info('返回码: ' + str(response.status))

logging.info('返回数据: ' + response.read())

#print response.getheaders() #获取头信息

sys.stdout.flush()

now_count+=1

except Exception, e:

print e

logging.info(e)

finally:

if httpClient:

httpClient.close()

def test_func(run_count):

global now_count

global requst_interval

global lock_obj

cnt = 0

while cnt < run_count:

lock_obj.acquire()

print ''

print '***************************请求次数:' + str(now_count) + '*******************************'

print 'Thread:(%d) Time:%s '%(thread.get_ident(), time.ctime())

logging.info(' ')

logging.info('***************************请求次数:' + str(now_count) + '*******************************')

logging.info('Thread:(%d) Time:%s '%(thread.get_ident(), time.ctime()))

cnt+=1

send_http()

sys.stdout.flush()

lock_obj.release()

time.sleep(requst_interval)

def test(ct):

global thread_count

for i in range(thread_count):

thread.start_new_thread(test_func,(ct,))

if __name__=='__main__':

global test_count

test(test_count)

while True:

time.sleep(100)