Add trailing metadata and fix test

This commit is contained in:
Lucio Franco
2019-08-28 20:09:43 -04:00
parent f750dfa111
commit bfab56d42d
4 changed files with 54 additions and 28 deletions
+10 -2
View File
@@ -418,6 +418,7 @@ pub async fn custom_metadata(client: &mut Client, assertions: &mut Vec<TestAsser
// format!("result={:?}", response.metadata().get_bin(key1))
// ));
let response = client
.full_duplex_call(req_stream)
.await
@@ -428,10 +429,17 @@ pub async fn custom_metadata(client: &mut Client, assertions: &mut Vec<TestAsser
response.metadata().get(key1) == Some(&value1),
format!("result={:?}", response.metadata().get(key1))
));
let mut stream = response.into_inner();
// while let Some(_) = stream.next().await {}
let trailers = stream.trailers().await.unwrap().unwrap();
assertions.push(test_assert!(
"metadata bin must match in unary",
response.metadata().get_bin(key2) == Some(&value2),
format!("result={:?}", response.metadata().get_bin(key1))
trailers.get_bin(key2) == Some(&value2),
format!("result={:?}", trailers.get_bin(key1))
));
}
+1 -3
View File
@@ -1,6 +1,4 @@
cargo run -p tonic-interop --bin client -- \
--test_case=empty_unary,large_unary,client_streaming,server_streaming,ping_pong,\
empty_stream,status_code_and_message,special_status_message,unimplemented_method,\
unimplemented_service
# DISABLED: ,custom_metadata
unimplemented_service,custom_metadata