Fix flaky testAddVisualStateCallbackAfterRendererGone

android_webview/javatests/src/org/chromium/android_webview/test/RenderProcessGoneHelper.java

Index: android_webview/javatests/src/org/chromium/android_webview/test/RenderProcessGoneHelper.java
diff --git a/android_webview/javatests/src/org/chromium/android_webview/test/RenderProcessGoneHelper.java b/android_webview/javatests/src/org/chromium/android_webview/test/RenderProcessGoneHelper.java
index d6940efb15356707fd80a9997b42b7c0dea9e7d7..b32d48e1507be73414740f68ec2d609f77d3a654 100644
--- a/android_webview/javatests/src/org/chromium/android_webview/test/RenderProcessGoneHelper.java
+++ b/android_webview/javatests/src/org/chromium/android_webview/test/RenderProcessGoneHelper.java
@@ -19,6 +19,7 @@ public class RenderProcessGoneHelper implements RenderProcessGoneObserver {
 
     private int mState;
     private CallbackHelper mCallbackHelper;
+    private Runnable mOnRenderProcessGoneTask;
 
     public RenderProcessGoneHelper() {
         mCallbackHelper = new CallbackHelper();
@@ -48,6 +49,8 @@ public class RenderProcessGoneHelper implements RenderProcessGoneObserver {
 
     @Override
     public void onRenderProcessGone() {
+        if (mOnRenderProcessGoneTask != null) mOnRenderProcessGoneTask.run();
+
         mState = RENDER_PROCESS_GONE_NOTIFIED_TO_AW_CONTENTS;
         mCallbackHelper.notifyCalled();
     }
@@ -63,4 +66,8 @@ public class RenderProcessGoneHelper implements RenderProcessGoneObserver {
         mState = AW_CONTENTS_DESTROYED;
         mCallbackHelper.notifyCalled();
     }
+
+    public void setOnRenderProcessGoneTask(Runnable task) {
+        mOnRenderProcessGoneTask = task;
+    }
 }