Update topology imports to new module structure

This commit is contained in:
andrussal 2025-12-10 08:29:41 +01:00
parent 253b0e7a0e
commit a76630d3e1
27 changed files with 40 additions and 41 deletions

View File

@ -18,7 +18,7 @@ use async_trait::async_trait;
use testing_framework_core::scenario::{
DynError, Expectation, RunContext, RunMetrics, Workload,
};
use testing_framework_core::topology::GeneratedTopology;
use testing_framework_core::topology::generation::GeneratedTopology;
pub struct ReachabilityWorkload {
target_idx: usize,

View File

@ -25,7 +25,7 @@
workloads can rely on consistent signals.
## Adding topology helpers
- Extend `testing_framework_core::topology::TopologyBuilder` with new layouts or
- Extend `testing_framework_core::topology::config::TopologyBuilder` with new layouts or
configuration presets (e.g., specialized DA parameters). Keep defaults safe:
ensure at least one participant and clamp dispersal factors as the current
helpers do.

View File

@ -9,7 +9,7 @@ use serde_with::serde_as;
use crate::{
constants::kzg_container_path,
topology::{GeneratedTopology, configs::wallet::WalletConfig},
topology::{configs::wallet::WalletConfig, generation::GeneratedTopology},
};
#[serde_as]

View File

@ -5,8 +5,9 @@ use super::{
workload::Workload,
};
use crate::topology::{
GeneratedTopology, TopologyBuilder, TopologyConfig,
config::{TopologyBuilder, TopologyConfig},
configs::{network::Libp2pNetworkLayout, wallet::WalletConfig},
generation::GeneratedTopology,
};
const DEFAULT_FUNDS_PER_WALLET: u64 = 100;

View File

@ -1,7 +1,7 @@
use async_trait::async_trait;
use super::{DynError, RunContext, runtime::context::RunMetrics};
use crate::topology::GeneratedTopology;
use crate::topology::generation::GeneratedTopology;
#[async_trait]
/// Defines a check evaluated during or after a scenario run.

View File

@ -4,7 +4,9 @@ use super::{block_feed::BlockFeed, metrics::Metrics, node_clients::ClusterClient
use crate::{
nodes::ApiClient,
scenario::{NodeClients, NodeControlHandle},
topology::{GeneratedTopology, Topology, configs::wallet::WalletAccount},
topology::{
configs::wallet::WalletAccount, deployment::Topology, generation::GeneratedTopology,
},
};
/// Shared runtime context available to workloads and expectations.

View File

@ -5,7 +5,7 @@ use rand::{Rng as _, seq::SliceRandom as _, thread_rng};
use crate::{
nodes::ApiClient,
scenario::DynError,
topology::{GeneratedTopology, Topology},
topology::{deployment::Topology, generation::GeneratedTopology},
};
/// Collection of API clients for the validator and executor set.

View File

@ -1,7 +1,7 @@
use async_trait::async_trait;
use super::{DynError, Expectation, RunContext, runtime::context::RunMetrics};
use crate::topology::GeneratedTopology;
use crate::topology::generation::GeneratedTopology;
#[async_trait]
/// Describes an action sequence executed during a scenario run.

View File

@ -20,9 +20,9 @@ use testing_framework_config::topology::configs::{
};
use crate::topology::{
GeneratedNodeConfig, GeneratedTopology, NodeRole,
configs::{GeneralConfig, time::default_time_config},
create_kms_configs, resolve_ids, resolve_ports,
generation::{GeneratedNodeConfig, GeneratedTopology, NodeRole},
utils::{create_kms_configs, resolve_ids, resolve_ports},
};
/// High-level topology settings used to generate node configs for a scenario.

View File

@ -8,7 +8,7 @@ use crate::{
validator::{Validator, create_validator_config},
},
topology::{
TopologyBuilder, TopologyConfig,
config::{TopologyBuilder, TopologyConfig},
configs::GeneralConfig,
generation::find_expected_peer_counts,
readiness::{

View File

@ -3,8 +3,9 @@ use std::{collections::HashSet, time::Duration};
use reqwest::{Client, Url};
use crate::topology::{
Topology, TopologyConfig,
config::TopologyConfig,
configs::{GeneralConfig, wallet::WalletAccount},
deployment::Topology,
readiness::{HttpMembershipReadiness, HttpNetworkReadiness, ReadinessCheck, ReadinessError},
};

View File

@ -2,17 +2,8 @@ pub mod configs {
pub use testing_framework_config::topology::configs::*;
}
mod config;
mod deployment;
pub mod config;
pub mod deployment;
pub mod generation;
pub mod readiness;
mod utils;
pub use config::{TopologyBuilder, TopologyConfig};
pub use deployment::Topology;
pub use generation::{GeneratedNodeConfig, GeneratedTopology, NodeRole, find_expected_peer_counts};
pub use readiness::{
DaBalancerReadiness, HttpMembershipReadiness, HttpNetworkReadiness, MembershipReadiness,
NetworkReadiness, ReadinessCheck, ReadinessError,
};
pub use utils::{create_kms_configs, multiaddr_port, resolve_ids, resolve_ports};
pub mod utils;

View File

@ -1,7 +1,7 @@
use nomos_da_network_core::swarm::BalancerStats;
use super::ReadinessCheck;
use crate::topology::Topology;
use crate::topology::deployment::Topology;
pub struct DaBalancerReadiness<'a> {
pub(crate) topology: &'a Topology,

View File

@ -3,7 +3,7 @@ use nomos_da_network_service::MembershipResponse;
use reqwest::{Client, Url};
use super::ReadinessCheck;
use crate::topology::Topology;
use crate::topology::deployment::Topology;
pub struct MembershipReadiness<'a> {
pub(crate) topology: &'a Topology,

View File

@ -3,7 +3,7 @@ use reqwest::{Client, Url};
use tracing::warn;
use super::ReadinessCheck;
use crate::topology::Topology;
use crate::topology::deployment::Topology;
pub struct NetworkReadiness<'a> {
pub(crate) topology: &'a Topology,

View File

@ -2,7 +2,7 @@ use std::{path::Path, process::Command as StdCommand};
use testing_framework_core::{
scenario::cfgsync::{apply_topology_overrides, load_cfgsync_template, write_cfgsync_template},
topology::GeneratedTopology,
topology::generation::GeneratedTopology,
};
/// Handle that tracks a cfgsync server started for compose runs.

View File

@ -11,7 +11,7 @@ use tera::Context as TeraContext;
use testing_framework_core::{
adjust_timeout,
constants::{DEFAULT_CFGSYNC_PORT, kzg_container_path},
topology::{GeneratedNodeConfig, GeneratedTopology},
topology::generation::{GeneratedNodeConfig, GeneratedTopology},
};
use tokio::{process::Command, time::timeout};
@ -611,7 +611,7 @@ fn handle_compose_status(
#[cfg(test)]
mod tests {
use testing_framework_core::topology::{TopologyBuilder, TopologyConfig};
use testing_framework_core::topology::config::{TopologyBuilder, TopologyConfig};
use super::*;

View File

@ -10,7 +10,7 @@ use testing_framework_core::{
BlockFeed, BlockFeedTask, CleanupGuard, Deployer, NodeClients, NodeControlHandle,
RequiresNodeControl, RunContext, Runner, Scenario,
},
topology::GeneratedTopology,
topology::generation::GeneratedTopology,
};
use tracing::{debug, info};
@ -296,7 +296,9 @@ mod tests {
use nomos_tracing_service::TracingSettings;
use testing_framework_core::{
scenario::ScenarioBuilder,
topology::{GeneratedNodeConfig, GeneratedTopology, NodeRole as TopologyNodeRole},
topology::generation::{
GeneratedNodeConfig, GeneratedTopology, NodeRole as TopologyNodeRole,
},
};
use zksign::PublicKey;

View File

@ -5,7 +5,9 @@ use std::{
};
use anyhow::anyhow;
use testing_framework_core::{adjust_timeout, scenario::CleanupGuard, topology::GeneratedTopology};
use testing_framework_core::{
adjust_timeout, scenario::CleanupGuard, topology::generation::GeneratedTopology,
};
use tokio::process::Command;
use tracing::{debug, info, warn};
use uuid::Uuid;

View File

@ -5,7 +5,7 @@ use testing_framework_core::{
MetricsError,
http_probe::{HttpReadinessError, NodeRole},
},
topology::ReadinessError,
topology::readiness::ReadinessError,
};
use url::ParseError;

View File

@ -5,7 +5,7 @@ use reqwest::Url;
use testing_framework_core::{
adjust_timeout,
scenario::http_probe::NodeRole as HttpNodeRole,
topology::{GeneratedTopology, NodeRole as TopologyNodeRole},
topology::generation::{GeneratedTopology, NodeRole as TopologyNodeRole},
};
use tokio::{process::Command, time::timeout};
use url::ParseError;

View File

@ -4,7 +4,7 @@ use reqwest::Url;
use testing_framework_core::{
nodes::ApiClient,
scenario::{Metrics, MetricsError, NodeClients, http_probe::NodeRole as HttpNodeRole},
topology::{GeneratedTopology, NodeRole as TopologyNodeRole},
topology::generation::{GeneratedTopology, NodeRole as TopologyNodeRole},
};
use tokio::time::sleep;

View File

@ -10,7 +10,7 @@ use tempfile::TempDir;
use testing_framework_core::{
constants::cfgsync_port,
scenario::cfgsync::{apply_topology_overrides, load_cfgsync_template, render_cfgsync_yaml},
topology::GeneratedTopology,
topology::generation::GeneratedTopology,
};
use thiserror::Error;

View File

@ -5,7 +5,7 @@ use reqwest::Url;
use testing_framework_core::{
nodes::ApiClient,
scenario::{CleanupGuard, Metrics, MetricsError, NodeClients, http_probe::NodeRole},
topology::{GeneratedTopology, ReadinessError},
topology::{generation::GeneratedTopology, readiness::ReadinessError},
};
use tracing::info;
use url::ParseError;

View File

@ -3,7 +3,7 @@ use async_trait::async_trait;
use kube::Client;
use testing_framework_core::{
scenario::{BlockFeedTask, CleanupGuard, Deployer, MetricsError, RunContext, Runner, Scenario},
topology::GeneratedTopology,
topology::generation::GeneratedTopology,
};
use tracing::{error, info};

View File

@ -4,7 +4,7 @@ use testing_framework_core::{
BlockFeed, BlockFeedTask, Deployer, DynError, Metrics, NodeClients, RunContext, Runner,
Scenario, ScenarioError, spawn_block_feed,
},
topology::{ReadinessError, Topology},
topology::{deployment::Topology, readiness::ReadinessError},
};
use thiserror::Error;
use tracing::info;

View File

@ -12,7 +12,7 @@ use nomos_core::mantle::{
use testing_framework_config::topology::configs::wallet::WalletAccount;
use testing_framework_core::{
scenario::{DynError, Expectation, RunContext, RunMetrics, Workload as ScenarioWorkload},
topology::{GeneratedNodeConfig, GeneratedTopology},
topology::generation::{GeneratedNodeConfig, GeneratedTopology},
};
use tokio::time::sleep;
use zksign::{PublicKey, SecretKey};