mirror of https://github.com/rwf2/Rocket.git
34 lines
994 B
Rust
34 lines
994 B
Rust
use std::fs::{self, File};
|
|
use std::io::prelude::*;
|
|
|
|
use rocket::local::Client;
|
|
|
|
#[test]
|
|
fn test_root() {
|
|
let client = Client::new(super::rocket()).unwrap();
|
|
let mut res = client.get("/").dispatch();
|
|
|
|
// Check that we have exactly 25,000 'a'.
|
|
let res_str = res.body_string().unwrap();
|
|
assert_eq!(res_str.len(), 25000);
|
|
for byte in res_str.as_bytes() {
|
|
assert_eq!(*byte, b'a');
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_file() {
|
|
// Create the 'big_file'
|
|
const CONTENTS: &str = "big_file contents...not so big here";
|
|
let mut file = File::create(super::FILENAME).expect("create big_file");
|
|
file.write_all(CONTENTS.as_bytes()).expect("write to big_file");
|
|
|
|
// Get the big file contents, hopefully.
|
|
let client = Client::new(super::rocket()).unwrap();
|
|
let mut res = client.get("/big_file").dispatch();
|
|
assert_eq!(res.body_string(), Some(CONTENTS.into()));
|
|
|
|
// Delete the 'big_file'.
|
|
fs::remove_file(super::FILENAME).expect("remove big_file");
|
|
}
|