lighthouse/beacon_node/http_metrics/tests/tests.rs

use beacon_chain::test_utils::BlockingMigratorEphemeralHarnessType;
use environment::null_logger;
use http_metrics::Config;
use reqwest::StatusCode;
use std::net::Ipv4Addr;
use std::sync::Arc;
use tokio::sync::oneshot;
use types::MainnetEthSpec;

type Context = http_metrics::Context<BlockingMigratorEphemeralHarnessType<MainnetEthSpec>>;

#[tokio::test(core_threads = 2)]
async fn returns_200_ok() {
    let log = null_logger().unwrap();

    let context = Arc::new(Context {
        config: Config {
            enabled: true,
            listen_addr: Ipv4Addr::new(127, 0, 0, 1),
            listen_port: 0,
            allow_origin: None,
        },
        chain: None,
        db_path: None,
        freezer_db_path: None,
        log,
    });

    let ctx = context.clone();
    let (_shutdown_tx, shutdown_rx) = oneshot::channel::<()>();
    let server_shutdown = async {
        // It's not really interesting why this triggered, just that it happened.
        let _ = shutdown_rx.await;
    };
    let (listening_socket, server) = http_metrics::serve(ctx, server_shutdown).unwrap();

    tokio::spawn(async { server.await });

    let url = format!(
        "http://{}:{}/metrics",
        listening_socket.ip(),
        listening_socket.port()
    );

    assert_eq!(reqwest::get(&url).await.unwrap().status(), StatusCode::OK);
}