wasmcloud-test-util 0.19.0

Utilities for testing wasmCloud hosts, providers, and components. 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

use std::env;

use anyhow::{Context, Result};
use async_nats::connection::State;
use async_nats::Client as NatsClient;
use tokio::process::Command;
use tokio::time::{sleep, timeout, Duration};
use url::Url;

use super::{free_port, tempdir, BackgroundServer};

pub async fn start_nats() -> Result<(BackgroundServer, Url, NatsClient)> {
    let port = free_port().await?;
    let url =
        Url::parse(&format!("nats://localhost:{port}")).context("failed to parse NATS URL")?;
    let jetstream_dir = tempdir()?;
    let server = BackgroundServer::spawn(
        Command::new(
            env::var("TEST_NATS_BIN")
                .as_deref()
                .unwrap_or("nats-server"),
        )
        .args([
            "-js",
            "-D",
            "-T=false",
            "-p",
            &port.to_string(),
            "-sd",
            jetstream_dir.path().display().to_string().as_str(),
        ]),
    )
    .await
    .context("failed to start NATS")?;

    // Wait until nats is ready to take connections
    let nats_client = async_nats::connect_with_options(
        url.as_str(),
        async_nats::ConnectOptions::new().retry_on_initial_connect(),
    )
    .await
    .context("failed to build nats client")?;
    let nats_client = timeout(Duration::from_secs(3), async move {
        loop {
            if nats_client.connection_state() == State::Connected {
                return nats_client;
            }
            sleep(Duration::from_millis(100)).await;
        }
    })
    .await
    .context("failed to ensure connection to NATS server")?;

    Ok((server, url, nats_client))
}