actix-server/src/main.rs

   1 use actix_web::{get, web, App, HttpRequest, HttpServer, Result};
   2 use actix_web::http::header;
   3 use actix_web::middleware::Logger;
   4
   5 #[get("/hello/{name}")]
   6 async fn hello(params: web::Path<String>, req: HttpRequest) -> Result<String> {
   7     let name = params.into_inner();
   8
   9     match req.headers().get(header::USER_AGENT) {
  10         Some(agent) => Ok(format!("Hello {} from {}!", name, agent.to_str().unwrap())),
  11         None => Ok(format!("Hello {}!", name))
  12     }
  13 }
  14
  15 #[actix_web::main]
  16 async fn main() -> std::io::Result<()> {
  17     env_logger::init_from_env(env_logger::Env::default().default_filter_or("info"));
  18
  19     HttpServer::new(|| {
  20         App::new()
  21             .service(hello)
  22             .wrap(Logger::default())
  23     })
  24         .bind(("127.0.0.1", 3030))?
  25         .run()
  26         .await
  27 }
  28
  29 #[cfg(test)]
  30 mod tests {
  31     use actix_web::{App, body, test, web};
  32     use actix_web::http::header;
  33
  34     use super::{hello};
  35
  36     #[actix_web::test]
  37     async fn test_hello() {
  38         let app = test::init_service(App::new().service(hello)).await;
  39
  40         // no user-agent
  41         let req = test::TestRequest::get().uri("/hello/rust").to_request();
  42         let res = test::call_service(&app, req).await;
  43         assert!(res.status().is_success());
  44         assert_eq!(body::to_bytes(res.into_body()).await.unwrap(),
  45                    web::Bytes::from_static(b"Hello rust!"));
  46
  47         // with user-agent
  48         let req = test::TestRequest::get()
  49             .uri("/hello/rust")
  50             .insert_header((header::USER_AGENT, "TestBrowser 0.1"))
  51             .to_request();
  52         let res = test::call_service(&app, req).await;
  53         assert!(res.status().is_success());
  54         assert_eq!(body::to_bytes(res.into_body()).await.unwrap(),
  55                    web::Bytes::from_static(b"Hello rust from TestBrowser 0.1!"));
  56     }
  57 }