multi-chain-bridge.bats

#!/usr/bin/env bats
# bats file_tags=lxly,multi-chain-bridge
# shellcheck disable=SC1091,SC2154

setup_file() {
    # shellcheck source=core/helpers/common.bash
    source "$BATS_TEST_DIRNAME/../../core/helpers/common.bash"
    _setup_vars

    # Multi-chain network configurations
    declare -A l2_rpc_urls=(
        ["network1"]="$(kurtosis port print "$kurtosis_enclave_name" cdk-erigon-rpc-001 rpc)"
        ["network2"]="$(kurtosis port print "$kurtosis_enclave_name" op-el-1-op-geth-op-node-002 rpc)"
    )
    
    declare -A bridge_service_urls=(
        ["network1"]="$(kurtosis port print "$kurtosis_enclave_name" zkevm-bridge-service-001 rpc)"
        ["network2"]="$(kurtosis port print "$kurtosis_enclave_name" zkevm-bridge-service-002 rpc)"
    )
    export l2_rpc_urls
    export bridge_service_urls
    
    # Default to network1 for backward compatibility
    export default_network=${NETWORK_TARGET:-"network1"}
    export l2_rpc_url=${L2_RPC_URL:-"${l2_rpc_urls[$default_network]}"}
    export l2_bridge_addr=${L2_BRIDGE_ADDR:-"0x78908F7A87d589fdB46bdd5EfE7892C5aD6001b6"}
    export bridge_service_url=${BRIDGE_SERVICE_URL:-"${bridge_service_urls[$default_network]}"}
    
    network_id=$(cast call --rpc-url "$l2_rpc_url" "$l2_bridge_addr" 'networkID()(uint32)')
    export network_id
    export claimtxmanager_addr=${CLAIMTXMANAGER_ADDR:-"0x5f5dB0D4D58310F53713eF4Df80ba6717868A9f8"}
    export claim_wait_duration=${CLAIM_WAIT_DURATION:-"10m"}

    export erc20_token_name="e2e test"
    export erc20_token_symbol="E2E"

    load "$BATS_TEST_DIRNAME/../../core/helpers/scripts/bridging.bash"
    fund_claim_tx_manager
}

setup() {
    load "$BATS_TEST_DIRNAME/../../core/helpers/scripts/bridging.bash"
}


# Helper function to get network configuration
function get_network_config() {
    local network_name="$1"
    local config_type="$2"
    
    case $config_type in
        "rpc_url")
            case $network_name in
                "network1") kurtosis port print "$kurtosis_enclave_name" cdk-erigon-rpc-001 rpc ;;
                "network2") kurtosis port print "$kurtosis_enclave_name" op-el-1-op-geth-op-node-002 rpc ;;
                *) echo "" ;;
            esac
            ;;
        "bridge_service_url")
            case $network_name in
                "network1") kurtosis port print "$kurtosis_enclave_name" zkevm-bridge-service-001 rpc ;;
                "network2") kurtosis port print "$kurtosis_enclave_name" zkevm-bridge-service-002 rpc ;;
                *) echo "" ;;
            esac
            ;;
    esac
}

# bats test_tags=bridge
@test "bridge native eth from L1 to L2 ("$NETWORK_TARGET")" {
    echo "Stopping the bridge-spammer service" >&3
    kurtosis service stop "$kurtosis_enclave_name" bridge-spammer-001

    echo "Starting bridge native ETH test for network: $NETWORK_TARGET" >&3
    echo "L1 RPC URL: $l1_rpc_url" >&3
    echo "L2 RPC URL: $l2_rpc_url" >&3
    echo "Network ID: $network_id" >&3
    
    bridge_amount=$(date +%s)
    echo "Bridge amount: $bridge_amount wei" >&3
    echo "Bridging ETH from L1 to L2..." >&3
    run polycli ulxly bridge asset \
            --bridge-address "$l1_bridge_addr" \
            --destination-address "$l2_eth_address" \
            --destination-network "$network_id" \
            --private-key "$l1_private_key" \
            --rpc-url "$l1_rpc_url" \
            --value "$bridge_amount"

    run polycli_bridge_asset_get_info "$output" "$l1_rpc_url" "$l1_bridge_addr"
    if [[ $status -ne 0 ]]; then
        echo "Failed to get deposit info" >&3
        echo "$output" >&3
        exit 1
    fi
    deposit_count=$(echo "$output" | jq -r '.depositCount')
    echo "Deposit count: $deposit_count" >&3

    echo "ETH bridge transaction completed, attempting to claim on L2..." >&3
    # It's possible this command will fail due to the auto claimer
    set +e
    polycli ulxly claim asset \
            --bridge-address "$l2_bridge_addr" \
            --private-key "$l2_private_key" \
            --rpc-url "$l2_rpc_url" \
            --deposit-count "$deposit_count" \
            --deposit-network "0" \
            --bridge-service-url "$bridge_service_url" \
            --wait "$claim_wait_duration"
    set -e
    echo "L1 to L2 ETH bridge test completed for network: $NETWORK_TARGET" >&3
}

