81c667b58e
Adds additional metrics for network monitoring and evaluation. Co-authored-by: Mark Mackey <mark@sigmaprime.io>
52 lines
1.5 KiB
Rust
52 lines
1.5 KiB
Rust
use beacon_chain::test_utils::EphemeralHarnessType;
|
|
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<EphemeralHarnessType<MainnetEthSpec>>;
|
|
|
|
#[tokio::test(flavor = "multi_thread", worker_threads = 2)]
|
|
async fn returns_200_ok() {
|
|
async {
|
|
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,
|
|
allocator_metrics_enabled: true,
|
|
},
|
|
chain: None,
|
|
db_path: None,
|
|
freezer_db_path: None,
|
|
gossipsub_registry: 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);
|
|
}
|
|
.await
|
|
}
|