From df7ef8ee871c53a26a7075ab21efd2f3a7638036 Mon Sep 17 00:00:00 2001
From: Antonio Zugaldia <antonio@silicalabs.com>
Date: Tue, 15 Sep 2015 14:18:22 -0400
Subject: [PATCH] flag to enable automatic screenshots on teardown

---
 .../com/mapbox/mapboxgl/testapp/BaseTest.java | 21 ++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/android/java/MapboxGLAndroidSDKTestApp/src/androidTest/java/com/mapbox/mapboxgl/testapp/BaseTest.java b/android/java/MapboxGLAndroidSDKTestApp/src/androidTest/java/com/mapbox/mapboxgl/testapp/BaseTest.java
index 7ec3c3181..038cc3099 100644
--- a/android/java/MapboxGLAndroidSDKTestApp/src/androidTest/java/com/mapbox/mapboxgl/testapp/BaseTest.java
+++ b/android/java/MapboxGLAndroidSDKTestApp/src/androidTest/java/com/mapbox/mapboxgl/testapp/BaseTest.java
@@ -14,6 +14,7 @@ public class BaseTest
         extends ActivityInstrumentationTestCase2<MainActivity> {
 
     private MainActivity mActivity;
+    private final static boolean screenshotOnTearDown = false;
 
     public BaseTest() {
         super(MainActivity.class);
@@ -36,12 +37,16 @@ public class BaseTest
 
     @Override
     protected void tearDown() throws Exception {
+        if (!screenshotOnTearDown) {
+            return;
+        }
+
         try {
             // Screenshots should be taken on the UI thread
             runTestOnUiThread(new Runnable() {
                 @Override
                 public void run() {
-                    ScreenshotUtil.take(getActivity());
+                    ScreenshotUtil.take(getActivity(), null);
                 }
             });
         } catch (Throwable throwable) {
@@ -51,4 +56,18 @@ public class BaseTest
         super.tearDown();
     }
 
+    protected void takeNamedScreenshot(final String name) {
+        try {
+            // Screenshots should be taken on the UI thread
+            runTestOnUiThread(new Runnable() {
+                @Override
+                public void run() {
+                    ScreenshotUtil.take(getActivity(), name);
+                }
+            });
+        } catch (Throwable throwable) {
+            throwable.printStackTrace();
+        }
+    }
+
 }