kellnr-webhooks 7.0.0

Kellnr is a self-hosted registry for Rust crates with support for rustdocs and crates.io caching.
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

use std::time::Duration;

use axum::Router;
use axum::body::Bytes;
use axum::http::{HeaderMap, StatusCode, header};
use kellnr_http::test_support::TestServer;
use tokio::sync::mpsc::{UnboundedReceiver, unbounded_channel};

/// A request received by a [`TestListener`].
pub(crate) struct Received {
    pub headers: HeaderMap,
    pub body: Bytes,
}

/// Webhook target on a free port that answers every request the same way and
/// reports each received request on `rx`.
pub(crate) struct TestListener {
    server: TestServer,
    pub rx: UnboundedReceiver<Received>,
}

impl TestListener {
    pub(crate) fn url(&self) -> String {
        self.server.url("/")
    }

    pub(crate) fn port(&self) -> u16 {
        self.server.addr().port()
    }
}

pub(crate) async fn get_test_listener(status_code: u16) -> TestListener {
    get_test_listener_with_response(status_code, HeaderMap::new(), String::new()).await
}

pub(crate) async fn get_redirect_listener(location: &str) -> TestListener {
    let mut headers = HeaderMap::new();
    headers.insert(header::LOCATION, location.parse().unwrap());
    get_test_listener_with_response(302, headers, String::new()).await
}

pub(crate) async fn get_test_listener_with_response(
    status_code: u16,
    headers: HeaderMap,
    body: String,
) -> TestListener {
    let status = StatusCode::from_u16(status_code).unwrap();
    let (tx, rx) = unbounded_channel();
    let router = Router::new().fallback(move |request_headers: HeaderMap, request_body: Bytes| {
        let _ = tx.send(Received {
            headers: request_headers,
            body: request_body,
        });
        let response = (status, headers.clone(), body.clone());
        async move { response }
    });
    TestListener {
        server: TestServer::start(router).await,
        rx,
    }
}

/// Webhook target on a free port that answers every request with 200 after
/// `delay`, handling requests in parallel.
pub(crate) async fn get_slow_listener(delay: Duration) -> TestServer {
    TestServer::start(Router::new().fallback(move || async move {
        tokio::time::sleep(delay).await;
        StatusCode::OK
    }))
    .await
}