Index: base/test/android/javatests/src/org/chromium/base/test/BaseJUnit4ClassRunner.java |
diff --git a/base/test/android/javatests/src/org/chromium/base/test/BaseJUnit4ClassRunner.java b/base/test/android/javatests/src/org/chromium/base/test/BaseJUnit4ClassRunner.java |
index 123545294772b3b0c7c667946ef031d1110dde25..7548efa4d1bbebf9066ae96f138b2e9a210c239c 100644 |
--- a/base/test/android/javatests/src/org/chromium/base/test/BaseJUnit4ClassRunner.java |
+++ b/base/test/android/javatests/src/org/chromium/base/test/BaseJUnit4ClassRunner.java |
@@ -12,6 +12,7 @@ import android.support.test.internal.util.AndroidRunnerParams; |
import org.junit.runner.notification.RunNotifier; |
import org.junit.runners.model.FrameworkMethod; |
import org.junit.runners.model.InitializationError; |
+import org.junit.runners.model.Statement; |
import org.chromium.base.CollectionUtil; |
import org.chromium.base.test.BaseTestResult.PreTestHook; |
@@ -156,4 +157,9 @@ public class BaseJUnit4ClassRunner extends AndroidJUnit4ClassRunner { |
} |
return false; |
} |
+ |
+ @Override |
+ protected Statement withAfters(FrameworkMethod method, Object test, Statement base) { |
+ return super.withAfters(method, test, new ScreenshotOnFailureStatement(base)); |
+ } |
} |