logos-blockchain/logos-blockchain-testing
1
0
Fork 0
mirror of https://github.com/logos-blockchain/logos-blockchain-testing.git synced 2026-01-02 13:23:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add topology_with helper and format scenario builder usage

Browse Source
This commit is contained in:
andrussal 2025-12-03 05:46:32 +01:00
parent dca7d79393
commit c0555ca347
5 changed files with 27 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
examples/src/bin/compose_runner.rs
View File

@ -43,11 +43,11 @@ async fn run_compose_case(
"building scenario plan"
);
let mut plan = ScenarioBuilder::topology()
.network_star()
.validators(validators)
.executors(executors)
.apply()
let mut plan = ScenarioBuilder::topology_with(|t| {
t.network_star()
.validators(validators)
.executors(executors)
})
.enable_node_control()
.chaos()
.restart()

10
examples/src/bin/k8s_runner.rs
View File

@ -39,11 +39,11 @@ async fn run_k8s_case(
duration_secs = run_duration.as_secs(),
"building scenario plan"
);
let mut plan = ScenarioBuilder::topology()
.network_star()
.validators(validators)
.executors(executors)
.apply()
let mut plan = ScenarioBuilder::topology_with(|t| {
t.network_star()
.validators(validators)
.executors(executors)
})
.wallets(TOTAL_WALLETS)
.transactions()
.rate(MIXED_TXS_PER_BLOCK)

10
examples/src/bin/local_runner.rs
View File

@ -47,11 +47,11 @@ async fn run_local_case(
duration_secs = run_duration.as_secs(),
"building scenario plan"
);
let mut plan = ScenarioBuilder::topology()
.network_star()
.validators(validators)
.executors(executors)
.apply()
let mut plan = ScenarioBuilder::topology_with(|t| {
t.network_star()
.validators(validators)
.executors(executors)
})
.wallets(TOTAL_WALLETS)
.transactions()
.rate(MIXED_TXS_PER_BLOCK)

9
testing-framework/core/src/scenario/definition.rs
View File

@ -115,6 +115,15 @@ impl<Caps: Default> Builder<Caps> {
pub fn topology() -> TopologyConfigurator<Caps> {
TopologyConfigurator::new(Self::new(TopologyBuilder::new(TopologyConfig::empty())))
}
/// Configure topology via a closure and return the scenario builder.
#[must_use]
pub fn topology_with(
f: impl FnOnce(TopologyConfigurator<Caps>) -> TopologyConfigurator<Caps>,
) -> Builder<Caps> {
let configurator = Self::topology();
f(configurator).apply()
}
}
impl<Caps> Builder<Caps> {

18
testing-framework/runners/compose/src/deployer.rs
View File

@ -302,11 +302,7 @@ mod tests {
#[test]
fn cfgsync_prebuilt_configs_preserve_genesis() {
let scenario = ScenarioBuilder::topology()
.validators(1)
.executors(1)
.apply()
.build();
let scenario = ScenarioBuilder::topology_with(|t| t.validators(1).executors(1)).build();
let topology = scenario.topology().clone();
let hosts = hosts_from_topology(&topology);
let tracing_settings = tracing_settings(&topology);
@ -358,11 +354,7 @@ mod tests {
#[test]
fn cfgsync_genesis_proofs_verify_against_ledger() {
let scenario = ScenarioBuilder::topology()
.validators(1)
.executors(1)
.apply()
.build();
let scenario = ScenarioBuilder::topology_with(|t| t.validators(1).executors(1)).build();
let topology = scenario.topology().clone();
let hosts = hosts_from_topology(&topology);
let tracing_settings = tracing_settings(&topology);
@ -398,11 +390,7 @@ mod tests {
#[test]
fn cfgsync_docker_overrides_produce_valid_genesis() {
let scenario = ScenarioBuilder::topology()
.validators(1)
.executors(1)
.apply()
.build();
let scenario = ScenarioBuilder::topology_with(|t| t.validators(1).executors(1)).build();
let topology = scenario.topology().clone();
let tracing_settings = tracing_settings(&topology);
let hosts = docker_style_hosts(&topology);