lovasoa
104 lines
3.5 KiB
Rust
104 lines
3.5 KiB
Rust
use actix_web::{
|
|
http::StatusCode,
|
|
test,
|
|
};
|
|
use sqlpage::webserver::http::main_handler;
|
|
|
|
use crate::common::get_request_to;
|
|
|
|
#[actix_web::test]
|
|
async fn test_request_body() -> actix_web::Result<()> {
|
|
let req = get_request_to("/tests/requests/request_body_test.sql")
|
|
.await?
|
|
.insert_header(("content-type", "text/plain"))
|
|
.set_payload("Hello, world!")
|
|
.to_srv_request();
|
|
let resp = main_handler(req).await?;
|
|
|
|
assert_eq!(resp.status(), StatusCode::OK);
|
|
let body = test::read_body(resp).await;
|
|
let body_str = String::from_utf8(body.to_vec()).unwrap();
|
|
assert!(
|
|
body_str.contains("Hello, world!"),
|
|
"{body_str}\nexpected to contain: Hello, world!"
|
|
);
|
|
|
|
// Test with form data - should return NULL
|
|
let req = get_request_to("/tests/requests/request_body_test.sql")
|
|
.await?
|
|
.insert_header(("content-type", "application/x-www-form-urlencoded"))
|
|
.set_payload("key=value")
|
|
.to_srv_request();
|
|
let resp = main_handler(req).await?;
|
|
|
|
assert_eq!(resp.status(), StatusCode::OK);
|
|
let body = test::read_body(resp).await;
|
|
let body_str = String::from_utf8(body.to_vec()).unwrap();
|
|
assert!(
|
|
body_str.contains("NULL"),
|
|
"{body_str}\nexpected NULL for form data"
|
|
);
|
|
Ok(())
|
|
}
|
|
|
|
#[actix_web::test]
|
|
async fn test_request_body_base64() -> actix_web::Result<()> {
|
|
let binary_data = (0u8..=255u8).collect::<Vec<_>>();
|
|
let expected_base64 =
|
|
base64::Engine::encode(&base64::engine::general_purpose::STANDARD, &binary_data);
|
|
|
|
let req = get_request_to("/tests/requests/request_body_base64_test.sql")
|
|
.await?
|
|
.insert_header(("content-type", "application/octet-stream"))
|
|
.set_payload(binary_data)
|
|
.to_srv_request();
|
|
let resp = main_handler(req).await?;
|
|
|
|
assert_eq!(resp.status(), StatusCode::OK);
|
|
let body = test::read_body(resp).await;
|
|
let body_str = String::from_utf8(body.to_vec()).unwrap();
|
|
assert!(
|
|
body_str.contains(&expected_base64),
|
|
"{body_str}\nexpected to contain base64: {expected_base64}"
|
|
);
|
|
|
|
// Test with form data - should return NULL
|
|
let req = get_request_to("/tests/requests/request_body_base64_test.sql")
|
|
.await?
|
|
.insert_header(("content-type", "application/x-www-form-urlencoded"))
|
|
.set_payload("key=value")
|
|
.to_srv_request();
|
|
let resp = main_handler(req).await?;
|
|
|
|
assert_eq!(resp.status(), StatusCode::OK);
|
|
let body = test::read_body(resp).await;
|
|
let body_str = String::from_utf8(body.to_vec()).unwrap();
|
|
assert!(
|
|
body_str.contains("NULL"),
|
|
"{body_str}\nexpected NULL for form data"
|
|
);
|
|
Ok(())
|
|
}
|
|
|
|
#[actix_web::test]
|
|
async fn test_large_form_field_roundtrip() -> actix_web::Result<()> {
|
|
let long_string = "a".repeat(123454);
|
|
let req = get_request_to("/tests/components/display_form_field.sql")
|
|
.await?
|
|
.insert_header(("content-type", "application/x-www-form-urlencoded"))
|
|
.set_payload(["x=", &long_string].concat()) // total size is 123454 + 2 = 123456 bytes
|
|
.to_srv_request();
|
|
let resp = main_handler(req).await?;
|
|
assert_eq!(resp.status(), StatusCode::OK);
|
|
let body = test::read_body(resp).await;
|
|
let body_str = String::from_utf8(body.to_vec()).unwrap();
|
|
assert!(
|
|
!body_str.contains("error"),
|
|
"{body_str}\nshouldn't have errors"
|
|
);
|
|
assert!(
|
|
body_str.contains(&long_string),
|
|
"{body_str}\nexpected to contain long string submitted"
|
|
);
|
|
Ok(())
|
|
}
|