optimistic-mode.bats

#!/usr/bin/env bats
# bats file_tags=op-fep
# shellcheck disable=SC2034

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

    timeout=${TIMEOUT:-3000}
    retry_interval=${RETRY_INTERVAL:-15}

    load "${BATS_TEST_DIRNAME}/../../core/helpers/agglayer-certificates-checks.bash"
}

# Helper function to manage bridge spammer service
manage_bridge_spammer() {
    local action="$1"  # start or stop
    local service="bridge-spammer-001"

    if [[ "$action" == "stop" ]]; then
        if docker ps | grep -q "$service"; then
            echo "Stopping $service..." >&3
            kurtosis service stop "$kurtosis_enclave_name" "$service" || {
                echo "Error: Failed to stop $service" >&3
                return 1
            }
            echo "$service stopped." >&3
        else
            echo "$service does not exist in enclave $kurtosis_enclave_name. Skipping stop operation." >&3
        fi
    elif [[ "$action" == "start" ]]; then
        echo "Starting $service..." >&3
        kurtosis service start "$kurtosis_enclave_name" "$service" || {
            echo "Error: Failed to start $service" >&3
            return 1
        }
        echo "$service started." >&3
    fi
}

# Helper function to toggle optimistic mode
toggle_optimistic_mode() {
    local enabled=$1  # true or false

    local method="enableOptimisticMode"
    [[ "$enabled" == "false" ]] && method="disableOptimisticMode"

    echo "Executing $method..." >&3
    cast send "$rollup_address" "$method()" --rpc-url "$l1_rpc_url" --private-key "$l2_sovereignadmin_private_key" >&3

    local result
    result=$(cast call "$rollup_address" "optimisticMode()(bool)" --rpc-url "$l1_rpc_url")
    if [[ "$result" == "$enabled" ]]; then
        echo "Success: optimisticMode() returned $enabled" >&3
    else
        echo "Error: optimisticMode() did not return $enabled, got $result" >&3
        return 1
    fi
}

# Helper function to check network is using optimisticMode compatible FEP consensus
check_fep_consensus_version() {
    contract_version=$(cast call --rpc-url "$l1_rpc_url" "$rollup_address" "version()(string)")
    if [[ -z $contract_version ]]; then
        echo "It seems like the rollup network is not using FEP consensus contract..." >&3
        return 1
    else
        echo "FEP contract on version: $contract_version" >&3
    fi
}

# bats test_tags=optimistic-mode
@test "Enable OptimisticMode" {
    check_fep_consensus_version

    ensure_non_null_cert >&3
    manage_bridge_spammer "stop"
    print_settlement_info >&3
    wait_for_null_cert >&3

    echo "Checking last settled certificate" >&3
    latest_settled_agglayer_height=$(cast rpc --rpc-url "$(kurtosis port print "$kurtosis_enclave_name" agglayer aglr-readrpc)" interop_getLatestSettledCertificateHeader 1 | jq -r '.height')
    echo "Latest settled L2 height: $latest_settled_agglayer_height" >&3

    cast rpc --rpc-url "$l2_node_url" admin_stopSequencer >stop.out
    kurtosis service stop "$kurtosis_enclave_name" aggkit-001 >&3

    toggle_optimistic_mode "true"

    # Restart sequencer with the same configuration used to stop it
    cast rpc --rpc-url "$l2_node_url" admin_startSequencer "$(cat stop.out)" >&3
    kurtosis service start "$kurtosis_enclave_name" aggkit-001 >&3
    manage_bridge_spammer "start"
}

# bats test_tags=optimistic-mode
@test "Disable OptimisticMode" {
    ensure_non_null_cert >&3
    manage_bridge_spammer "stop"
    print_settlement_info >&3
    wait_for_null_cert >&3

    kurtosis service stop "$kurtosis_enclave_name" aggkit-001 >&3

    toggle_optimistic_mode "false"

    manage_bridge_spammer "start"
    kurtosis service start "$kurtosis_enclave_name" aggkit-001 >&3

    check_height_increase >&3
    print_settlement_info >&3

    wait_for_null_cert >&3
    print_settlement_info >&3
}