提示词专家

tests/test_wx_api.py

@@ -0,0 +1,103 @@
+import requests
+import json
+
+# API基础URL
+BASE_URL = 'http://localhost:5000/api/wx'
+
+def test_generate_prompt():
+    """测试生成提示词接口"""
+    print("\n=== 测试生成提示词 ===")
+    
+    url = f'{BASE_URL}/generate'
+    data = {
+        'input_text': '写一篇关于人工智能的文章',
+        'template_id': 1  # 可选
+    }
+    
+    try:
+        response = requests.post(url, json=data)
+        result = response.json()
+        
+        print(f"状态码: {response.status_code}")
+        print(f"响应数据: {json.dumps(result, ensure_ascii=False, indent=2)}")
+        
+        if response.status_code == 200 and result['code'] == 200:
+            print("✓ 生成提示词测试通过")
+        else:
+            print("✗ 生成提示词测试失败")
+            
+    except Exception as e:
+        print(f"✗ 测试出错: {str(e)}")
+
+def test_get_templates():
+    """测试获取模板列表接口"""
+    print("\n=== 测试获取模板列表 ===")
+    
+    # 测试不同的筛选条件
+    test_cases = [
+        {},  # 无筛选
+        {'industry': '移动开发'},  # 按行业筛选
+        {'profession': 'Android工程师'},  # 按职业筛选
+        {'category': '功能开发'},  # 按分类筛选
+        {'industry': '移动开发', 'profession': 'Android工程师'}  # 组合筛选
+    ]
+    
+    for params in test_cases:
+        print(f"\n测试筛选条件: {params}")
+        url = f'{BASE_URL}/templates'
+        
+        try:
+            response = requests.get(url, params=params)
+            result = response.json()
+            
+            print(f"状态码: {response.status_code}")
+            print(f"返回模板数量: {len(result['data']) if result['code'] == 200 else 0}")
+            
+            if response.status_code == 200 and result['code'] == 200:
+                print("✓ 获取模板列表测试通过")
+            else:
+                print("✗ 获取模板列表测试失败")
+                
+        except Exception as e:
+            print(f"✗ 测试出错: {str(e)}")
+
+def test_get_template_detail():
+    """测试获取模板详情接口"""
+    print("\n=== 测试获取模板详情 ===")
+    
+    # 先获取模板列表
+    templates_url = f'{BASE_URL}/templates'
+    try:
+        response = requests.get(templates_url)
+        result = response.json()
+        
+        if response.status_code == 200 and result['code'] == 200 and result['data']:
+            # 获取第一个模板的ID
+            template_id = result['data'][0]['id']
+            
+            # 测试获取详情
+            detail_url = f'{BASE_URL}/template/{template_id}'
+            detail_response = requests.get(detail_url)
+            detail_result = detail_response.json()
+            
+            print(f"状态码: {detail_response.status_code}")
+            print(f"响应数据: {json.dumps(detail_result, ensure_ascii=False, indent=2)}")
+            
+            if detail_response.status_code == 200 and detail_result['code'] == 200:
+                print("✓ 获取模板详情测试通过")
+            else:
+                print("✗ 获取模板详情测试失败")
+        else:
+            print("✗ 无法获取模板列表进行测试")
+            
+    except Exception as e:
+        print(f"✗ 测试出错: {str(e)}")
+
+def run_all_tests():
+    """运行所有测试"""
+    test_generate_prompt()
+    test_get_templates()
+    test_get_template_detail()
+
+if __name__ == '__main__':
+    run_all_tests()