From 1bcc50ba310d5a8b50b03d487e201cb0038a4d34 Mon Sep 17 00:00:00 2001
From: Sven-Hendrik Haase <svenstaro@gmail.com>
Date: Wed, 24 Apr 2019 03:38:22 +0200
Subject: Add a couple more integration tests

---
 src/file_upload.rs | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

(limited to 'src/file_upload.rs')

diff --git a/src/file_upload.rs b/src/file_upload.rs
index 534083c..88f8a7c 100644
--- a/src/file_upload.rs
+++ b/src/file_upload.rs
@@ -101,10 +101,11 @@ fn handle_multipart(
 /// invalid.
 /// This method returns future.
 pub fn upload_file(req: &HttpRequest<crate::MiniserveConfig>) -> FutureResponse<HttpResponse> {
-    let return_path: String = req.headers()[header::REFERER]
-        .to_str()
-        .unwrap_or("/")
-        .to_owned();
+    let return_path = if let Some(header) = req.headers().get(header::REFERER) {
+        header.to_str().unwrap_or("/").to_owned()
+    } else {
+        "/".to_string()
+    };
     let app_root_dir = if let Ok(dir) = req.state().path.canonicalize() {
         dir
     } else {
-- 
cgit v1.2.3