update main.star

This commit is contained in:
0xFugue
2023-08-10 22:20:11 +05:30
parent 9dc3b0c8be
commit df3ce5e879

View File

@@ -6,6 +6,7 @@ prometheus = import_module(vars.PROMETHEUS_MODULE)
grafana = import_module(vars.GRAFANA_MODULE)
args_parser = import_module(vars.ARGUMENT_PARSER_MODULE)
wls = import_module(vars.WLS_MODULE)
collectnet = import_module(vars.COLLECTNET_MODULE)
nodes = import_module(vars.NODE_BUILDERS_MODULE)
assertions = import_module(vars.ASSERTIONS_MODULE)
@@ -36,6 +37,9 @@ def run(plan, args):
if kurtosis_config[vars.INTERCONNECT_NODES]:
nodes.interconnect_nodes(plan, network_topology, interconnection_batch)
# Start topology collection if requested
if kurtosis_config["collectnet"]:
collectnet_service = collectnet.init(plan, network_topology, config_file, prometheus_service)
# Setup WLS & Start the Simulation
if kurtosis_config["injection"]:
wls_service = wls.init(plan, network_topology, config_file, prometheus_service)
@@ -43,3 +47,4 @@ def run(plan, args):
# Tests
if kurtosis_config["testing"]:
assertions.start_test(plan, kurtosis_config, network_topology)