mirror of
https://github.com/rwf2/Rocket.git
synced 2025-01-24 10:22:23 +00:00
93 lines
3.5 KiB
Rust
93 lines
3.5 KiB
Rust
use super::{rocket, index, PasteId};
|
|
use rocket::local::blocking::Client;
|
|
use rocket::http::{Status, ContentType};
|
|
use rocket::request::FromParam;
|
|
|
|
fn extract_id(from: &str) -> Option<String> {
|
|
from.rfind('/').map(|i| &from[(i + 1)..]).map(|s| s.trim_end().to_string())
|
|
}
|
|
|
|
#[test]
|
|
fn check_index() {
|
|
let client = Client::tracked(rocket()).unwrap();
|
|
|
|
// Ensure the index returns what we expect.
|
|
let response = client.get(uri!(super::index)).dispatch();
|
|
assert_eq!(response.status(), Status::Ok);
|
|
assert_eq!(response.content_type(), Some(ContentType::Plain));
|
|
assert_eq!(response.into_string(), Some(index().into()))
|
|
}
|
|
|
|
fn upload_paste(client: &Client, body: &str) -> String {
|
|
let response = client.post(uri!(super::upload)).body(body).dispatch();
|
|
assert_eq!(response.status(), Status::Ok);
|
|
assert_eq!(response.content_type(), Some(ContentType::Plain));
|
|
extract_id(&response.into_string().unwrap()).unwrap()
|
|
}
|
|
|
|
fn download_paste(client: &Client, id: &str) -> Option<String> {
|
|
let id = PasteId::from_param(id).expect("valid ID");
|
|
let response = client.get(uri!(super::retrieve(id))).dispatch();
|
|
if response.status().class().is_success() {
|
|
Some(response.into_string().unwrap())
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
|
|
fn delete_paste(client: &Client, id: &str) {
|
|
let id = PasteId::from_param(id).expect("valid ID");
|
|
let response = client.delete(uri!(super::delete(id))).dispatch();
|
|
assert_eq!(response.status(), Status::Ok);
|
|
}
|
|
|
|
#[test]
|
|
fn pasting() {
|
|
let client = Client::tracked(rocket()).unwrap();
|
|
|
|
// Do a trivial upload, just to make sure it works.
|
|
let body_1 = "Hello, world!";
|
|
let id_1 = upload_paste(&client, body_1);
|
|
assert_eq!(download_paste(&client, &id_1).unwrap(), body_1);
|
|
|
|
// Make sure we can keep getting that paste.
|
|
assert_eq!(download_paste(&client, &id_1).unwrap(), body_1);
|
|
assert_eq!(download_paste(&client, &id_1).unwrap(), body_1);
|
|
assert_eq!(download_paste(&client, &id_1).unwrap(), body_1);
|
|
|
|
// Upload some unicode.
|
|
let body_2 = "こんにちは";
|
|
let id_2 = upload_paste(&client, body_2);
|
|
assert_eq!(download_paste(&client, &id_2).unwrap(), body_2);
|
|
|
|
// Make sure we can get both pastes.
|
|
assert_eq!(download_paste(&client, &id_1).unwrap(), body_1);
|
|
assert_eq!(download_paste(&client, &id_2).unwrap(), body_2);
|
|
assert_eq!(download_paste(&client, &id_1).unwrap(), body_1);
|
|
assert_eq!(download_paste(&client, &id_2).unwrap(), body_2);
|
|
|
|
// Now a longer upload.
|
|
let body_3 = "Lorem ipsum dolor sit amet, consectetur adipisicing elit, sed
|
|
do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim
|
|
ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut
|
|
aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit
|
|
in voluptate velit esse cillum dolore eu fugiat nulla pariatur.
|
|
Excepteur sint occaecat cupidatat non proident, sunt in culpa qui
|
|
officia deserunt mollit anim id est laborum.";
|
|
|
|
let id_3 = upload_paste(&client, body_3);
|
|
assert_eq!(download_paste(&client, &id_3).unwrap(), body_3);
|
|
assert_eq!(download_paste(&client, &id_1).unwrap(), body_1);
|
|
assert_eq!(download_paste(&client, &id_2).unwrap(), body_2);
|
|
|
|
// Delete everything we uploaded.
|
|
delete_paste(&client, &id_1);
|
|
assert!(download_paste(&client, &id_1).is_none());
|
|
|
|
delete_paste(&client, &id_2);
|
|
assert!(download_paste(&client, &id_2).is_none());
|
|
|
|
delete_paste(&client, &id_3);
|
|
assert!(download_paste(&client, &id_3).is_none());
|
|
}
|