From 2e16c3ca8a68493434e8a191069fc7ff337a3126 Mon Sep 17 00:00:00 2001
From: Ms2ger <Ms2ger@gmail.com>
Date: Tue, 6 Oct 2015 18:10:48 +0200
Subject: [PATCH] Avoid an unnecessary unwrap() call in handle_take_screenshot.

---
 components/webdriver_server/lib.rs | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/components/webdriver_server/lib.rs b/components/webdriver_server/lib.rs
index 22776126a94..ecead687c1f 100644
--- a/components/webdriver_server/lib.rs
+++ b/components/webdriver_server/lib.rs
@@ -615,12 +615,13 @@ impl Handler {
             sleep_ms(interval)
         }
 
-        if img.is_none() {
-            return Err(WebDriverError::new(ErrorStatus::Timeout,
-                                           "Taking screenshot timed out"));
-        }
+        let mut img = match img {
+            Some(img) => img,
+            None => return Err(WebDriverError::new(ErrorStatus::Timeout,
+                                                   "Taking screenshot timed out")),
+        };
 
-        let img_vec = match png::to_vec(&mut img.unwrap()) {
+        let img_vec = match png::to_vec(&mut img) {
            Ok(x) => x,
            Err(_) => return Err(WebDriverError::new(ErrorStatus::UnknownError,
                                                     "Taking screenshot failed"))