124 lines
3.8 KiB
Python
124 lines
3.8 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
测试修复后的监控器查询
|
||
"""
|
||
|
||
import sys
|
||
import os
|
||
import datetime
|
||
sys.path.append(os.path.dirname(__file__))
|
||
|
||
from influx_service import InfluxService, InfluxConnectionParams
|
||
from config_model import AppConfig
|
||
from pathlib import Path
|
||
|
||
def test_time_range_formats():
|
||
"""测试不同的时间范围格式"""
|
||
print("🧪 测试时间范围格式")
|
||
|
||
# 加载配置
|
||
config_path = Path("default.json")
|
||
config = AppConfig.load(config_path)
|
||
|
||
# 创建InfluxDB服务
|
||
params = InfluxConnectionParams(
|
||
url=config.influx.url,
|
||
org=config.influx.org,
|
||
token=config.influx.token
|
||
)
|
||
|
||
service = InfluxService(params)
|
||
|
||
bucket = getattr(config.influx, 'bucket', 'PCM')
|
||
measurement = getattr(config.influx, 'measurement', 'experiment_status')
|
||
fields = ['status']
|
||
filters = {}
|
||
|
||
# 测试不同的时间范围格式
|
||
test_cases = [
|
||
("-1h", "相对时间（1小时前）"),
|
||
("-30m", "相对时间（30分钟前）"),
|
||
('"2025-11-25T15:00:00Z"', "绝对时间（带引号和时区）"),
|
||
('2025-11-25T15:00:00Z', "绝对时间（无引号）- 应该失败"),
|
||
]
|
||
|
||
for time_range, description in test_cases:
|
||
print(f"\n🔍 测试: {description}")
|
||
print(f" 时间范围: {time_range}")
|
||
|
||
try:
|
||
# 生成查询
|
||
flux = service.build_flux(
|
||
bucket=bucket,
|
||
measurement=measurement,
|
||
fields=fields,
|
||
filters=filters,
|
||
time_range=time_range
|
||
)
|
||
|
||
print(" 生成的查询:")
|
||
for i, line in enumerate(flux.split('\n'), 1):
|
||
if line.strip():
|
||
print(f" {i}: {line}")
|
||
|
||
# 执行查询
|
||
df = service.query(
|
||
bucket=bucket,
|
||
measurement=measurement,
|
||
fields=fields,
|
||
filters=filters,
|
||
time_range=time_range
|
||
)
|
||
|
||
print(f" ✅ 查询成功! 返回 {len(df)} 行数据")
|
||
|
||
except Exception as e:
|
||
print(f" ❌ 查询失败: {e}")
|
||
|
||
def test_monitor_time_logic():
|
||
"""测试监控器的时间逻辑"""
|
||
print("\n🕐 测试监控器时间逻辑")
|
||
|
||
# 模拟不同的start_time情况
|
||
test_times = [
|
||
(datetime.datetime.now(), "当前时间（无时区）"),
|
||
(datetime.datetime.now(datetime.timezone.utc), "当前时间（UTC时区）"),
|
||
("2025-11-25T15:00:00", "字符串时间"),
|
||
(None, "None值"),
|
||
]
|
||
|
||
for start_time, description in test_times:
|
||
print(f"\n🔍 测试: {description}")
|
||
print(f" start_time: {start_time}")
|
||
|
||
try:
|
||
# 模拟监控器的时间处理逻辑
|
||
if hasattr(start_time, 'isoformat'):
|
||
if start_time.tzinfo is None:
|
||
start_time_utc = start_time.replace(tzinfo=datetime.timezone.utc)
|
||
else:
|
||
start_time_utc = start_time.astimezone(datetime.timezone.utc)
|
||
start_iso = start_time_utc.isoformat()
|
||
else:
|
||
start_iso = "-1h"
|
||
|
||
if start_iso.startswith('-'):
|
||
time_range = start_iso
|
||
else:
|
||
time_range = f'"{start_iso}"'
|
||
|
||
print(f" 处理后的时间范围: {time_range}")
|
||
print(" ✅ 时间处理成功")
|
||
|
||
except Exception as e:
|
||
print(f" ❌ 时间处理失败: {e}")
|
||
|
||
def main():
|
||
print("监控器修复测试工具")
|
||
print("=" * 40)
|
||
test_time_range_formats()
|
||
test_monitor_time_logic()
|
||
|
||
if __name__ == "__main__":
|
||
main()
|