diff --git a/performance_test.py b/performance_test.py index bf7621b..98636d7 100644 --- a/performance_test.py +++ b/performance_test.py @@ -9,7 +9,7 @@ import requests import smtplib import time -import datetime +from datetime import datetime as dt import random import string import argparse @@ -177,7 +177,7 @@ def test_api_responsiveness(api_base_url="http://localhost:5000/api", num_reques def send_test_email(smtp_host, smtp_port, from_addr, to_addr): """发送测试邮件并计时""" message_id = f"{generate_random_string()}@test.local" - subject = f"测试邮件 {datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}" + subject = f"测试邮件 {dt.now().strftime('%Y-%m-%d %H:%M:%S')}" # 创建邮件 msg = MIMEMultipart() @@ -190,7 +190,7 @@ def send_test_email(smtp_host, smtp_port, from_addr, to_addr): body = f""" 这是一封测试邮件。 - 发送时间: {datetime.datetime.now().isoformat()} + 发送时间: {dt.now().isoformat()} 唯一ID: {message_id} 此邮件用于测试系统性能。 @@ -412,7 +412,7 @@ def save_results(results, filename="performance_results.json"): def run_all_tests(args): """运行所有测试""" results = { - 'timestamp': datetime.datetime.now().isoformat(), + 'timestamp': dt.now().isoformat(), 'config': vars(args), 'tests': {} }