hoc/tests/util/mod.rs

use hoc::{config::Settings, telemetry};

use std::net::TcpListener;

use tempfile::{tempdir, TempDir};

lazy_static::lazy_static! {
    static ref TRACING: () = {
        let filter = if std::env::var("TEST_LOG").is_ok() { "debug" } else { "" };
        let subscriber = telemetry::get_subscriber("test", filter);
        telemetry::init_subscriber(subscriber);
    };
}

pub struct TestApp {
    pub address: String,
    // Those are unused but are hold to be dropped and deleted after the TestApp goes out of scope
    _repo_dir: TempDir,
    _cache_dir: TempDir,
}

pub async fn spawn_app() -> TestApp {
    lazy_static::initialize(&TRACING);

    let listener = TcpListener::bind("127.0.0.1:0").expect("Failed to bind random port");

    let port = listener.local_addr().unwrap().port();
    let address = format!("http://127.0.0.1:{}", port);

    let _repo_dir = tempdir().expect("Cannot create repo_dir");
    let _cache_dir = tempdir().expect("Cannot create cache_dir");

    let mut settings = Settings::load().expect("Failed to read configuration.");
    settings.repodir = _repo_dir.path().to_path_buf();
    settings.cachedir = _cache_dir.path().to_path_buf();

    let server = hoc::run(listener, settings)
        .await
        .expect("Failed to bind address");

    let _ = tokio::spawn(server);

    TestApp {
        address,
        _repo_dir,
        _cache_dir,
    }
}