| | |
|---|
| | | """ |
|---|
| | | import sys |
|---|
| | | import os |
|---|
| | | # 将上一级目录加入模块搜索路径 |
|---|
| | | sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) |
|---|
| | | import asyncio |
|---|
| | | import aiohttp |
|---|
| | | import time |
|---|
| | | import traceback |
|---|
| | | import datetime |
|---|
| | | from tqdm import tqdm |
|---|
| | | from Util.random_util import RandomUtil |
|---|
| | | from Util.dingtalk_helper import DingTalkHelper |
|---|
| | | import pymysql |
|---|
| | | import pymysql.cursors |
|---|
| | | import random |
|---|
| | | def get_parent_directory(file_path, levels=1): |
|---|
| | | """获取指定层级的父目录""" |
|---|
| | | path = os.path.abspath(file_path) |
|---|
| | | for _ in range(levels): |
|---|
| | | path = os.path.dirname(path) |
|---|
| | | return path |
|---|
| | | parent_dir = get_parent_directory(__file__, 5) # 获取上五级目录 |
|---|
| | | sys.path.append(parent_dir) |
|---|
| | | from 测试组.脚本.造数脚本2.Util import TokenValidator, DingTalkHelper, RandomUtil, RequestRecord, LoadTestReportGenerator |
|---|
| | | |
|---|
| | | |
|---|
| | | # --- 配置 --- |
|---|
| | |
|---|
| | | 'charset': 'utf8mb4', |
|---|
| | | 'cursorclass': pymysql.cursors.DictCursor |
|---|
| | | } |
|---|
| | | # 账号密码配置 |
|---|
| | | username = "gly" |
|---|
| | | password = "Baoyi@1341" |
|---|
| | | # 创建获取token实例 |
|---|
| | | token_validator = TokenValidator() |
|---|
| | | |
|---|
| | | domain = "http://192.168.6.190:5561" |
|---|
| | | token = token_validator.get_token(domain, username, password) |
|---|
| | | |
|---|
| | | apiname = "入驻笼位" |
|---|
| | | url = "http://192.168.6.190:5561/api/base/cage/cage/enterCage" |
|---|
| | | url = f"{domain}/api/base/cage/cage/enterCage" |
|---|
| | | headers = { |
|---|
| | | "token": "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJleHAiOjE3NjgwMzY0MzksInVzZXJuYW1lIjoiZ2x5In0.-LYYNbSJ-zb5RKaiBiPjntgUfnGRfvajA2B1N2v7a-o", |
|---|
| | | "token": token, |
|---|
| | | "Content-Type": "application/json" |
|---|
| | | } |
|---|
| | | |
|---|
| | | NUM_WORKERS = 100 |
|---|
| | | TOTAL_REQUESTS = 10000 |
|---|
| | | TOTAL_REQUESTS = 1 |
|---|
| | | MAX_RETRIES = 3 |
|---|
| | | REQUEST_TIMEOUT = 60 |
|---|
| | | OUTPUT_DIR = './load_test_report' |
|---|
| | |
|---|
| | | try: |
|---|
| | | import importlib.util |
|---|
| | | script_dir = os.path.dirname(os.path.abspath(__file__)) |
|---|
| | | report_path = os.path.join(script_dir, 'H:\\项目\\造数脚本\\Util\\stress_test_report_generator.py') |
|---|
| | | report_path = os.path.join(script_dir, 'H:\\项目\\archive\\测试组\\脚本\\造数脚本2\\Util\\stress_test_report_generator.py') |
|---|
| | | if os.path.exists(report_path): |
|---|
| | | spec = importlib.util.spec_from_file_location('report_module', report_path) |
|---|
| | | report_module = importlib.util.module_from_spec(spec) |
|---|
