test_setup.py

#!/usr/bin/env python3
"""
Test script to verify the Relay listener setup
"""

import sys
import os

def test_imports():
    """Test if all required modules can be imported"""
    print("🔍 Testing imports...")
    
    try:
        import yaml
        print("✅ PyYAML imported successfully")
    except ImportError as e:
        print(f"❌ PyYAML import failed: {e}")
        return False
    
    try:
        import web3
        print("✅ Web3.py imported successfully")
    except ImportError as e:
        print(f"❌ Web3.py import failed: {e}")
        return False
    
    try:
        from dotenv import load_dotenv
        print("✅ python-dotenv imported successfully")
    except ImportError as e:
        print(f"❌ python-dotenv import failed: {e}")
        return False
    
    try:
        from config.config_loader import get_config
        print("✅ Config loader imported successfully")
    except ImportError as e:
        print(f"❌ Config loader import failed: {e}")
        return False
    
    return True

def test_config_loading():
    """Test if configuration can be loaded"""
    print("\n🔍 Testing configuration loading...")
    
    try:
        from config.config_loader import get_config
        config = get_config()
        print("✅ Configuration loaded successfully")
        
        # Test some config methods
        api_key = config.get_alchemy_api_key()
        if api_key:
            print(f"✅ Alchemy API key found: {api_key[:10]}...")
        else:
            print("⚠️  No Alchemy API key found (expected if .env not set)")
        
        addresses = config.get_all_shapeshift_addresses()
        print(f"✅ Found {len(addresses)} ShapeShift affiliate addresses")
        
        chains = config.get_supported_chains()
        print(f"✅ Supported chains: {', '.join(chains)}")
        
        return True
        
    except Exception as e:
        print(f"❌ Configuration loading failed: {e}")
        return False

def test_directory_structure():
    """Test if required directories exist"""
    print("\n🔍 Testing directory structure...")
    
    required_dirs = ["config", "data"]
    required_files = [
        "config/config.yaml",
        "config/config.example.yaml", 
        "config/config_loader.py",
        "relay_listener.py",
        "requirements.txt",
        "README.md"
    ]
    
    for directory in required_dirs:
        if os.path.exists(directory):
            print(f"✅ Directory exists: {directory}")
        else:
            print(f"❌ Directory missing: {directory}")
            return False
    
    for file_path in required_files:
        if os.path.exists(file_path):
            print(f"✅ File exists: {file_path}")
        else:
            print(f"❌ File missing: {file_path}")
            return False
    
    return True

def main():
    """Run all tests"""
    print("🚀 Relay Affiliate Fee Listener - Setup Test")
    print("=" * 50)
    
    tests = [
        test_imports,
        test_config_loading,
        test_directory_structure
    ]
    
    passed = 0
    total = len(tests)
    
    for test in tests:
        if test():
            passed += 1
        print()
    
    print("=" * 50)
    print(f"📊 Test Results: {passed}/{total} tests passed")
    
    if passed == total:
        print("🎉 All tests passed! Setup is complete.")
        print("\nNext steps:")
        print("1. Copy env.example to .env and add your API keys")
        print("2. Run: python relay_listener.py --help")
        print("3. Start monitoring: python relay_listener.py")
    else:
        print("❌ Some tests failed. Please check the errors above.")
        return 1
    
    return 0

if __name__ == "__main__":
    sys.exit(main())