proby 1.0.2

Check whether hosts are reachable on certain ports and return result on HTTP
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

mod utils;

use surf;

use utils::{Error, ProbyProcess};

/// If querying a connectable service, Proby returns 200 by default.
#[actix_rt::test]
async fn connectable_service() -> Result<(), Error> {
    let mut dh = ProbyProcess::new(Vec::<String>::new())?;

    let mut resp_body = surf::get(&dh.selfcheck()).await?;
    assert_eq!(resp_body.status(), 200);
    assert_eq!(
        resp_body.body_string().await?,
        format!("{}:{} is connectable", dh.host, dh.port)
    );

    dh.child.kill()?;

    Ok(())
}

/// If querying an unconnectable service, Proby returns 503 by default.
#[actix_rt::test]
async fn unconnectable_service() -> Result<(), Error> {
    let mut dh = ProbyProcess::new(Vec::<String>::new())?;

    // Generate a URL that's not connectable.
    let url = format!("{}/{}:{}", dh.url, dh.host, 1);
    let mut resp_body = surf::get(&url).await?;
    assert_eq!(resp_body.status(), 503);
    assert_eq!(
        resp_body.body_string().await?,
        format!("{}:{} is NOT connectable", dh.host, 1)
    );

    dh.child.kill()?;

    Ok(())
}

/// We can set a different good status code.
#[actix_rt::test]
async fn can_configure_good_status_code() -> Result<(), Error> {
    let mut dh = ProbyProcess::new(Vec::<String>::new())?;

    let url = format!("{}?good=201", dh.selfcheck());
    let mut resp_body = surf::get(url).await?;
    assert_eq!(resp_body.status(), 201);
    assert_eq!(
        resp_body.body_string().await?,
        format!("{}:{} is connectable", dh.host, dh.port)
    );

    dh.child.kill()?;

    Ok(())
}

/// We can set a different bad status code.
#[actix_rt::test]
async fn can_configure_bad_status_code() -> Result<(), Error> {
    let mut dh = ProbyProcess::new(Vec::<String>::new())?;

    let url = format!("{}?good=500", dh.selfcheck());
    let mut resp_body = surf::get(url).await?;
    assert_eq!(resp_body.status(), 500);
    assert_eq!(
        resp_body.body_string().await?,
        format!("{}:{} is connectable", dh.host, dh.port)
    );

    dh.child.kill()?;

    Ok(())
}