mirror of
https://github.com/tokio-rs/axum.git
synced 2025-09-27 13:00:39 +00:00
131 lines
3.4 KiB
Rust
131 lines
3.4 KiB
Rust
//! Run with
|
|
//!
|
|
//! ```not_rust
|
|
//! cargo run -p example-form
|
|
//! ```
|
|
|
|
use axum::{extract::Form, response::Html, routing::get, Router};
|
|
use serde::Deserialize;
|
|
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
tracing_subscriber::registry()
|
|
.with(
|
|
tracing_subscriber::EnvFilter::try_from_default_env()
|
|
.unwrap_or_else(|_| format!("{}=debug", env!("CARGO_CRATE_NAME")).into()),
|
|
)
|
|
.with(tracing_subscriber::fmt::layer())
|
|
.init();
|
|
|
|
// build our application with some routes
|
|
let app = app();
|
|
|
|
// run it
|
|
let listener = tokio::net::TcpListener::bind("127.0.0.1:3000")
|
|
.await
|
|
.unwrap();
|
|
tracing::debug!("listening on {}", listener.local_addr().unwrap());
|
|
axum::serve(listener, app).await.unwrap();
|
|
}
|
|
|
|
fn app() -> Router {
|
|
Router::new().route("/", get(show_form).post(accept_form))
|
|
}
|
|
|
|
async fn show_form() -> Html<&'static str> {
|
|
Html(
|
|
r#"
|
|
<!doctype html>
|
|
<html>
|
|
<head></head>
|
|
<body>
|
|
<form action="/" method="post">
|
|
<label for="name">
|
|
Enter your name:
|
|
<input type="text" name="name">
|
|
</label>
|
|
|
|
<label>
|
|
Enter your email:
|
|
<input type="text" name="email">
|
|
</label>
|
|
|
|
<input type="submit" value="Subscribe!">
|
|
</form>
|
|
</body>
|
|
</html>
|
|
"#,
|
|
)
|
|
}
|
|
|
|
#[derive(Deserialize, Debug)]
|
|
#[allow(dead_code)]
|
|
struct Input {
|
|
name: String,
|
|
email: String,
|
|
}
|
|
|
|
async fn accept_form(Form(input): Form<Input>) -> Html<String> {
|
|
dbg!(&input);
|
|
Html(format!(
|
|
"email='{}'\nname='{}'\n",
|
|
&input.email, &input.name
|
|
))
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use axum::{
|
|
body::Body,
|
|
http::{self, Request, StatusCode},
|
|
};
|
|
use http_body_util::BodyExt;
|
|
use tower::ServiceExt; // for `call`, `oneshot`, and `ready` // for `collect`
|
|
|
|
#[tokio::test]
|
|
async fn test_get() {
|
|
let app = app();
|
|
|
|
let response = app
|
|
.oneshot(Request::builder().uri("/").body(Body::empty()).unwrap())
|
|
.await
|
|
.unwrap();
|
|
|
|
assert_eq!(response.status(), StatusCode::OK);
|
|
|
|
let body = response.into_body().collect().await.unwrap().to_bytes();
|
|
let body = std::str::from_utf8(&body).unwrap();
|
|
|
|
assert!(body.contains(r#"<input type="submit" value="Subscribe!">"#));
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn test_post() {
|
|
let app = app();
|
|
|
|
let response = app
|
|
.oneshot(
|
|
Request::builder()
|
|
.method(http::Method::POST)
|
|
.uri("/")
|
|
.header(
|
|
http::header::CONTENT_TYPE,
|
|
mime::APPLICATION_WWW_FORM_URLENCODED.as_ref(),
|
|
)
|
|
.body(Body::from("name=foo&email=bar@axum"))
|
|
.unwrap(),
|
|
)
|
|
.await
|
|
.unwrap();
|
|
|
|
assert_eq!(response.status(), StatusCode::OK);
|
|
|
|
let body = response.into_body().collect().await.unwrap().to_bytes();
|
|
let body = std::str::from_utf8(&body).unwrap();
|
|
|
|
assert_eq!(body, "email='bar@axum'\nname='foo'\n");
|
|
}
|
|
}
|