diff options
| -rw-r--r-- | askama_derive/src/generator.rs | 24 | ||||
| -rw-r--r-- | askama_hyper/tests/basic.rs | 44 |
2 files changed, 62 insertions, 6 deletions
diff --git a/askama_derive/src/generator.rs b/askama_derive/src/generator.rs index 173fab1..762f356 100644 --- a/askama_derive/src/generator.rs +++ b/askama_derive/src/generator.rs @@ -426,12 +426,13 @@ impl<'a> Generator<'a> { buf.writeln("}") } - // Implement `From<Template> for hyper::Response<Body>`. + // Implement `From<Template> for hyper::Response<Body>` and `From<Template> for hyper::Body. #[cfg(feature = "with-hyper")] fn impl_hyper_into_response(&mut self, buf: &mut Buffer) -> Result<(), CompileError> { let (impl_generics, orig_ty_generics, where_clause) = self.input.ast.generics.split_for_impl(); let ident = &self.input.ast.ident; + // From<Template> for hyper::Response<Body> buf.writeln(&format!( "{} {{", quote!( @@ -447,6 +448,27 @@ impl<'a> Generator<'a> { ))?; buf.writeln("::askama_hyper::respond(value)")?; buf.writeln("}")?; + buf.writeln("}")?; + + // From<Template> for hyper::Body + buf.writeln(&format!( + "{} {{", + quote!( + impl #impl_generics ::core::convert::From<&#ident #orig_ty_generics> + for ::askama_hyper::hyper::Body + #where_clause + ) + ))?; + buf.writeln("#[inline]")?; + buf.writeln(&format!( + "{} {{", + quote!(fn from(value: &#ident #orig_ty_generics) -> Self) + ))?; + buf.writeln( + "::askama::Template::render(value).ok().map(Into::into)\ + .unwrap_or_else(|| ::askama_hyper::hyper::Body::empty())", + )?; + buf.writeln("}")?; buf.writeln("}") } diff --git a/askama_hyper/tests/basic.rs b/askama_hyper/tests/basic.rs index efa8ba5..831bcbb 100644 --- a/askama_hyper/tests/basic.rs +++ b/askama_hyper/tests/basic.rs @@ -2,6 +2,7 @@ use std::convert::Infallible; use askama::Template; use hyper::body::to_bytes; +use hyper::http::response::Builder; use hyper::{Body, Client, Request, Response, Server}; use routerify::ext::RequestExt; use routerify::{Router, RouterService}; @@ -21,9 +22,16 @@ async fn hello_handler(req: Request<Body>) -> Result<Response<Body>, Infallible> Ok(template.into()) } +async fn body_handler(req: Request<Body>) -> Result<Response<Body>, Infallible> { + let name = req.param("name").unwrap(); + let template = &HelloTemplate { name: &name }; + Ok(Builder::new().body(template.into()).unwrap()) +} + fn router() -> Router<Body, Infallible> { Router::builder() .get("/hello/:name", hello_handler) + .get("/body/:name", body_handler) .build() .unwrap() } @@ -35,14 +43,40 @@ async fn test_hyper() { let server = Server::bind(&addr).serve(service); let local_addr = server.local_addr(); - let (tx, rx) = tokio::sync::oneshot::channel::<()>(); + let (tx1, rx1) = tokio::sync::oneshot::channel::<()>(); + let (tx2, rx2) = tokio::sync::oneshot::channel::<()>(); let serve = async move { let server = server.with_graceful_shutdown(async { - rx.await.expect("Could not await signal to stop"); + rx1.await.expect("Could not await signal to stop"); + rx2.await.expect("Could not await signal to stop"); }); server.await.expect("Could not serve"); }; - let query = async move { + let hello_query = async move { + let uri = format!("http://{local_addr}/hello/world") + .parse() + .expect("Could not format URI"); + let client = Client::new(); + + let res = client.get(uri).await.expect("Could not query client"); + assert_eq!(res.status(), hyper::StatusCode::OK); + + let content_type = res + .headers() + .get("content-type") + .expect("Response did not contain content-type header") + .to_str() + .expect("Content-type was not a UTF-8 string"); + assert_eq!(content_type, mime::TEXT_HTML_UTF_8.to_string()); + + let body = to_bytes(res).await.expect("No body returned"); + let body = std::str::from_utf8(&body).expect("Body was not UTF-8"); + assert_eq!(body, "Hello, world!"); + + tx1.send(()).unwrap(); + }; + + let body_query = async move { let uri = format!("http://{local_addr}/hello/world") .parse() .expect("Could not format URI"); @@ -63,8 +97,8 @@ async fn test_hyper() { let body = std::str::from_utf8(&body).expect("Body was not UTF-8"); assert_eq!(body, "Hello, world!"); - tx.send(()).unwrap(); + tx2.send(()).unwrap(); }; - tokio::join!(serve, query); + tokio::join!(serve, body_query, hello_query); } |