use http::{
HeaderMap,
HeaderValue,
};
use qubit_http::AsyncHttpHeaderInjector;
#[tokio::test]
async fn test_async_http_header_injector_runs_async_mutation() {
let injector = AsyncHttpHeaderInjector::new(|headers| {
Box::pin(async move {
tokio::task::yield_now().await;
headers.insert("x-token", HeaderValue::from_static("fresh"));
Ok(())
})
});
let mut headers = HeaderMap::new();
injector
.apply(&mut headers)
.await
.expect("injector should mutate headers");
assert_eq!(headers.get("x-token").expect("token header"), "fresh");
}