From e6615af7e6aa1f71f7a6a69f4777a4530310272c Mon Sep 17 00:00:00 2001
From: Sergio Benitez <sb@sergio.bz>
Date: Fri, 14 Apr 2017 14:39:17 -0700
Subject: [PATCH] Add tests for pastebin example.

---
 .gitignore                     |  3 ++
 examples/pastebin/Cargo.toml   |  3 ++
 examples/pastebin/src/main.rs  |  9 +++-
 examples/pastebin/src/tests.rs | 76 ++++++++++++++++++++++++++++++++++
 4 files changed, 89 insertions(+), 2 deletions(-)
 create mode 100644 examples/pastebin/src/tests.rs

diff --git a/.gitignore b/.gitignore
index 8dcb1cb3..03fcdce2 100644
--- a/.gitignore
+++ b/.gitignore
@@ -22,3 +22,6 @@ scripts/upload-docs.sh
 
 # Backup files.
 *.bak
+
+# Uploads in pastebin example.
+examples/pastebin/upload/*
diff --git a/examples/pastebin/Cargo.toml b/examples/pastebin/Cargo.toml
index 907ff59a..de14c65e 100644
--- a/examples/pastebin/Cargo.toml
+++ b/examples/pastebin/Cargo.toml
@@ -7,3 +7,6 @@ workspace = "../../"
 rocket = { path = "../../lib" }
 rocket_codegen = { path = "../../codegen" }
 rand = "0.3"
+
+[dev-dependencies]
+rocket = { path = "../../lib", features = ["testing"] }
diff --git a/examples/pastebin/src/main.rs b/examples/pastebin/src/main.rs
index b15cdf1b..2a95e527 100644
--- a/examples/pastebin/src/main.rs
+++ b/examples/pastebin/src/main.rs
@@ -5,6 +5,7 @@ extern crate rocket;
 extern crate rand;
 
 mod paste_id;
+#[cfg(test)] mod tests;
 
 use std::io;
 use std::fs::File;
@@ -52,6 +53,10 @@ fn index() -> &'static str {
     "
 }
 
-fn main() {
-    rocket::ignite().mount("/", routes![index, upload, retrieve]).launch();
+fn rocket() -> rocket::Rocket {
+    rocket::ignite().mount("/", routes![index, upload, retrieve])
+}
+
+fn main() {
+    rocket().launch();
 }
diff --git a/examples/pastebin/src/tests.rs b/examples/pastebin/src/tests.rs
new file mode 100644
index 00000000..01f9a6e8
--- /dev/null
+++ b/examples/pastebin/src/tests.rs
@@ -0,0 +1,76 @@
+use super::{rocket, index};
+use rocket::testing::MockRequest;
+use rocket::http::Method::*;
+use rocket::http::{Status, ContentType};
+use rocket::Rocket;
+
+fn extract_id(from: &str) -> Option<String> {
+    from.rfind('/').map(|i| &from[(i + 1)..]).map(|s| s.trim_right().to_string())
+}
+
+#[test]
+fn check_index() {
+    let rocket = rocket();
+
+    // Ensure the index returns what we expect.
+    let mut req = MockRequest::new(Get, "/");
+    let mut response = req.dispatch_with(&rocket);
+    assert_eq!(response.status(), Status::Ok);
+    assert_eq!(response.content_type(), Some(ContentType::Plain));
+    assert_eq!(response.body_string(), Some(index().into()))
+}
+
+fn upload_paste(rocket: &Rocket, body: &str) -> String {
+    let mut req = MockRequest::new(Post, "/").body(body);
+    let mut response = req.dispatch_with(rocket);
+    assert_eq!(response.status(), Status::Ok);
+    assert_eq!(response.content_type(), Some(ContentType::Plain));
+    extract_id(&response.body_string().unwrap()).unwrap()
+}
+
+
+fn download_paste(rocket: &Rocket, id: &str) -> String {
+    let mut req = MockRequest::new(Get, format!("/{}", id));
+    let mut response = req.dispatch_with(rocket);
+    assert_eq!(response.status(), Status::Ok);
+    response.body_string().unwrap()
+}
+
+#[test]
+fn pasting() {
+    let rocket = rocket();
+
+    // Do a trivial upload, just to make sure it works.
+    let body_1 = "Hello, world!";
+    let id_1 = upload_paste(&rocket, body_1);
+    assert_eq!(download_paste(&rocket, &id_1), body_1);
+
+    // Make sure we can keep getting that paste.
+    assert_eq!(download_paste(&rocket, &id_1), body_1);
+    assert_eq!(download_paste(&rocket, &id_1), body_1);
+    assert_eq!(download_paste(&rocket, &id_1), body_1);
+
+    // Upload some unicode.
+    let body_2 = "こんにちは";
+    let id_2 = upload_paste(&rocket, body_2);
+    assert_eq!(download_paste(&rocket, &id_2), body_2);
+
+    // Make sure we can get both pastes.
+    assert_eq!(download_paste(&rocket, &id_1), body_1);
+    assert_eq!(download_paste(&rocket, &id_2), body_2);
+    assert_eq!(download_paste(&rocket, &id_1), body_1);
+    assert_eq!(download_paste(&rocket, &id_2), 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(&rocket, body_3);
+    assert_eq!(download_paste(&rocket, &id_3), body_3);
+    assert_eq!(download_paste(&rocket, &id_1), body_1);
+    assert_eq!(download_paste(&rocket, &id_2), body_2);
+}