Android content_browsertests: Allow processing java tasks in nested mode.

content/test/browser_test_message_pump_android.cc

 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "content/test/browser_test_message_pump_android.h"
 
+#include "base/android/jni_android.h"
+#include "base/android/scoped_java_ref.h"
+#include "base/lazy_instance.h"
 #include "base/logging.h"
 #include "base/synchronization/waitable_event.h"
 #include "base/time.h"
+#include "jni/BrowserTestSystemMessageHandler_jni.h"
+
+namespace {
+
+base::LazyInstance<base::android::ScopedJavaGlobalRef<jobject> >
+    g_message_handler_obj = LAZY_INSTANCE_INITIALIZER;
+
+}  // namespace
+
 
 namespace content {
 
 struct BrowserTestMessagePumpAndroid::RunState {
   RunState(base::MessagePump::Delegate* delegate, int run_depth)
       : delegate(delegate),
         run_depth(run_depth),
         should_quit(false),
         waitable_event(false, false) {
   }
@@ skipping 19 matching lines @@
 
 BrowserTestMessagePumpAndroid::~BrowserTestMessagePumpAndroid() {
 }
 
 void BrowserTestMessagePumpAndroid::Run(Delegate* delegate) {
   RunState state(delegate, state_ ? state_->run_depth + 1 : 1);
   RunState* previous_state = state_;
   state_ = &state;
   DCHECK(state_->run_depth <= 1) << "Only one level nested loops supported";
 
+  JNIEnv* env = base::android::AttachCurrentThread();
+  DCHECK(env);
+
+  // Need to cap the wait time to allow task processing on the java
+  // side. Otherwise, a long wait time on the native will starve java
+  // tasks.
+  base::TimeDelta max_delay = base::TimeDelta::FromMilliseconds(100);
+
   for (;;) {
     if (state_->should_quit)
       break;
 
     bool did_work = state_->delegate->DoWork();
     if (state_->should_quit)
       break;
 
     did_work |= state_->delegate->DoDelayedWork(&state_->delayed_work_time);
     if (state_->should_quit)
       break;
 
     if (did_work) {
       continue;
     }
 
     did_work = state_->delegate->DoIdleWork();
     if (state_->should_quit)
       break;
 
     if (did_work)
       continue;
 
+    // No native tasks to process right now. Process tasks from the Java
+    // System message handler. This will return when the java message queue
+    // is idle.
+    bool ret = Java_BrowserTestSystemMessageHandler_runNestedLoopTillIdle(env,
+        g_message_handler_obj.Get().obj());
+    CHECK(ret) << "Error running java message loop, tests will likely fail.";
+
     if (state_->delayed_work_time.is_null()) {
-      state_->waitable_event.Wait();
+      state_->waitable_event.TimedWait(max_delay);
     } else {
       base::TimeDelta delay =
           state_->delayed_work_time - base::TimeTicks::Now();
+      if (delay > max_delay)
+        delay = max_delay;
       if (delay > base::TimeDelta()) {
         state_->waitable_event.TimedWait(delay);
       } else {
         // It looks like delayed_work_time indicates a time in the past, so we
         // need to call DoDelayedWork now.
         state_->delayed_work_time = base::TimeTicks();
       }
     }
   }
 
   state_ = previous_state;
 }
 
+void BrowserTestMessagePumpAndroid::Start(
+    base::MessagePump::Delegate* delegate) {
+  JNIEnv* env = base::android::AttachCurrentThread();
+  DCHECK(env);
+  g_message_handler_obj.Get().Reset(
+      Java_BrowserTestSystemMessageHandler_create(env));
+
+  base::MessagePumpForUI::Start(delegate);
+}
+
 void BrowserTestMessagePumpAndroid::Quit() {
   if (state_) {
     state_->should_quit = true;
     state_->waitable_event.Signal();
     return;
   }
-
   base::MessagePumpForUI::Quit();
 }
 
 void BrowserTestMessagePumpAndroid::ScheduleWork() {
   if (state_) {
     state_->waitable_event.Signal();
     return;
   }
 
   base::MessagePumpForUI::ScheduleWork();
 }
 
 void BrowserTestMessagePumpAndroid::ScheduleDelayedWork(
     const base::TimeTicks& delayed_work_time) {
   if (state_) {
     // We know that we can't be blocked on Wait right now since this method can
     // only be called on the same thread as Run, so we only need to update our
     // record of how long to sleep when we do sleep.
     state_->delayed_work_time = delayed_work_time;
     return;
   }
 
   base::MessagePumpForUI::ScheduleDelayedWork(delayed_work_time);
 }
 
-}  // namespace base
+// static
+bool BrowserTestMessagePumpAndroid::RegisterJni(JNIEnv* env) {
+  return RegisterNativesImpl(env);
+}
+
+}  // namespace content