test_installation.py

"""
Test script to verify valkeyrie package installation
Run this AFTER installing from PyPI: pip install valkeyrie
"""

def test_import():
    """Test that package can be imported"""
    try:
        from valkeyrie import ValkeyreClient
        print("✅ Import successful!")
        return True
    except ImportError as e:
        print(f"❌ Import failed: {e}")
        return False

def test_connection():
    """Test connection to Valkeyrie server (requires running server)"""
    try:
        from valkeyrie import ValkeyreClient
        
        client = ValkeyreClient('127.0.0.1', 6379)
        if client.connect():
            print("✅ Connection successful!")
            
            # Test basic operations
            client.set('test_key', 'test_value')
            value = client.get('test_key')
            
            if value == 'test_value':
                print("✅ Basic operations work!")
            else:
                print(f"❌ Unexpected value: {value}")
            
            client.delete('test_key')
            client.disconnect()
            return True
        else:
            print("⚠️  Could not connect to server (make sure Valkeyrie is running)")
            return False
    except Exception as e:
        print(f"❌ Connection test failed: {e}")
        return False

def main():
    print("🧪 Testing Valkeyrie Package Installation\n")
    
    # Test 1: Import
    print("Test 1: Package Import")
    import_ok = test_import()
    print()
    
    # Test 2: Connection (optional)
    if import_ok:
        print("Test 2: Server Connection")
        print("(Make sure Valkeyrie server is running on localhost:6379)")
        test_connection()
    
    print("\n✨ Tests complete!")

if __name__ == '__main__':
    main()