perfgate-server 0.15.1

REST API server for centralized baseline management
Documentation 
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
56
57
58
59
60
61

//! Testing utilities for perfgate-server.
//!
//! This module is only available when compiled for tests.

use std::sync::Arc;
use tokio::net::TcpListener;

use crate::auth::AuthState;
use crate::server::{
    AppState, ServerConfig, create_fleet_store, create_key_store, create_router, create_storage,
};
use crate::storage::{InMemoryKeyStore, KeyStore};

/// A running test server.
pub struct TestServer {
    /// The base URL of the server (e.g., "http://127.0.0.1:1234/api/v1")
    pub url: String,
    /// The root URL of the server (e.g., "http://127.0.0.1:1234")
    pub root_url: String,
    /// Join handle for the server task
    pub handle: tokio::task::JoinHandle<()>,
}

/// Spawns a real perfgate server on a random port for testing.
pub async fn spawn_test_server(config: ServerConfig) -> TestServer {
    let (store, audit) = create_storage(&config)
        .await
        .expect("failed to create storage");
    let key_store = create_key_store(&config)
        .await
        .expect("failed to create key store");
    let persistent_key_store: Arc<dyn KeyStore> = Arc::new(InMemoryKeyStore::new());
    let fleet_store = create_fleet_store();
    let auth_state = AuthState::new(key_store, config.jwt.clone(), Default::default())
        .with_persistent_key_store(persistent_key_store.clone());
    let app_state = AppState { store, audit };
    let app = create_router(
        app_state,
        persistent_key_store,
        fleet_store,
        None,
        auth_state,
        &config,
        None,
    );

    let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
    let addr = listener.local_addr().unwrap();
    let root_url = format!("http://{}", addr);
    let url = format!("{}/api/v1", root_url);

    let handle = tokio::spawn(async move {
        axum::serve(listener, app).await.unwrap();
    });

    TestServer {
        url,
        root_url,
        handle,
    }
}