test_basic.py

#!/usr/bin/env python3
"""
Quick test script to verify OpenLens core functionality
"""

import sys
sys.path.insert(0, 'src')

from openlens.utils import (
    keccak256,
    parse_event_signature,
    encode_event_signature,
    encode_address,
    encode_uint256,
    bytes_to_hex,
)

def test_keccak():
    """Test keccak256 hashing"""
    result = keccak256(b"hello")
    print(f"✅ keccak256('hello') = {bytes_to_hex(result)}")
    assert len(result) == 32

def test_event_signature_parsing():
    """Test event signature parsing"""
    sig = "Transfer(address,address,uint256)"
    name, params = parse_event_signature(sig)
    print(f"✅ Parsed '{sig}':")
    print(f"   Name: {name}")
    print(f"   Params: {params}")
    assert name == "Transfer"
    assert params == ["address", "address", "uint256"]

def test_event_signature_encoding():
    """Test event signature encoding (topic[0])"""
    sig = "Transfer(address,address,uint256)"
    topic0 = encode_event_signature(sig)
    topic0_hex = bytes_to_hex(topic0)
    print(f"✅ Event signature hash for '{sig}':")
    print(f"   {topic0_hex}")
    # This is the well-known Transfer event signature hash
    expected = "0xddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef"
    assert topic0_hex == expected

def test_address_encoding():
    """Test address encoding"""
    addr = "0x742d35Cc6634C0532925a3b844Bc9e7595f0bEb0"
    encoded = encode_address(addr)
    print(f"✅ Encoded address {addr}:")
    print(f"   {bytes_to_hex(encoded)}")
    assert len(encoded) == 32
    # Should be left-padded
    assert encoded[:12] == b'\x00' * 12

def test_uint256_encoding():
    """Test uint256 encoding"""
    value = 1000000000000000000  # 1 ETH in wei
    encoded = encode_uint256(value)
    print(f"✅ Encoded uint256 {value}:")
    print(f"   {bytes_to_hex(encoded)}")
    assert len(encoded) == 32
    assert int.from_bytes(encoded, byteorder='big') == value

def main():
    print("=" * 70)
    print("OpenLens Core Functionality Tests")
    print("=" * 70)
    print()
    
    try:
        test_keccak()
        print()
        test_event_signature_parsing()
        print()
        test_event_signature_encoding()
        print()
        test_address_encoding()
        print()
        test_uint256_encoding()
        print()
        print("=" * 70)
        print("✅ All tests passed!")
        print("=" * 70)
        return 0
    except Exception as e:
        print()
        print("=" * 70)
        print(f"❌ Test failed: {e}")
        print("=" * 70)
        import traceback
        traceback.print_exc()
        return 1

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