Add kairos server test hook

flake.nix

@@ -41,7 +41,7 @@
       ];
       perSystem = { config, self', inputs', system, pkgs, lib, ... }:
         let
-          rustToolchain = inputs'.fenix.packages.stable.toolchain;
+          rustToolchain = inputs'.fenix.packages.complete.toolchain;
           craneLib = inputs.crane.lib.${system}.overrideToolchain rustToolchain;
 
           kairosNodeAttrs = {

kairos-test-utils/Cargo.toml

@@ -16,4 +16,5 @@ nom = "7"
 tokio = { version = "1", features = [ "full", "tracing", "macros" ] }
 tempfile = "3"
 tracing = "0.1"
+reqwest = { version = "*", features = ["json"] }
 

kairos-test-utils/src/cctl.rs

@@ -1,6 +1,6 @@
 pub mod parsers;
 use anyhow::anyhow;
-use backoff::{self, backoff::Constant, future::retry};
+use backoff::{backoff::Constant, future::retry};
 use casper_client::{get_node_status, rpcs::results::ReactorState, Error, JsonRpcId, Verbosity};
 use std::io::{self, Write};
 use std::path::PathBuf;
@@ -133,7 +133,6 @@ impl Drop for CCTLNetwork {
 #[cfg(test)]
 mod tests {
     use super::*;
-    use casper_client::{get_node_status, rpcs::results::ReactorState, JsonRpcId, Verbosity};
     #[tokio::test]
     async fn test_cctl_network_starts_and_terminates() {
         let network = CCTLNetwork::run().await.unwrap();

kairos-test-utils/src/kairos.rs

@@ -0,0 +1,68 @@
+use backoff::future::retry;
+use backoff::ExponentialBackoff;
+use reqwest::Url;
+use std::env;
+use std::io;
+use std::net::{SocketAddr, TcpListener};
+use std::path::PathBuf;
+use std::process::{Child, Command};
+use tokio::net::TcpStream;
+
+// A hacky way to get the cargo binary directory path
+pub fn bin_dir() -> PathBuf {
+    let mut path = env::current_exe().unwrap();
+    path.pop(); // pop kairos_test_utils-hash
+    path.pop(); // pop deps
+    path
+}
+
+async fn wait_for_port(address: &SocketAddr) -> Result<(), io::Error> {
+    retry(ExponentialBackoff::default(), || async {
+        Ok(TcpStream::connect(address).await.map(|_| ())?)
+    })
+    .await
+}
+
+pub struct Kairos {
+    pub url: Url,
+    process_handle: Child,
+}
+
+impl Kairos {
+    pub async fn run() -> Result<Kairos, io::Error> {
+        let port = TcpListener::bind("127.0.0.1:0")?
+            .local_addr()?
+            .port()
+            .to_string();
+        let url = Url::parse(format!("http://127.0.0.1:{}", port).as_str()).unwrap();
+        let kairos = bin_dir().join("kairos-server");
+        let process_handle = Command::new(kairos)
+            .env("KAIROS_SERVER_PORT", &port)
+            .spawn()
+            .expect("Failed to start the kairos-server");
+
+        wait_for_port(url.socket_addrs(|| Option::None).unwrap().first().unwrap())
+            .await
+            .unwrap();
+
+        Ok(Kairos {
+            url,
+            process_handle,
+        })
+    }
+}
+
+impl Drop for Kairos {
+    fn drop(&mut self) {
+        let _ = self.process_handle.kill();
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    #[tokio::test]
+    async fn test_kairos_starts_and_terminates() {
+        let _kairos = Kairos::run().await.unwrap();
+    }
+}

kairos-test-utils/src/lib.rs

@@ -1 +1,3 @@
+#![feature(async_closure)]
 pub mod cctl;
+pub mod kairos;