Ai_GirlFriend/test_invite_code.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
测试邀请码功能
"""

import requests

# 配置
BASE_URL = "http://localhost:30101"
TOKEN_USER1 = "3932cd35-4238-4c3f-ad5c-ad6ce9454e79"  # 用户1的 token（邀请人）
TOKEN_USER2 = "test-token-user2"  # 用户2的 token（被邀请人，需要替换为真实 token）

def test_get_invite_info(token):
    """测试获取邀请信息"""
    url = f"{BASE_URL}/config/invite/info"
    headers = {
        "Authorization": f"Bearer {token}"
    }
    
    print(f"\n=== 测试获取邀请信息 ===")
    print(f"请求 URL: {url}")
    
    response = requests.get(url, headers=headers)
    
    print(f"状态码: {response.status_code}")
    print(f"响应内容: {response.json()}")
    
    if response.status_code == 200:
        data = response.json()
        if data.get("code") == 1:
            invite_code = data.get("data", {}).get("invite_code")
            print(f"\n✅ 邀请码: {invite_code}")
            return invite_code
        else:
            print(f"\n❌ 获取失败: {data.get('message')}")
    else:
        print(f"\n❌ HTTP 错误: {response.status_code}")
    
    return None

def test_apply_invite_code(token, invite_code):
    """测试使用邀请码"""
    url = f"{BASE_URL}/config/invite/apply"
    headers = {
        "Authorization": f"Bearer {token}",
        "Content-Type": "application/json"
    }
    data = {
        "invite_code": invite_code
    }
    
    print(f"\n=== 测试使用邀请码 ===")
    print(f"请求 URL: {url}")
    print(f"邀请码: {invite_code}")
    
    response = requests.post(url, headers=headers, json=data)
    
    print(f"状态码: {response.status_code}")
    print(f"响应内容: {response.json()}")
    
    if response.status_code == 200:
        data = response.json()
        if data.get("code") == 1:
            print(f"\n✅ 使用成功: {data.get('data', {}).get('message')}")
            print(f"奖励: {data.get('data', {}).get('reward')} 金币")
            print(f"余额: {data.get('data', {}).get('balance')} 金币")
        else:
            print(f"\n❌ 使用失败: {data.get('message')}")
    else:
        print(f"\n❌ HTTP 错误: {response.status_code}")
        print(f"错误详情: {response.text}")

def test_check_user_info(token):
    """测试检查用户信息"""
    url = f"{BASE_URL}/user/basic"
    headers = {
        "Authorization": f"Bearer {token}"
    }
    
    print(f"\n=== 测试获取用户信息 ===")
    print(f"请求 URL: {url}")
    
    response = requests.get(url, headers=headers)
    
    print(f"状态码: {response.status_code}")
    if response.status_code == 200:
        data = response.json()
        print(f"响应内容: {data}")
        if data.get("code") == 1:
            user_data = data.get("data", {})
            print(f"\n✅ 用户ID: {user_data.get('user_id')}")
            print(f"金币余额: {user_data.get('money')}")
            print(f"邀请码: {user_data.get('invite_code')}")
            print(f"被邀请码: {user_data.get('invited_by')}")
    else:
        print(f"\n❌ HTTP 错误: {response.status_code}")

if __name__ == "__main__":
    print("=" * 60)
    print("邀请码功能测试")
    print("=" * 60)
    
    # 测试1：获取用户1的邀请码
    print("\n【步骤1】获取用户1（邀请人）的邀请码")
    invite_code = test_get_invite_info(TOKEN_USER1)
    
    if invite_code:
        print(f"\n【步骤2】用户1的邀请码是: {invite_code}")
        print("\n提示：请使用另一个用户的 token 来测试使用邀请码")
        print(f"修改脚本中的 TOKEN_USER2 变量，然后取消注释下面的代码")
        
        # 取消注释下面的代码来测试使用邀请码
        # print("\n【步骤3】用户2使用邀请码")
        # test_apply_invite_code(TOKEN_USER2, invite_code)
        
        # print("\n【步骤4】检查用户1的信息（应该增加了邀请计数和奖励）")
        # test_check_user_info(TOKEN_USER1)
        
        # print("\n【步骤5】检查用户2的信息（应该有 invited_by 字段）")
        # test_check_user_info(TOKEN_USER2)
    
    print("\n" + "=" * 60)