LQAdminPlatform/test/test_user_menus.py

#!/usr/bin/env python3
import requests
import json

# First login to get token
login_data = {
    'username': 'admin',
    'password': 'Admin123456'
}

try:
    # Login
    login_response = requests.post('http://localhost:8000/api/v1/auth/login', json=login_data)
    if login_response.status_code == 200:
        login_result = login_response.json()
        if login_result.get('code') == 0:
            token = login_result['data']['access_token']
            print('✅ Login successful')
            
            # Get user menus
            headers = {'Authorization': f'Bearer {token}'}
            menus_response = requests.get('http://localhost:8000/api/v1/user/menus', headers=headers)
            
            if menus_response.status_code == 200:
                menus_result = menus_response.json()
                if menus_result.get('code') == 0:
                    menus = menus_result['data']
                    print(f'✅ Got {len(menus)} total items for admin user')
                    
                    # Count menu types
                    menu_count = 0
                    for menu in menus:
                        if menu.get('menu_type') == 'menu':
                            menu_count += 1
                            print(f'📁 Menu: {menu["title"]} ({menu["name"]})')
                            if menu.get('children'):
                                for child in menu['children']:
                                    if child.get('menu_type') == 'menu':
                                        menu_count += 1
                                        print(f'   📄 {child["title"]} ({child["name"]})')
                    
                    print(f'📊 Total menu items (type=menu): {menu_count}')
                else:
                    print(f'❌ Menu API error: {menus_result.get("message")}')
            else:
                print(f'❌ Menu API HTTP error: {menus_response.status_code}')
        else:
            print(f'❌ Login error: {login_result.get("message")}')
    else:
        print(f'❌ Login HTTP error: {login_response.status_code}')
        
except Exception as e:
    print(f'❌ Request error: {e}')