ai_la/base7/test_achievement_initialization.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
测试成就初始化功能
验证新用户注册和老用户登录时的成就初始化
"""

import requests
import json
import time
import random
import string

# 服务器配置
BASE_URL = "http://localhost:8080/api"

def generate_random_username():
    """生成随机用户名"""
    return 'test_user_' + ''.join(random.choices(string.ascii_lowercase + string.digits, k=8))

def test_new_user_registration():
    """测试新用户注册时的成就初始化"""
    print("\n=== 测试新用户注册时的成就初始化 ===")
    
    # 生成随机用户名
    username = generate_random_username()
    password = "test123"
    grade = "小学三年级"
    
    print(f"注册新用户: {username}")
    
    # 注册用户
    register_data = {
        "username": username,
        "password": password,
        "grade": grade
    }
    
    try:
        response = requests.post(f"{BASE_URL}/users/register", json=register_data)
        print(f"注册响应状态: {response.status_code}")
        
        if response.status_code == 200:
            user_data = response.json()
            print(f"注册成功: {json.dumps(user_data, ensure_ascii=False, indent=2)}")
            user_id = user_data['data']['userId']
            
            # 等待一下确保成就初始化完成
            time.sleep(1)
            
            # 检查用户成就
            achievements_response = requests.get(f"{BASE_URL}/achievements/user/{user_id}")
            print(f"\n获取成就响应状态: {achievements_response.status_code}")
            
            if achievements_response.status_code == 200:
                achievements = achievements_response.json()
                print(f"\n用户成就数量: {len(achievements)}")
                
                if achievements:
                    print("\n成就列表:")
                    for achievement in achievements:
                        status = "已获得" if achievement.get('isAchieved') else "未获得"
                        progress = achievement.get('currentValue', 0)
                        target = achievement.get('targetValue', 0)
                        print(f"- {achievement.get('title')}: {status} ({progress}/{target})")
                    
                    # 验证所有成就都是未获得状态
                    unachieved_count = sum(1 for a in achievements if not a.get('isAchieved'))
                    print(f"\n未获得成就数量: {unachieved_count}/{len(achievements)}")
                    
                    if unachieved_count == len(achievements):
                        print("✅ 新用户注册成就初始化测试通过 - 所有成就都是未获得状态")
                        return True, user_data
                    else:
                        print("❌ 新用户注册成就初始化测试失败 - 存在已获得的成就")
                        return False, user_data
                else:
                    print("❌ 新用户注册成就初始化测试失败 - 没有找到任何成就")
                    return False, user_data
            else:
                print(f"❌ 获取成就失败: {achievements_response.text}")
                return False, user_data
        else:
            print(f"❌ 注册失败: {response.text}")
            return False, None
            
    except Exception as e:
        print(f"❌ 测试过程中发生错误: {str(e)}")
        return False, None

def test_existing_user_login(user_data):
    """测试已存在用户登录时的成就检查"""
    print("\n=== 测试已存在用户登录时的成就检查 ===")
    
    if not user_data:
        print("❌ 没有可用的用户数据进行测试")
        return False
    
    username = user_data['data']['username']
    password = "test123"
    
    print(f"用户登录: {username}")
    
    # 用户登录
    login_data = {
        "username": username,
        "password": password
    }
    
    try:
        response = requests.post(f"{BASE_URL}/users/login", json=login_data)
        print(f"登录响应状态: {response.status_code}")
        
        if response.status_code == 200:
            login_result = response.json()
            print(f"登录成功: {json.dumps(login_result, ensure_ascii=False, indent=2)}")
            user_id = login_result['data']['userId']
            
            # 再次检查用户成就
            achievements_response = requests.get(f"{BASE_URL}/achievements/user/{user_id}")
            print(f"\n获取成就响应状态: {achievements_response.status_code}")
            
            if achievements_response.status_code == 200:
                achievements = achievements_response.json()
                print(f"\n用户成就数量: {len(achievements)}")
                
                if achievements:
                    unachieved_count = sum(1 for a in achievements if not a.get('isAchieved'))
                    print(f"未获得成就数量: {unachieved_count}/{len(achievements)}")
                    
                    if len(achievements) > 0:
                        print("✅ 用户登录成就检查测试通过 - 成就数据保持一致")
                        return True
                    else:
                        print("❌ 用户登录成就检查测试失败 - 成就数据丢失")
                        return False
                else:
                    print("❌ 用户登录成就检查测试失败 - 没有找到任何成就")
                    return False
            else:
                print(f"❌ 获取成就失败: {achievements_response.text}")
                return False
        else:
            print(f"❌ 登录失败: {response.text}")
            return False
            
    except Exception as e:
        print(f"❌ 测试过程中发生错误: {str(e)}")
        return False

def test_achievement_types():
    """测试成就类型和数量"""
    print("\n=== 测试成就类型和数量 ===")
    
    # 使用第一个测试用户
    username = generate_random_username()
    password = "test123"
    grade = "小学五年级"
    
    # 注册用户
    register_data = {
        "username": username,
        "password": password,
        "grade": grade
    }
    
    try:
        response = requests.post(f"{BASE_URL}/users/register", json=register_data)
        if response.status_code == 200:
            user_data = response.json()
            user_id = user_data['data']['userId']
            
            time.sleep(1)  # 等待成就初始化
            
            # 获取成就
            achievements_response = requests.get(f"{BASE_URL}/achievements/user/{user_id}")
            if achievements_response.status_code == 200:
                achievements = achievements_response.json()
                
                # 按类型分组统计
                type_counts = {}
                for achievement in achievements:
                    achievement_type = achievement.get('type', 'UNKNOWN')
                    type_counts[achievement_type] = type_counts.get(achievement_type, 0) + 1
                
                print(f"\n成就类型统计:")
                for achievement_type, count in type_counts.items():
                    print(f"- {achievement_type}: {count}个")
                
                expected_types = [
                    'STUDY_TIME_1H', 'STUDY_TIME_10H', 'STUDY_TIME_50H', 'STUDY_TIME_100H',
                    'CONSECUTIVE_3D', 'CONSECUTIVE_7D', 'CONSECUTIVE_30D',
                    'GOAL_10', 'GOAL_50',
                    'ACCURACY_90', 'ACCURACY_100',
                    'EARLY_BIRD', 'NIGHT_OWL'
                ]
                
                missing_types = [t for t in expected_types if t not in type_counts]
                if missing_types:
                    print(f"\n❌ 缺少的成就类型: {missing_types}")
                    return False
                else:
                    print(f"\n✅ 所有预期的成就类型都已创建")
                    return True
            else:
                print(f"❌ 获取成就失败: {achievements_response.text}")
                return False
        else:
            print(f"❌ 注册失败: {response.text}")
            return False
            
    except Exception as e:
        print(f"❌ 测试过程中发生错误: {str(e)}")
        return False

def main():
    """主测试函数"""
    print("开始测试成就初始化功能...")
    
    # 测试1: 新用户注册
    success1, user_data = test_new_user_registration()
    
    # 测试2: 用户登录
    success2 = test_existing_user_login(user_data)
    
    # 测试3: 成就类型和数量
    success3 = test_achievement_types()
    
    # 总结
    print("\n" + "="*50)
    print("测试结果总结:")
    print(f"新用户注册成就初始化: {'✅ 通过' if success1 else '❌ 失败'}")
    print(f"用户登录成就检查: {'✅ 通过' if success2 else '❌ 失败'}")
    print(f"成就类型和数量验证: {'✅ 通过' if success3 else '❌ 失败'}")
    
    if success1 and success2 and success3:
        print("\n🎉 所有测试通过！成就初始化功能正常工作")
    else:
        print("\n⚠️ 部分测试失败，请检查相关功能")

if __name__ == "__main__":
    main()