amigan/rustypaste-pretty
1
0
Fork 0
mirror of https://github.com/amigan/rustypaste-pretty.git synced 2024-11-21 11:59:48 -05:00
Code Issues Projects Releases Packages Wiki Activity

test(server): add tests for server configuration

Browse source
This commit is contained in:
Orhun Parmaksız 2022-05-17 22:38:57 +03:00
parent 1f8f462299
commit 013c6ac810
No known key found for this signature in database
GPG key ID: F83424824B3E4B90
1 changed files with 59 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

74
src/server.rs
View file

@ -222,18 +222,7 @@ mod tests {
use std::thread;
use std::time::Duration;
#[actix_web::test]
async fn test_index() {
let app = test::init_service(App::new().service(index)).await;
let request = TestRequest::default()
.insert_header(("content-type", "text/plain"))
.to_request();
let resp = test::call_service(&app, request).await;
assert!(resp.status().is_redirection());
assert_eq!(http::StatusCode::FOUND, resp.status());
}
fn get_multipart_request(data: &str, name: &str, file_name: &str) -> TestRequest {
fn get_multipart_request(data: &str, name: &str, filename: &str) -> TestRequest {
let multipart_data = format!(
"\r\n\
--multipart_bound\r\n\
@ -242,7 +231,7 @@ mod tests {
{}\r\n\
--multipart_bound--\r\n",
name,
file_name,
filename,
data.bytes().len(),
data,
);
@ -267,6 +256,61 @@ mod tests {
}
}
#[actix_web::test]
async fn test_index() {
let app = test::init_service(App::new().service(index)).await;
let request = TestRequest::default()
.insert_header(("content-type", "text/plain"))
.to_request();
let resp = test::call_service(&app, request).await;
assert!(resp.status().is_redirection());
assert_eq!(http::StatusCode::FOUND, resp.status());
}
#[actix_web::test]
async fn test_auth() -> Result<(), Error> {
let mut config = Config::default();
config.server.auth_token = Some(String::from("test"));
let app = test::init_service(
App::new()
.app_data(Data::new(RwLock::new(config)))
.app_data(Data::new(Client::default()))
.service(serve)
.service(upload),
)
.await;
let response =
test::call_service(&app, get_multipart_request("", "", "").to_request()).await;
assert_eq!(http::StatusCode::UNAUTHORIZED, response.status());
assert_body(response, "unauthorized").await?;
Ok(())
}
#[actix_web::test]
async fn test_payload_limit() -> Result<(), Error> {
let app = test::init_service(
App::new()
.app_data(Data::new(RwLock::new(Config::default())))
.app_data(Data::new(Client::default()))
.service(serve)
.service(upload),
)
.await;
let response = test::call_service(
&app,
get_multipart_request("test", "file", "test").to_request(),
)
.await;
assert_eq!(http::StatusCode::PAYLOAD_TOO_LARGE, response.status());
assert_body(response, "upload limit exceeded").await?;
Ok(())
}
#[actix_web::test]
async fn test_upload_file() -> Result<(), Error> {
let mut config = Config::default();
@ -331,7 +375,7 @@ mod tests {
get_multipart_request(&timestamp, "file", file_name)
.insert_header((
header::HeaderName::from_static("expire"),
header::HeaderValue::from_static("10ms"),
header::HeaderValue::from_static("20ms"),
))
.to_request(),
)
@ -346,7 +390,7 @@ mod tests {
assert_eq!(http::StatusCode::OK, response.status());
assert_body(response, &timestamp).await?;
thread::sleep(Duration::from_millis(20));
thread::sleep(Duration::from_millis(40));
let serve_request = TestRequest::get()
.uri(&format!("/{}", file_name))