85 lines
2.3 KiB
Rust
85 lines
2.3 KiB
Rust
#![recursion_limit="128"]
|
|
#![deny(warnings)]
|
|
#[macro_use]
|
|
mod support;
|
|
use self::support::*;
|
|
use support::tap::TapEventExt;
|
|
|
|
// Flaky: sometimes the admin thread hasn't had a chance to register
|
|
// the Taps before the `client.get` is called.
|
|
#[test]
|
|
#[cfg_attr(not(feature = "flaky_tests"), ignore)]
|
|
fn inbound_http1() {
|
|
let _ = env_logger_init();
|
|
let srv = server::http1()
|
|
.route("/", "hello")
|
|
.run();
|
|
|
|
let proxy = proxy::new()
|
|
.inbound(srv)
|
|
.run();
|
|
|
|
let mut tap = tap::client(proxy.control);
|
|
let events = tap.observe(
|
|
tap::observe_request()
|
|
);
|
|
|
|
let authority = "tap.test.svc.cluster.local";
|
|
let client = client::http1(proxy.inbound, authority);
|
|
|
|
assert_eq!(client.get("/"), "hello");
|
|
|
|
let mut events = events.wait().take(3);
|
|
|
|
let ev1 = events.next().expect("next1").expect("stream1");
|
|
assert!(ev1.is_inbound());
|
|
assert_eq!(ev1.request_init_authority(), authority);
|
|
assert_eq!(ev1.request_init_path(), "/");
|
|
|
|
let ev2 = events.next().expect("next2").expect("stream2");
|
|
assert!(ev2.is_inbound());
|
|
assert_eq!(ev2.response_init_status(), 200);
|
|
|
|
let ev3 = events.next().expect("next3").expect("stream3");
|
|
assert!(ev3.is_inbound());
|
|
assert_eq!(ev3.response_end_bytes(), 5);
|
|
}
|
|
|
|
#[test]
|
|
#[cfg_attr(not(feature = "flaky_tests"), ignore)]
|
|
fn grpc_headers_end() {
|
|
let _ = env_logger_init();
|
|
let srv = server::http2()
|
|
.route_fn("/", |_req| {
|
|
Response::builder()
|
|
.header("grpc-status", "1")
|
|
.body(Default::default())
|
|
.unwrap()
|
|
})
|
|
.run();
|
|
|
|
let proxy = proxy::new()
|
|
.inbound(srv)
|
|
.run();
|
|
|
|
let mut tap = tap::client(proxy.control);
|
|
let events = tap.observe(
|
|
tap::observe_request()
|
|
);
|
|
|
|
let authority = "tap.test.svc.cluster.local";
|
|
let client = client::http2(proxy.inbound, authority);
|
|
|
|
let res = client.request(client
|
|
.request_builder("/")
|
|
.header("content-type", "application/grpc+nope")
|
|
);
|
|
assert_eq!(res.status(), 200);
|
|
assert_eq!(res.headers()["grpc-status"], "1");
|
|
assert_eq!(res.into_body().concat2().wait().unwrap().len(), 0);
|
|
|
|
let ev = events.wait().nth(2).expect("nth").expect("stream");
|
|
|
|
assert_eq!(ev.response_end_eos_grpc(), 1);
|
|
}
|