增加接口测试功能

run_tests.py

@@ -0,0 +1,77 @@
+#!/usr/bin/env python3
+"""
+测试运行脚本
+"""
+
+import subprocess
+import sys
+import os
+
+def run_tests():
+    """运行所有测试"""
+    print("🧪 开始运行AI Talk回调接口测试...")
+    
+    # 设置环境变量为测试模式
+    os.environ["TESTING"] = "1"
+    
+    try:
+        # 运行pytest
+        result = subprocess.run([
+            sys.executable, "-m", "pytest", 
+            "test_callback.py", 
+            "-v",
+            "--tb=short",
+            "--color=yes"
+        ], capture_output=True, text=True)
+        
+        print(result.stdout)
+        if result.stderr:
+            print("错误信息:")
+            print(result.stderr)
+        
+        if result.returncode == 0:
+            print("✅ 所有测试通过!")
+            return True
+        else:
+            print("❌ 测试失败!")
+            return False
+            
+    except Exception as e:
+        print(f"❌ 运行测试时出错: {e}")
+        return False
+
+def run_specific_test(test_name: str):
+    """运行特定测试"""
+    print(f"🧪 运行特定测试: {test_name}")
+    
+    os.environ["TESTING"] = "1"
+    
+    try:
+        result = subprocess.run([
+            sys.executable, "-m", "pytest", 
+            "test_callback.py", 
+            f"-k={test_name}",
+            "-v",
+            "--tb=short",
+            "--color=yes"
+        ], capture_output=True, text=True)
+        
+        print(result.stdout)
+        if result.stderr:
+            print("错误信息:")
+            print(result.stderr)
+        
+        return result.returncode == 0
+        
+    except Exception as e:
+        print(f"❌ 运行测试时出错: {e}")
+        return False
+
+if __name__ == "__main__":
+    if len(sys.argv) > 1:
+        # 运行特定测试
+        test_name = sys.argv[1]
+        run_specific_test(test_name)
+    else:
+        # 运行所有测试
+        run_tests()