#!/usr/bin/env python3 # -*- coding: utf-8 -*- """ 测试生成新题目,验证difficulty字段是否正确设置 """ import requests import json def test_generate_questions(): base_url = "http://localhost:8080" print("=== 测试生成新题目 ===") # 1. 先创建一个新目标 create_goal_url = f"{base_url}/api/goals" goal_data = { "userId": 1, "subject": "数学", "goalType": "选择", "difficulty": "中等", "totalQuantity": 2, "goalContent": "测试difficulty字段的题目生成" } print(f"创建新目标: {goal_data}") create_response = requests.post(create_goal_url, json=goal_data) if create_response.status_code == 200: create_result = create_response.json() if create_result.get('code') == 200: goal_id = create_result['data']['goalId'] print(f"✅ 目标创建成功,ID: {goal_id}") # 2. 等待一下让自动生成题目完成 import time print("等待5秒让题目生成完成...") time.sleep(5) # 3. 获取生成的题目 questions_url = f"{base_url}/api/ai/questions/{goal_id}" questions_response = requests.get(questions_url) if questions_response.status_code == 200: questions_result = questions_response.json() if questions_result.get('code') == 200: questions = questions_result['data'] print(f"\n✅ 获取到 {len(questions)} 道题目") # 检查每道题目的difficulty字段 for i, question in enumerate(questions, 1): difficulty = question.get('difficulty') print(f"\n题目 {i}:") print(f" - detailId: {question.get('detailId')}") print(f" - difficulty: {difficulty}") print(f" - 题目内容: {question.get('question', '')[:50]}...") if difficulty: print(f" ✅ difficulty字段正常: {difficulty}") else: print(f" ❌ difficulty字段为空或null") else: print(f"❌ 获取题目失败: {questions_result.get('message')}") else: print(f"❌ 获取题目请求失败: {questions_response.status_code}") print(f"响应内容: {questions_response.text}") else: print(f"❌ 创建目标失败: {create_result.get('message')}") else: print(f"❌ 创建目标请求失败: {create_response.status_code}") print(f"响应内容: {create_response.text}") if __name__ == "__main__": test_generate_questions()