From bb7affb9eb664d08c2aaa25d290c426abd4e7608 Mon Sep 17 00:00:00 2001
From: bootstraponline <code@bootstraponline.com>
Date: Wed, 11 Sep 2013 11:22:42 -0400
Subject: [PATCH] Fix screenshotStatus

---
 .../appium/android/bootstrap/handler/TakeScreenshot.java | 5 +++--
 app/android.js                                           | 9 +++++++--
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/android/bootstrap/src/io/appium/android/bootstrap/handler/TakeScreenshot.java b/android/bootstrap/src/io/appium/android/bootstrap/handler/TakeScreenshot.java
index bd162a200..5e0e21391 100644
--- a/android/bootstrap/src/io/appium/android/bootstrap/handler/TakeScreenshot.java
+++ b/android/bootstrap/src/io/appium/android/bootstrap/handler/TakeScreenshot.java
@@ -26,8 +26,9 @@ public class TakeScreenshot extends CommandHandler {
    */
   @Override
   public AndroidCommandResult execute(final AndroidCommand command) {
-    final boolean result = UiDevice.getInstance().takeScreenshot(
-        new File("/data/local/tmp/screenshot.png"));
+    final File screenshot = new File("/data/local/tmp/screenshot.png");
+    screenshot.mkdirs();
+    final boolean result = UiDevice.getInstance().takeScreenshot(screenshot);
     return getSuccessResult(result);
   }
 }
\ No newline at end of file
diff --git a/app/android.js b/app/android.js
index fc376894b..7767e1ed0 100644
--- a/app/android.js
+++ b/app/android.js
@@ -747,9 +747,14 @@ Android.prototype.getScreenshot = function(cb) {
     }
   ],
   // Top level cb
-  function(){
+  function(err, res) {
+    var screenshotStatus = status.codes.Success.code;
+    try {
+      screenshotStatus = res[0].value === false ? status.codes.UnknownError.code : screenshotStatus;
+    } catch(e) {
+    }
     cb(null, {
-      status: status.codes.Success.code
+      status: screenshotStatus
       , value: b64data
     });
   });