quick_test.py

#!/usr/bin/env python3
"""
Quick test script for Rust JWT service
"""

import requests
import json
import sys

def quick_test(base_url="http://localhost:8080"):
    """Run a quick test of all endpoints"""
    
    print(f"Testing JWT Authentication Service at {base_url}")
    print("="*60)
    
    # Test 1: Try to access protected route without token
    print("\n1. Testing protected route without token...")
    response = requests.get(f"{base_url}/api/protected")
    print(f"   Status: {response.status_code} (expected: 401)")
    print(f"   Response: {response.text[:100]}")
    
    # Test 2: Register a new user
    print("\n2. Registering new user...")
    user_data = {
        "username": "testuser_py",
        "password": "testpassword123"
    }
    response = requests.post(f"{base_url}/api/auth/register", json=user_data)
    print(f"   Status: {response.status_code} (expected: 201 or 409 if exists)")
    
    if response.status_code == 201:
        print("   ✓ User registered successfully")
    elif response.status_code == 409:
        print("   ℹ User already exists (trying to login instead)")
    else:
        print(f"   ✗ Unexpected response: {response.text}")
    
    # Test 3: Login
    print("\n3. Logging in...")
    response = requests.post(f"{base_url}/api/auth/login", json=user_data)
    print(f"   Status: {response.status_code} (expected: 200)")
    
    if response.status_code == 200:
        data = response.json()
        if data.get('success') and data.get('data', {}).get('token'):
            token = data['data']['token']
            print(f"   ✓ Login successful")
            print(f"   Token: {token[:50]}...")
            
            # Test 4: Access protected route with token
            print("\n4. Accessing protected route with token...")
            headers = {'Authorization': f'Bearer {token}'}
            response = requests.get(f"{base_url}/api/protected", headers=headers)
            print(f"   Status: {response.status_code} (expected: 200)")
            
            if response.status_code == 200:
                protected_data = response.json()
                print(f"   ✓ Protected route accessed successfully")
                print(f"   Response: {json.dumps(protected_data, indent=2)}")
            else:
                print(f"   ✗ Failed to access protected route: {response.text}")
        else:
            print(f"   ✗ Login response missing token: {data}")
    else:
        print(f"   ✗ Login failed: {response.text}")
    
    # Test 5: Test invalid login
    print("\n5. Testing invalid login...")
    invalid_data = {
        "username": "nonexistent",
        "password": "wrongpassword"
    }
    response = requests.post(f"{base_url}/api/auth/login", json=invalid_data)
    print(f"   Status: {response.status_code} (expected: 401)")
    print(f"   ✓ Invalid credentials correctly rejected" if response.status_code == 401 else f"   ✗ Unexpected response")
    
    print("\n" + "="*60)
    print("Test completed!")

if __name__ == "__main__":
    base_url = sys.argv[1] if len(sys.argv) > 1 else "http://localhost:8080"
    quick_test(base_url)