structured-proxy 7.0.0

Universal gRPC→REST transcoding proxy — config-driven, works with any gRPC service
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

use super::*;

use std::time::Duration;

/// Whether `future` has resolved, without waiting.
async fn ready<F: Future>(future: Pin<&mut F>) -> bool {
    tokio::select! {
        biased;
        _ = future => true,
        () = std::future::ready(()) => false,
    }
}

#[tokio::test(start_paused = true)]
async fn a_connection_with_nothing_in_flight_is_idle_after_the_timeout() {
    let activity = Arc::new(Activity::default());
    let idle = activity.idle_for(Duration::from_secs(1));
    tokio::pin!(idle);
    assert!(!ready(idle.as_mut()).await);
    tokio::time::advance(Duration::from_millis(1001)).await;
    assert!(ready(idle.as_mut()).await);
}

#[tokio::test(start_paused = true)]
async fn a_request_in_flight_keeps_its_connection() {
    // A long stream must not be cut because it outlasts the idle timeout.
    let activity = Arc::new(Activity::default());
    let in_flight = activity.enter();
    let idle = activity.idle_for(Duration::from_secs(1));
    tokio::pin!(idle);
    tokio::time::advance(Duration::from_secs(10)).await;
    assert!(!ready(idle.as_mut()).await);

    // The timeout counts from the end of the request.
    drop(in_flight);
    assert!(!ready(idle.as_mut()).await);
    tokio::time::advance(Duration::from_millis(500)).await;
    assert!(!ready(idle.as_mut()).await);
    tokio::time::advance(Duration::from_millis(501)).await;
    assert!(ready(idle.as_mut()).await);
}

#[tokio::test(start_paused = true)]
async fn a_request_that_comes_and_goes_restarts_the_timeout() {
    let activity = Arc::new(Activity::default());
    let idle = activity.idle_for(Duration::from_secs(1));
    tokio::pin!(idle);
    assert!(!ready(idle.as_mut()).await);
    tokio::time::advance(Duration::from_millis(800)).await;
    drop(activity.enter());
    assert!(!ready(idle.as_mut()).await);
    tokio::time::advance(Duration::from_millis(800)).await;
    assert!(!ready(idle.as_mut()).await);
    tokio::time::advance(Duration::from_millis(201)).await;
    assert!(ready(idle.as_mut()).await);
}