test_kms_server 5.23.0

Run a test KMS server for testing purposes
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

//! Validates that the KMS server serves the `OpenAPI` specification correctly.

use crate::{init_test_logging, start_default_test_kms_server};

/// Verify that `GET /openapi.yaml` returns a valid `OpenAPI` 3.1 document.
#[tokio::test]
async fn test_openapi_yaml_endpoint() -> Result<(), String> {
    init_test_logging();
    let ctx = start_default_test_kms_server().await;

    let client = reqwest::Client::builder()
        .danger_accept_invalid_certs(true)
        .build()
        .map_err(|e| format!("failed to build reqwest client: {e}"))?;

    let url = format!(
        "{}/openapi.yaml",
        ctx.owner_client_config.http_config.server_url
    );
    let response = client
        .get(&url)
        .send()
        .await
        .map_err(|e| format!("GET /openapi.yaml failed: {e}"))?;

    if response.status() != 200 {
        return Err(format!("expected status 200, got {}", response.status()));
    }

    let content_type = response
        .headers()
        .get("content-type")
        .ok_or("missing content-type header")?
        .to_str()
        .map_err(|e| format!("invalid content-type header: {e}"))?;

    if !content_type.contains("application/yaml") {
        return Err(format!("unexpected content-type: {content_type}"));
    }

    let body = response
        .text()
        .await
        .map_err(|e| format!("failed to read body: {e}"))?;

    if !body.contains("openapi: 3.1.0") {
        return Err("missing openapi version".to_owned());
    }
    if !body.contains("title: Cosmian KMS") {
        return Err("missing title".to_owned());
    }
    if body.len() <= 50_000 {
        return Err(format!("spec body too small: {} bytes", body.len()));
    }

    Ok(())
}

/// Verify that `GET /swagger` returns the Swagger UI HTML page.
#[tokio::test]
async fn test_swagger_ui_endpoint() -> Result<(), String> {
    init_test_logging();
    let ctx = start_default_test_kms_server().await;

    let client = reqwest::Client::builder()
        .danger_accept_invalid_certs(true)
        .build()
        .map_err(|e| format!("failed to build reqwest client: {e}"))?;

    let url = format!("{}/swagger", ctx.owner_client_config.http_config.server_url);
    let response = client
        .get(&url)
        .send()
        .await
        .map_err(|e| format!("GET /swagger failed: {e}"))?;

    if response.status() != 200 {
        return Err(format!("expected status 200, got {}", response.status()));
    }

    let content_type = response
        .headers()
        .get("content-type")
        .ok_or("missing content-type header")?
        .to_str()
        .map_err(|e| format!("invalid content-type header: {e}"))?;

    if !content_type.contains("text/html") {
        return Err(format!("unexpected content-type: {content_type}"));
    }

    let body = response
        .text()
        .await
        .map_err(|e| format!("failed to read body: {e}"))?;

    if !body.contains("/openapi.yaml") {
        return Err("HTML should reference /openapi.yaml".to_owned());
    }
    if !body.contains("swagger-ui") {
        return Err("HTML should contain swagger-ui div".to_owned());
    }
    if !body.contains("Cosmian KMS") {
        return Err("HTML should contain page title".to_owned());
    }

    Ok(())
}