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

Make scenario source transitions explicit

Browse Source
This commit is contained in:
andrussal 2026-03-08 13:44:25 +01:00
parent 6888c18275
commit 3f8e287c68
2 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -569,13 +569,13 @@ impl<E: Application, Caps> Builder<E, Caps> {
#[must_use] #[must_use]
pub fn with_attach_source(mut self, attach: AttachSource) -> Self { pub fn with_attach_source(mut self, attach: AttachSource) -> Self {
self.sources.set_attach(attach); self.sources = self.sources.with_attach(attach);
self self
} }
#[must_use] #[must_use]
pub fn with_external_node(mut self, node: ExternalNodeSource) -> Self { pub fn with_external_node(mut self, node: ExternalNodeSource) -> Self {
self.sources.add_external_node(node); self.sources = self.sources.with_external_node(node);
self self
} }
@ -591,7 +591,7 @@ impl<E: Application, Caps> Builder<E, Caps> {
#[must_use] #[must_use]
pub fn with_external_only_sources(mut self) -> Self { pub fn with_external_only_sources(mut self) -> Self {
self.sources.set_external_only(); self.sources = self.sources.into_external_only();
self self
} }

19
testing-framework/core/src/scenario/sources/model.rs
View File

@ -128,22 +128,29 @@ impl ScenarioSources {
Self::ExternalOnly { external } Self::ExternalOnly { external }
} }
pub(crate) fn add_external_node(&mut self, node: ExternalNodeSource) { #[must_use]
match self { pub fn with_external_node(mut self, node: ExternalNodeSource) -> Self {
match &mut self {
Self::Managed { external } Self::Managed { external }
| Self::Attached { external, .. } | Self::Attached { external, .. }
| Self::ExternalOnly { external } => external.push(node), | Self::ExternalOnly { external } => external.push(node),
} }
self
} }
pub(crate) fn set_attach(&mut self, attach: AttachSource) { #[must_use]
pub fn with_attach(self, attach: AttachSource) -> Self {
let external = self.external_nodes().to_vec(); let external = self.external_nodes().to_vec();
*self = Self::Attached { attach, external };
Self::Attached { attach, external }
} }
pub(crate) fn set_external_only(&mut self) { #[must_use]
pub fn into_external_only(self) -> Self {
let external = self.external_nodes().to_vec(); let external = self.external_nodes().to_vec();
*self = Self::ExternalOnly { external };
Self::ExternalOnly { external }
} }
#[must_use] #[must_use]