# bats test_tags=bridge,transaction-erc20
@test "bridge L2 ("$NETWORK_TARGET") originated token from L2 to L1" {
    echo "Starting ERC20 token bridge test for network: $NETWORK_TARGET" >&3
    echo "Setting up deterministic deployer and ERC20 token..." >&3
    
    salt="0x0000000000000000000000000000000000000000000000000000000000000000"
    deterministic_deployer_addr=0x4e59b44847b379578588920ca78fbf26c0b4956c
    deterministic_deployer_code=$(cast code --rpc-url "$l2_rpc_url" "$deterministic_deployer_addr")

    if [[ $deterministic_deployer_code == "0x" ]]; then
        echo "Deploying missing deterministic deployer proxy contract..." >&3
        cast send --legacy --value 0.1ether --rpc-url "$l2_rpc_url" --private-key "$l2_private_key" 0x3fab184622dc19b6109349b94811493bf2a45362
        cast publish --rpc-url "$l2_rpc_url" 0xf8a58085174876e800830186a08080b853604580600e600039806000f350fe7fffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffe03601600081602082378035828234f58015156039578182fd5b8082525050506014600cf31ba02222222222222222222222222222222222222222222222222222222222222222a02222222222222222222222222222222222222222222222222222222222222222
    fi

    erc_20_bytecode=$(cat core/contracts/bin/erc20permitmock.bin)
    constructor_args=$(cast abi-encode 'f(string,string,address,uint256)' "$erc20_token_name" "$erc20_token_symbol" "$l2_eth_address" 100000000000000000000 | sed 's/0x//')
    test_erc20_addr=$(cast create2 --salt $salt --init-code "$erc_20_bytecode$constructor_args")
    echo "ERC20 token address: $test_erc20_addr" >&3

    if [[ $(cast code --rpc-url "$l2_rpc_url" "$test_erc20_addr") == "0x" ]]; then
        echo "Deploying ERC20 test token..." >&3
        cast send --legacy --rpc-url "$l2_rpc_url" --private-key "$l2_private_key" "$deterministic_deployer_addr" "$salt$erc_20_bytecode$constructor_args"
        cast send --legacy --rpc-url "$l2_rpc_url" --private-key "$l2_private_key" "$test_erc20_addr" 'approve(address,uint256)' "$l2_bridge_addr" "$(cast max-uint)"
        echo "ERC20 token deployed and approved for bridge" >&3
    fi

    bridge_amount=$(date +%s)
    echo "Bridge amount: $bridge_amount" >&3
    echo "Bridging ERC20 token from L2 to L1..." >&3
    # Bridge some funds from L2 to L1
    run polycli ulxly bridge asset \
            --destination-network 0 \
            --token-address  "$test_erc20_addr" \
            --value "$bridge_amount" \
            --bridge-address "$l2_bridge_addr" \
            --rpc-url "$l2_rpc_url" \
            --private-key "$l2_private_key"

    run polycli_bridge_asset_get_info "$output" "$l2_rpc_url" "$l2_bridge_addr"
    if [[ $status -ne 0 ]]; then
        echo "Failed to get deposit info" >&3
        echo "$output" >&3
        exit 1
    fi
    deposit_count=$(echo "$output" | jq -r '.depositCount')
    echo "Deposit count: $deposit_count" >&3

    echo "Claiming ERC20 token on L1..." >&3

    polycli ulxly claim asset \
            --bridge-address "$l1_bridge_addr" \
            --private-key "$l1_private_key" \
            --rpc-url "$l1_rpc_url" \
            --deposit-count "$deposit_count" \
            --deposit-network "$network_id" \
            --bridge-service-url "$bridge_service_url" \
            --wait "$claim_wait_duration"

    token_hash=$(cast keccak "$(cast abi-encode --packed 'f(uint32, address)' "$network_id" "$test_erc20_addr")")
    wrapped_token_addr=$(cast call --rpc-url "$l1_rpc_url" "$l1_bridge_addr" 'tokenInfoToWrappedToken(bytes32)(address)' "$token_hash")
    echo "Wrapped token address on L1: $wrapped_token_addr" >&3

    echo "Bridging wrapped token from L1 back to L2..." >&3

    run polycli ulxly bridge asset \
        --destination-network "$network_id" \
        --token-address "$wrapped_token_addr" \
        --value "$bridge_amount" \
        --bridge-address "$l1_bridge_addr" \
        --rpc-url "$l1_rpc_url" \
        --private-key "$l1_private_key"

    run polycli_bridge_asset_get_info "$output" "$l1_rpc_url" "$l1_bridge_addr"
    if [[ $status -ne 0 ]]; then
        echo "Failed to get deposit info" >&3
        echo "$output" >&3
        exit 1
    fi
    deposit_count=$(echo "$output" | jq -r '.depositCount')
    echo "Deposit count: $deposit_count" >&3

    echo "Attempting to claim wrapped token back on L2..." >&3
    # It's possible this command will fail due to the auto claimer
    set +e
    polycli ulxly claim asset \
            --bridge-address "$l2_bridge_addr" \
            --private-key "$l2_private_key" \
            --rpc-url "$l2_rpc_url" \
            --deposit-count "$deposit_count" \
            --deposit-network "0" \
            --bridge-service-url "$bridge_service_url" \
            --wait "$claim_wait_duration"
    set -e

    echo "Performing second round of L2 to L1 bridging..." >&3
    # repeat the first step again to trigger another exit of l2 but with the added claim
    bridge_amount=$(date +%s)
    echo "Second round - Bridge amount: $bridge_amount" >&3
    run polycli ulxly bridge asset \
            --destination-network 0 \
            --token-address  "$test_erc20_addr" \
            --value "$bridge_amount" \
            --bridge-address "$l2_bridge_addr" \
            --rpc-url "$l2_rpc_url" \
            --private-key "$l2_private_key"

    run polycli_bridge_asset_get_info "$output" "$l2_rpc_url" "$l2_bridge_addr"
    if [[ $status -ne 0 ]]; then
        echo "Failed to get deposit info" >&3
        echo "$output" >&3
        exit 1
    fi
    deposit_count=$(echo "$output" | jq -r '.depositCount')
    echo "Deposit count: $deposit_count" >&3

    echo "Claiming second round ERC20 token on L1..." >&3
    # Wait for that exit to settle on L1
    set +e
    polycli ulxly claim asset \
            --bridge-address "$l1_bridge_addr" \
            --private-key "$l1_private_key" \
            --rpc-url "$l1_rpc_url" \
            --deposit-count "$deposit_count" \
            --deposit-network "$network_id" \
            --bridge-service-url "$bridge_service_url" \
            --wait "$claim_wait_duration"
    set -e
    echo "ERC20 token bridge test completed for network: $NETWORK_TARGET" >&3

    echo "Restarting the bridge-spammer service" >&3
    kurtosis service start "$kurtosis_enclave_name" bridge-spammer-001
}

