mirror of
https://github.com/tower-rs/tower.git
synced 2026-03-13 17:27:56 +00:00
e2f1a49cf3
* Update hedge, filter, load, load-shed, and more * Update ready cache * Prepare release for ready-cache * fix merge * Update balance * Prepare balance release
36 lines
948 B
Rust
36 lines
948 B
Rust
use std::{thread, time::Duration};
|
|
use tokio_test::{assert_pending, assert_ready, assert_ready_err, assert_ready_ok};
|
|
use tower_spawn_ready::SpawnReadyLayer;
|
|
use tower_test::mock;
|
|
|
|
#[tokio::test]
|
|
async fn when_inner_is_not_ready() {
|
|
let layer = SpawnReadyLayer::new();
|
|
let (mut service, mut handle) = mock::spawn_layer::<(), (), _>(layer);
|
|
|
|
// Make the service NotReady
|
|
handle.allow(0);
|
|
|
|
assert_pending!(service.poll_ready());
|
|
|
|
// Make the service is Ready
|
|
handle.allow(1);
|
|
thread::sleep(Duration::from_millis(100));
|
|
assert_ready_ok!(service.poll_ready());
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn when_inner_fails() {
|
|
let layer = SpawnReadyLayer::new();
|
|
let (mut service, mut handle) = mock::spawn_layer::<(), (), _>(layer);
|
|
|
|
// Make the service NotReady
|
|
handle.allow(0);
|
|
handle.send_error("foobar");
|
|
|
|
assert_eq!(
|
|
assert_ready_err!(service.poll_ready()).to_string(),
|
|
"foobar"
|
|
);
|
|
}
|