unleash-edge 19.2.1

Unleash edge is a proxy for Unleash. It can return both evaluated feature toggles as well as the raw data from Unleash's client API
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
109
110
111
112
113
114
115
116
117

use crate::cli::HealthCheckArgs;
use crate::error::EdgeError;
use crate::tls::build_upstream_certificate;
use reqwest::{ClientBuilder, Url};

fn build_health_url(url: &Url) -> Url {
    let mut with_path = url.clone();
    with_path
        .path_segments_mut()
        .expect("Could not build health check url")
        .push("internal-backstage")
        .push("health");
    with_path
}

pub async fn check_health(health_check_args: HealthCheckArgs) -> Result<(), EdgeError> {
    let client = match build_upstream_certificate(health_check_args.ca_certificate_file)? {
        Some(cert) => ClientBuilder::new()
            .add_root_certificate(cert)
            .build()
            .expect("Failed to build health check client"),
        None => reqwest::Client::default(),
    };
    let base_url = Url::parse(&health_check_args.edge_url)
        .map_err(|p| EdgeError::HealthCheckError(format!("Invalid health check url: {p:?}")))?;
    let health_check_url = build_health_url(&base_url);
    client
        .get(health_check_url)
        .send()
        .await
        .map_err(|e| EdgeError::HealthCheckError(format!("{e:?}")))
        .map(|r| {
            if r.status() == 200 {
                Ok(())
            } else {
                Err(EdgeError::HealthCheckError(
                    "Healthcheck had different status than 200".into(),
                ))
            }
        })?
}

#[cfg(test)]
mod tests {
    use crate::cli::HealthCheckArgs;
    use crate::health_checker::check_health;
    use crate::internal_backstage::health;
    use actix_http::HttpService;
    use actix_http_test::test_server;
    use actix_service::map_config;
    use actix_web::dev::AppConfig;
    use actix_web::{web, App, HttpResponse};

    #[tokio::test]
    pub async fn runs_health_check() {
        let srv = test_server(move || {
            HttpService::new(map_config(
                App::new().service(web::scope("/internal-backstage").service(health)),
                |_| AppConfig::default(),
            ))
            .tcp()
        })
        .await;
        let url = srv.url("/");
        let check_result = check_health(HealthCheckArgs {
            ca_certificate_file: None,
            edge_url: url,
        })
        .await;
        assert!(check_result.is_ok());
    }

    #[tokio::test]
    pub async fn errors_if_health_check_fails() {
        let check_result = check_health(HealthCheckArgs {
            ca_certificate_file: None,
            edge_url: "http://bogusurl".into(),
        })
        .await;
        assert!(check_result.is_err());
    }

    async fn conflict() -> HttpResponse {
        HttpResponse::Conflict().finish()
    }

    #[tokio::test]
    pub async fn errors_if_health_check_returns_different_status_than_200() {
        let srv = test_server(move || {
            HttpService::new(map_config(
                App::new().service(
                    web::scope("/internal-backstage").route("/health", web::get().to(conflict)),
                ),
                |_| AppConfig::default(),
            ))
            .tcp()
        })
        .await;
        let url = srv.url("/");
        let check_result = check_health(HealthCheckArgs {
            ca_certificate_file: None,
            edge_url: url,
        })
        .await;
        assert!(check_result.is_err());
    }

    #[tokio::test]
    pub async fn fails_if_given_an_invalid_url() {
        let check_result = check_health(HealthCheckArgs {
            ca_certificate_file: None,
            edge_url: ":\\///\\/".into(),
        })
        .await;
        assert!(check_result.is_err());
    }
}