mirror of
https://github.com/serai-dex/serai.git
synced 2025-12-08 20:29:23 +00:00
43 lines
1.4 KiB
Rust
43 lines
1.4 KiB
Rust
use std::time::Duration;
|
|
|
|
use serai_client_serai::Serai;
|
|
|
|
use dockertest::{StartPolicy, PullPolicy, Image, TestBodySpecification, DockerOperations};
|
|
|
|
pub struct Handle(String);
|
|
|
|
pub fn composition(name: &str, logs_path: String) -> (TestBodySpecification, Handle) {
|
|
let handle = serai_docker_tests::handle(&format!("serai-{name}"));
|
|
serai_docker_tests::build("serai".to_string());
|
|
(
|
|
TestBodySpecification::with_image(
|
|
Image::with_repository("serai-dev-serai").pull_policy(PullPolicy::Never),
|
|
)
|
|
.replace_env(
|
|
[("SERAI_NAME".to_string(), name.to_lowercase()), ("KEY".to_string(), " ".to_string())]
|
|
.into(),
|
|
)
|
|
.set_start_policy(StartPolicy::Strict)
|
|
.set_publish_all_ports(true)
|
|
.set_handle(handle.clone())
|
|
.set_log_options(Some(serai_docker_tests::log_options(logs_path))),
|
|
Handle(handle),
|
|
)
|
|
}
|
|
|
|
pub async fn rpc(ops: &DockerOperations, handle: Handle) -> Serai {
|
|
let serai_rpc = ops.handle(&handle.0).host_port(9944).unwrap();
|
|
let serai_rpc = format!("http://{}:{}", serai_rpc.0, serai_rpc.1);
|
|
|
|
// If the RPC server has yet to start, sleep for up to 5 minutes until it does
|
|
let client = Serai::new(serai_rpc.clone()).unwrap();
|
|
for _ in 0 .. 300 {
|
|
tokio::time::sleep(Duration::from_secs(1)).await;
|
|
if client.block_by_number(0).await.is_err() {
|
|
continue;
|
|
}
|
|
return client;
|
|
}
|
|
panic!("serai RPC server wasn't available after 5 minutes");
|
|
}
|