mirror of
https://github.com/logos-blockchain/logos-blockchain-testing.git
synced 2026-02-17 19:53:05 +00:00
feat(local-deployer): add per-node readiness wait for manual clusters
This commit is contained in:
parent
6c7b66db02
commit
514cae9e89
@ -73,6 +73,11 @@ impl<E: LocalDeployerEnv> ManualCluster<E> {
|
||||
pub async fn wait_network_ready(&self) -> Result<(), ReadinessError> {
|
||||
self.nodes.wait_network_ready().await
|
||||
}
|
||||
|
||||
pub async fn wait_node_ready(&self, name: &str) -> Result<(), ManualClusterError> {
|
||||
self.nodes.wait_node_ready(name).await?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
impl<E: LocalDeployerEnv> Drop for ManualCluster<E> {
|
||||
|
||||
@ -51,6 +51,11 @@ pub enum NodeManagerError {
|
||||
#[source]
|
||||
source: DynError,
|
||||
},
|
||||
#[error("failed readiness check: {source}")]
|
||||
Readiness {
|
||||
#[source]
|
||||
source: ReadinessError,
|
||||
},
|
||||
}
|
||||
|
||||
pub struct NodeManager<E: LocalDeployerEnv> {
|
||||
@ -191,6 +196,31 @@ impl<E: LocalDeployerEnv> NodeManager<E> {
|
||||
wait_for_http_ports(&ports, E::readiness_endpoint_path()).await
|
||||
}
|
||||
|
||||
pub async fn wait_node_ready(&self, name: &str) -> Result<(), NodeManagerError> {
|
||||
let port = {
|
||||
let state = self.lock_state();
|
||||
let index =
|
||||
*state
|
||||
.indices_by_name
|
||||
.get(name)
|
||||
.ok_or_else(|| NodeManagerError::NodeName {
|
||||
name: name.to_string(),
|
||||
})?;
|
||||
|
||||
state
|
||||
.nodes
|
||||
.get(index)
|
||||
.map(|node| node.endpoints().api.port())
|
||||
.ok_or_else(|| NodeManagerError::NodeName {
|
||||
name: name.to_string(),
|
||||
})?
|
||||
};
|
||||
|
||||
wait_for_http_ports(&[port], E::readiness_endpoint_path())
|
||||
.await
|
||||
.map_err(|source| NodeManagerError::Readiness { source })
|
||||
}
|
||||
|
||||
pub async fn start_node_with(
|
||||
&self,
|
||||
name: &str,
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user