dump hprof if onFailure causes OutOfMemoryError

Change-Id: If582e3815a3f4e303cecdf1319763474823f297e

1 files changed, 3 insertions, 0 deletions

diff --git a/src/com/google/android/droiddriver/helpers/BaseDroidDriverTest.java b/src/com/google/android/droiddriver/helpers/BaseDroidDriverTest.java
index b262f49..09b4a36 100644
--- a/src/com/google/android/droiddriver/helpers/BaseDroidDriverTest.java
+++ b/src/com/google/android/droiddriver/helpers/BaseDroidDriverTest.java
@@ -112,6 +112,9 @@ public abstract class BaseDroidDriverTest<T extends Activity> extends
       // This method is for troubleshooting. Do not throw new error; we'll
       // throw the original failure.
       Logs.log(Log.WARN, e);
+      if (e instanceof OutOfMemoryError) {
+        dumpHprof();
+      }
     } finally {
       throw failure;
     }