talksik/tonic
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix(transport): Correctly map hyper errors (#629)

Browse Source 
Co-authored-by: Lucio Franco <[email protected]>
This commit is contained in:
David Freese
2021-10-20 11:05:27 -04:00
committed by GitHub
co-authored by Lucio Franco
parent 2e471548d8
commit 4947b076f5
2 changed files with 42 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/integration_tests/tests/connection.rs
+6 -3
View File
@@ -5,7 +5,7 @@ use std::time::Duration;
use tokio::sync::oneshot;
use tonic::{
transport::{Endpoint, Server},
Request, Response, Status,
Code, Request, Response, Status,
};
struct Svc(Arc<Mutex<Option<oneshot::Sender<()>>>>);
@@ -52,7 +52,8 @@ async fn connect_returns_err_via_call_after_connected() {
let res = client.unary_call(Request::new(Input {})).await;
assert!(res.is_err());
let err = res.unwrap_err();
assert_eq!(err.code(), Code::Unavailable);
jh.await.unwrap();
}
@@ -84,7 +85,9 @@ async fn connect_lazy_reconnects_after_first_failure() {
// The server shut down, third call should fail
tokio::time::sleep(Duration::from_millis(100)).await;
client.unary_call(Request::new(Input {})).await.unwrap_err();
let err = client.unary_call(Request::new(Input {})).await.unwrap_err();
assert_eq!(err.code(), Code::Unavailable);
jh.await.unwrap();
}