Commit 8ff8bb35 authored by Victor Colombo's avatar Victor Colombo

Extract wait to custom file

parent 57a6a2de
shared_utils = import_module("github.com/kurtosis-tech/eth2-package/src/shared_utils/shared_utils.star") shared_utils = import_module("github.com/kurtosis-tech/eth2-package/src/shared_utils/shared_utils.star")
parse_input = import_module("github.com/kurtosis-tech/eth2-package/src/package_io/parse_input.star") parse_input = import_module("github.com/kurtosis-tech/eth2-package/src/package_io/parse_input.star")
el_client_context = import_module("github.com/kurtosis-tech/eth2-package/src/participant_network/el/el_client_context.star") el_client_context = import_module("github.com/kurtosis-tech/eth2-package/src/participant_network/el/el_client_context.star")
el_admin_node_info = import_module("github.com/kurtosis-tech/eth2-package/src/participant_network/el/el_admin_node_info.star")
package_io = import_module("github.com/kurtosis-tech/eth2-package/src/package_io/constants.star") package_io = import_module("github.com/kurtosis-tech/eth2-package/src/package_io/constants.star")
# The dirpath of the execution data directory on the client container # The dirpath of the execution data directory on the client container
...@@ -60,23 +60,12 @@ def launch( ...@@ -60,23 +60,12 @@ def launch(
service = add_service(service_id, config) service = add_service(service_id, config)
result_recipe = struct( enode, enr = el_admin_node_info.get_enode_enr_for_node(service_id, RPC_PORT_ID)
service_id = service_id,
method= "POST",
endpoint = "",
body = '{"method":"admin_nodeInfo","params":[],"id":1,"jsonrpc":"2.0"}',
content_type = "application/json",
port_id = RPC_PORT_ID,
extract = {
"enode": ".result.enode",
}
)
response = wait(result_recipe, "extract.enode", "!=", "")
return el_client_context.new_el_client_context( return el_client_context.new_el_client_context(
"besu", "besu",
"", # besu has no ENR "", # besu has no ENR
response["extract.enode"], enode,
service.ip_address, service.ip_address,
RPC_PORT_NUM, RPC_PORT_NUM,
WS_PORT_NUM, WS_PORT_NUM,
......
def get_enode_enr_for_node(service_id, port_id):
recipe = struct(
service_id = service_id,
method= "POST",
endpoint = "",
body = '{"method":"admin_nodeInfo","params":[],"id":1,"jsonrpc":"2.0"}',
content_type = "application/json",
port_id = port_id,
extract = {
"enode": ".result.enode",
"enr": ".result.enr",
}
)
response = wait(recipe, "extract.enode", "!=", "")
return (response["extract.enode"], response["extract.enr"])
\ No newline at end of file
shared_utils = import_module("github.com/kurtosis-tech/eth2-package/src/shared_utils/shared_utils.star") shared_utils = import_module("github.com/kurtosis-tech/eth2-package/src/shared_utils/shared_utils.star")
parse_input = import_module("github.com/kurtosis-tech/eth2-package/src/package_io/parse_input.star") parse_input = import_module("github.com/kurtosis-tech/eth2-package/src/package_io/parse_input.star")
el_admin_node_info = import_module("github.com/kurtosis-tech/eth2-package/src/participant_network/el/el_admin_node_info.star")
el_client_context = import_module("github.com/kurtosis-tech/eth2-package/src/participant_network/el/el_client_context.star") el_client_context = import_module("github.com/kurtosis-tech/eth2-package/src/participant_network/el/el_client_context.star")
package_io = import_module("github.com/kurtosis-tech/eth2-package/src/package_io/constants.star") package_io = import_module("github.com/kurtosis-tech/eth2-package/src/package_io/constants.star")
...@@ -57,24 +58,12 @@ def launch( ...@@ -57,24 +58,12 @@ def launch(
service = add_service(service_id, config) service = add_service(service_id, config)
result_recipe = struct( enode, enr = el_admin_node_info.get_enode_enr_for_node(service_id, RPC_PORT_ID)
service_id = service_id,
method= "POST",
endpoint = "",
body = '{"method":"admin_nodeInfo","params":[],"id":1,"jsonrpc":"2.0"}',
content_type = "application/json",
port_id = RPC_PORT_ID,
extract = {
"enr": ".result.enr",
"enode": ".result.enode",
}
)
response = wait(result_recipe, "extract.enode", "!=", "")
return el_client_context.new_el_client_context( return el_client_context.new_el_client_context(
"erigon", "erigon",
response["extract.enr"], enr,
response["extract.enode"], enode,
service.ip_address, service.ip_address,
RPC_PORT_NUM, RPC_PORT_NUM,
WS_PORT_NUM, WS_PORT_NUM,
......
shared_utils = import_module("github.com/kurtosis-tech/eth2-package/src/shared_utils/shared_utils.star") shared_utils = import_module("github.com/kurtosis-tech/eth2-package/src/shared_utils/shared_utils.star")
parse_input = import_module("github.com/kurtosis-tech/eth2-package/src/package_io/parse_input.star") parse_input = import_module("github.com/kurtosis-tech/eth2-package/src/package_io/parse_input.star")
el_client_context = import_module("github.com/kurtosis-tech/eth2-package/src/participant_network/el/el_client_context.star") el_client_context = import_module("github.com/kurtosis-tech/eth2-package/src/participant_network/el/el_client_context.star")
el_admin_node_info = import_module("github.com/kurtosis-tech/eth2-package/src/participant_network/el/el_admin_node_info.star")
package_io = import_module("github.com/kurtosis-tech/eth2-package/src/package_io/constants.star") package_io = import_module("github.com/kurtosis-tech/eth2-package/src/package_io/constants.star")
...@@ -70,25 +71,12 @@ def launch( ...@@ -70,25 +71,12 @@ def launch(
service = add_service(service_id, config) service = add_service(service_id, config)
result_recipe = struct( enode, enr = el_admin_node_info.get_enode_enr_for_node(service_id, RPC_PORT_ID)
service_id = service_id,
method= "POST",
endpoint = "",
body = '{"method":"admin_nodeInfo","params":[],"id":1,"jsonrpc":"2.0"}',
content_type = "application/json",
port_id = RPC_PORT_ID,
extract = {
"enr": ".result.enr",
"enode": ".result.enode",
}
)
response = wait(result_recipe, "extract.enode", "!=", "")
return el_client_context.new_el_client_context( return el_client_context.new_el_client_context(
"geth", "geth",
response["extract.enr"], enr,
response["extract.enode"], enode,
service.ip_address, service.ip_address,
RPC_PORT_NUM, RPC_PORT_NUM,
WS_PORT_NUM, WS_PORT_NUM,
......
shared_utils = import_module("github.com/kurtosis-tech/eth2-package/src/shared_utils/shared_utils.star") shared_utils = import_module("github.com/kurtosis-tech/eth2-package/src/shared_utils/shared_utils.star")
parse_input = import_module("github.com/kurtosis-tech/eth2-package/src/package_io/parse_input.star") parse_input = import_module("github.com/kurtosis-tech/eth2-package/src/package_io/parse_input.star")
el_client_context = import_module("github.com/kurtosis-tech/eth2-package/src/participant_network/el/el_client_context.star") el_client_context = import_module("github.com/kurtosis-tech/eth2-package/src/participant_network/el/el_client_context.star")
el_admin_node_info = import_module("github.com/kurtosis-tech/eth2-package/src/participant_network/el/el_admin_node_info.star")
package_io = import_module("github.com/kurtosis-tech/eth2-package/src/package_io/constants.star") package_io = import_module("github.com/kurtosis-tech/eth2-package/src/package_io/constants.star")
...@@ -54,24 +55,13 @@ def launch( ...@@ -54,24 +55,13 @@ def launch(
service = add_service(service_id, config) service = add_service(service_id, config)
result_recipe = struct( enode, enr = el_admin_node_info.get_enode_enr_for_node(service_id, RPC_PORT_ID)
service_id = service_id,
method= "POST",
endpoint = "",
body = '{"method":"admin_nodeInfo","params":[],"id":1,"jsonrpc":"2.0"}',
content_type = "application/json",
port_id = RPC_PORT_ID,
extract = {
"enode": ".result.enode",
}
)
response = wait(result_recipe, "extract.enode", "!=", "")
return el_client_context.new_el_client_context( return el_client_context.new_el_client_context(
"nethermind", "nethermind",
"", # nethermind has no ENR in the eth2-merge-kurtosis-module either "", # nethermind has no ENR in the eth2-merge-kurtosis-module either
# Nethermind node info endpoint doesn't return ENR field https://docs.nethermind.io/nethermind/ethereum-client/json-rpc/admin # Nethermind node info endpoint doesn't return ENR field https://docs.nethermind.io/nethermind/ethereum-client/json-rpc/admin
response["extract.enode"], enode,
service.ip_address, service.ip_address,
RPC_PORT_NUM, RPC_PORT_NUM,
WS_PORT_NUM, WS_PORT_NUM,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment