Python server性能测试 - REST request 模拟器 - 多线程并发-优快云博客

本文链接：https://blog.youkuaiyun.com/tidewang2k/article/details/42496915

该Python脚本用于模拟REST服务的删除和安装设备配置请求，使用多线程并发进行性能测试。通过初始化认证字符串、更新参数、发送REST请求等方法，对不同状态码的响应进行处理，并记录性能日志。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#! /usr/bin/python
#-*- coding: utf-8 -*-
#FileName: NotificationSimu.py
from BaseModule import BaseSimu
import urllib
import httplib
import base64
import json
import string
import ConfigParser
import multiprocessing
import uuid
import time
from datetime import datetime
import tmslog
import threading
import Queue
import sys

CFG_DATA = "test_cfg.txt"
DEVICES_DATA = "view_devices.txt"
SIMU_LOG = 'mdmi_simu_log.txt'
TEST_DATA = ConfigParser.ConfigParser()
TEST_DATA.read(CFG_DATA)

class NotificationSimu(BaseSimu):
    def __init__(self, flag=0, acct_index=0, url_tab=()):
        BaseSimu.__init__(self)
        auth_list = []
        auth = self._init_auth_str(self.username, self.password)
        auth_list.append(auth)
        auth = self._init_auth_str(self.username2, self.password)
        auth_list.append(auth)
        auth = self._init_auth_str(self.username3, self.password)
        auth_list.append(auth)
        auth = self._init_auth_str(self.username4, self.password)
        auth_list.append(auth)
        auth = self._init_auth_str(self.username, self.wrongpassword)
        auth_list.append(auth)
        self.auth_str_tab = tuple(auth_list)
        self.install_url_tab = url_tab
        pass

    def _init_auth_str(self, username, password):
        auth = base64.b64encode(username + ':' + password)
        auth = ' '.join(['Basic', auth])
        return auth
        pass

    def rest_removeprofile(self):
        """
        this method is used for simulating air-watch call to remove device profile to REST Service
        """
        params = {
                "RequestID": "123456789abcdef",
                "User": {
                      "Email": "xwang@websense.com"
                    },
                "Device": {
                      "Udid": "4B6D5E40DFB22BD57EC4AF54DA1F297A3221388B"
                    }
                }

        ns_url = TEST_DATA.get("Notification Server", "REST_RemoveProfile")
        response = self.execute_request('POST', ns_url, params, 0, 0)
        print response
        return response

    def rest_removeprofile_406(self):
        """
        this method is used for simulating air-watch call to remove device profile to REST Service
        """
        params = {
                "RequestID": "123456789abcdef",
                "User": {
                      "Email": "xwang@websense.com"
                    },
                "Device": {
                      "Udid": "4B6D5E40DFB22BD57EC4AF54DA1F297A3221388B"
                    }
                }

        ns_url = TEST_DATA.get("Notification Server", "REST_RemoveProfile")
        response = self.execute_request('POST', ns_url, params, 1, 0)
        print response
        return response

    def rest_removeprofile_401(self):
        """
        this method is used for simulating air-watch call to remove device profile to REST Service
        """
        params = {
                "RequestID": "123456789abcdef",
                "User": {
                      "Email": "xwang@websense.com"
                    },
                "Device": {
                      "Udid": "4B6D5E40DFB22BD57EC4AF54DA1F297A3221388B"
                    }
                }

        ns_url = TEST_DATA.get("Notification Server", "REST_RemoveProfile")
        response = self.execute_request('POST', ns_url, params, 0, 1)
        print response
        return response

    def update_params(self, device, params, acct_index):
        """
        this method is used to update the params payload with the device info.
        """
        device = device.split()
        print "%s - Account:%d " % (device, acct_index)
        request_id = base64.b64encode(uuid.uuid4().get_bytes())
        callback_url = "http://10.230.48.19:80/cgi-py/installvpn.py?requestid=%s" % request_id
        params['RequestID'] = request_id
        #params['OperationCallbackURL'] = callback_url
        #Update the params with the user info. got
        params['User']['UserName'] = device[9]
        params['User']['Group'] = device[6]
        params['User']['Email'] = device[10]
        #params['User']['LdapDN'] = 'mail=%s,%s' %(device[10], params['User']['LdapDN'])
        #params['User']['LdapGuid'] = base64.b64encode(uuid.uuid4().get_bytes())
        #Update the params with the deviec info. got
        para