# bats test_tags=bridge,multi-chain
@test "cross-chain bridge between different L2 networks (target:"$NETWORK_TARGET")" {    
    # Define all possible network combinations (source -> target)
    declare -A network_combinations=(
        ["network1"]="network2"
        ["network2"]="network1" 
    )
    
    # Use the combination based on NETWORK_TARGET, or default
    local target_network="${NETWORK_TARGET:-network2}"  
    local source_network="${network_combinations[$target_network]:-network1}"
    
    # Ensure we have valid networks
    if [[ -z "$source_network" ]]; then
        source_network="network1"
    fi
    
    echo "Testing bridge combination: $source_network -> $target_network" >&3
    
    local source_rpc_url
    source_rpc_url=$(get_network_config "$source_network" "rpc_url")
    local source_bridge_service_url
    source_bridge_service_url=$(get_network_config "$source_network" "bridge_service_url")
    local target_rpc_url
    target_rpc_url=$(get_network_config "$target_network" "rpc_url")
    # local target_bridge_service_url
    # target_bridge_service_url=$(get_network_config "$target_network" "bridge_service_url")
    
    # Validate that we can reach both networks
    if [[ -z "$source_rpc_url" || -z "$target_rpc_url" ]]; then
        skip "Required networks ($source_network, $target_network) are not available"
    fi
    
    # Get network IDs
    local source_network_id
    source_network_id=$(cast call --rpc-url "$source_rpc_url" "$l2_bridge_addr" 'networkID()(uint32)')
    local target_network_id
    target_network_id=$(cast call --rpc-url "$target_rpc_url" "$l2_bridge_addr" 'networkID()(uint32)')
    
    echo "Bridging from network $source_network_id to network $target_network_id" >&3
    
    # Bridge from source to target network
    local bridge_amount
    bridge_amount=$(date +%s)
    
    run polycli ulxly bridge asset \
            --destination-network "$target_network_id" \
            --destination-address "$l2_eth_address" \
            --bridge-address "$l2_bridge_addr" \
            --rpc-url "$source_rpc_url" \
            --private-key "$l2_private_key" \
            --value "$bridge_amount"

    run polycli_bridge_asset_get_info "$output" "$source_rpc_url" "$l2_bridge_addr"
    if [[ $status -ne 0 ]]; then
        echo "Failed to get deposit info" >&3
        echo "$output" >&3
        exit 1
    fi
    deposit_count=$(echo "$output" | jq -r '.depositCount')
    
    echo "Claiming bridged L2 token on another L2..." >&3

    # Claim on target network
    set +e
    polycli ulxly claim asset \
            --bridge-address "$l2_bridge_addr" \
            --private-key "$l2_private_key" \
            --rpc-url "$target_rpc_url" \
            --deposit-count "$deposit_count" \
            --deposit-network "$source_network_id" \
            --bridge-service-url "$source_bridge_service_url" \
            --wait "$claim_wait_duration"
    set -e
}