Function create_mock_ssv_server

tests/src/mock_ssv.rs:29–60

Creates a simple mock server to simulate the SSV API endpoint under various conditions for testing. Note this ignores

(
    port: u16,
    state: Option<SsvMockState>,
)

/// Creates a simple mock server to simulate the SSV API endpoint under
/// various conditions for testing. Note this ignores
pub async fn create_mock_ssv_server(
    port: u16,
    state: Option<SsvMockState>,
) -> Result<JoinHandle<()>, axum::Error> {
    let data = include_str!("../../tests/data/ssv_valid.json");
    let response = serde_json::from_str::<SSVResponse>(data).expect("failed to parse test data");
    let state = state.unwrap_or(SsvMockState {
        validators: Arc::new(RwLock::new(response.validators)),
        force_timeout: Arc::new(RwLock::new(false)),
    });
    let router = axum::Router::new()
        .route(
            "/api/v4/{chain_name}/validators/in_operator/{node_operator_id}",
            get(handle_validators),
        )
        .route("/big_data", get(handle_big_data))
        .with_state(state)
        .into_make_service();

    let address = SocketAddr::from(([127, 0, 0, 1], port));
    let listener = TcpListener::bind(address).await.map_err(axum::Error::new)?;
    let server = axum::serve(listener, router).with_graceful_shutdown(async {
        tokio::signal::ctrl_c().await.expect("Failed to listen for shutdown signal");
    });
    let result = Ok(tokio::spawn(async move {
        if let Err(e) = server.await {
            eprintln!("Server error: {e}");
        }
    }));
    info!("Mock server started on http://localhost:{port}/");
    result
}

/// Returns a valid SSV validators response, or a timeout if requested in
/// the server state