test_scraper.py

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

BASE_URL = "http://localhost:8800"
USERNAME = "klu2200031641"

def test_endpoint(endpoint, name):
    print(f"\n{'='*60}")
    print(f"Testing: {name}")
    print(f"{'='*60}")
    try:
        response = requests.get(f"{BASE_URL}{endpoint}", timeout=30)
        print(f"Status: {response.status_code}")
        data = response.json()
        print(json.dumps(data, indent=2))
        return response.status_code == 200 and data.get('success', False)
    except Exception as e:
        print(f"Error: {e}")
        return False

if __name__ == "__main__":
    print(f"Testing CodeChef Scraper for user: {USERNAME}")

    results = {
        "Health Check": test_endpoint("/health", "Health Check"),
        "Profile": test_endpoint(f"/api/profile/{USERNAME}", "Profile Data"),
        "Ratings": test_endpoint(f"/api/ratings/{USERNAME}", "Ratings History"),
        "Recent": test_endpoint(f"/api/recent/{USERNAME}", "Recent Submissions"),
        "Whole Profile": test_endpoint(f"/api/whole/{USERNAME}", "Complete Profile"),
        "Upcoming Contests": test_endpoint("/api/upcoming", "Upcoming Contests")
    }

    print(f"\n{'='*60}")
    print("SUMMARY")
    print(f"{'='*60}")
    for test, passed in results.items():
        status = "✓ PASS" if passed else "✗ FAIL"
        print(f"{test:20} {status}")

    total = len(results)
    passed = sum(results.values())
    print(f"\nTotal: {passed}/{total} tests passed")

    sys.exit(0 if passed == total else 1)