#!/usr/bin/env python3
"""
Test script to verify URL expansion functionality
"""
import sys
import os

# Add the current directory to Python path to import our modules
sys.path.append(os.path.dirname(os.path.abspath(__file__)))

from url_utils import expand_ecommerce_url, is_amazon_short_link, is_flipkart_short_link

def test_url_expansion():
    """Test URL expansion with different types of URLs"""
    
    test_urls = [
        # Amazon short links
        "https://amzn.to/4nDPQmT",
        "https://www.amzn.to/3xyz123",
        
        # Regular Amazon URLs (should not be expanded)
        "https://www.amazon.in/dp/B08N5WRWNW",
        "https://amazon.com/product/123",
        
        # Flipkart short links
        "https://fkrt.it/ABC123",
        
        # Regular Flipkart URLs (should not be expanded)
        "https://www.flipkart.com/product/p/123"
    ]
    
    print("🔗 Testing URL Expansion Functionality")
    print("=" * 50)
    
    for url in test_urls:
        print(f"\n📌 Testing URL: {url}")
        
        # Check if it's a short link
        is_amazon_short = is_amazon_short_link(url)
        is_flipkart_short = is_flipkart_short_link(url)
        
        print(f"   Is Amazon short link: {is_amazon_short}")
        print(f"   Is Flipkart short link: {is_flipkart_short}")
        
        if is_amazon_short or is_flipkart_short:
            print("   🔄 Expanding URL...")
            try:
                expanded_url = expand_ecommerce_url(url)
                if expanded_url != url:
                    print(f"   ✅ Expanded to: {expanded_url}")
                else:
                    print("   ⚠️  URL remained unchanged")
            except Exception as e:
                print(f"   ❌ Error expanding URL: {e}")
        else:
            print("   ➡️  No expansion needed")
    
    print("\n" + "=" * 50)
    print("🎉 URL expansion test completed!")

if __name__ == "__main__":
    test_url_expansion()