#!/usr/bin/env python3
"""
Test script for registration API endpoints
"""
import requests
import json
from datetime import date, timedelta

# API base URL
BASE_URL = "http://localhost:8009"

def test_daily_registered_users():
    """Test the daily registered users endpoint"""
    print("Testing daily registered users endpoint...")
    
    # Test with last 7 days
    end_date = date.today()
    start_date = end_date - timedelta(days=6)
    
    url = f"{BASE_URL}/api/metrics/daily-registered-users"
    params = {
        "start_date": str(start_date),
        "end_date": str(end_date),
        "product_id": "freeleaps"
    }
    
    try:
        response = requests.get(url, params=params)
        print(f"Status Code: {response.status_code}")
        
        if response.status_code == 200:
            data = response.json()
            print(f"Response: {json.dumps(data, indent=2)}")
            print(f"Number of days: {len(data['dates'])}")
            print(f"Total registrations: {data['total_registrations']}")
        else:
            print(f"Error: {response.text}")
            
    except Exception as e:
        print(f"Request failed: {e}")

def test_registration_summary():
    """Test the registration summary endpoint"""
    print("\nTesting registration summary endpoint...")
    
    end_date = date.today()
    start_date = end_date - timedelta(days=6)
    
    url = f"{BASE_URL}/api/metrics/registration-summary"
    params = {
        "start_date": str(start_date),
        "end_date": str(end_date),
        "product_id": "freeleaps"
    }
    
    try:
        response = requests.get(url, params=params)
        print(f"Status Code: {response.status_code}")
        
        if response.status_code == 200:
            data = response.json()
            print(f"Summary: {json.dumps(data, indent=2)}")
        else:
            print(f"Error: {response.text}")
            
    except Exception as e:
        print(f"Request failed: {e}")

def test_post_method():
    """Test the POST method for daily registered users"""
    print("\nTesting POST method for daily registered users...")
    
    end_date = date.today()
    start_date = end_date - timedelta(days=6)
    
    url = f"{BASE_URL}/api/metrics/daily-registered-users"
    payload = {
        "start_date": str(start_date),
        "end_date": str(end_date),
        "product_id": "freeleaps"
    }
    
    try:
        response = requests.post(url, json=payload)
        print(f"Status Code: {response.status_code}")
        
        if response.status_code == 200:
            data = response.json()
            print(f"Response: {json.dumps(data, indent=2)}")
        else:
            print(f"Error: {response.text}")
            
    except Exception as e:
        print(f"Request failed: {e}")

if __name__ == "__main__":
    print("Starting registration API tests...")
    print(f"Testing against: {BASE_URL}")
    print("=" * 50)
    
    test_daily_registered_users()
    test_registration_summary()
    test_post_method()
    
    print("\n" + "=" * 50)
    print("Tests completed!")