OobeLocalizationTest.LocalizationTest : implement test timeouts.

chrome/browser/chromeos/login/oobe_localization_browsertest.cc

 // Copyright 2014 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 <stddef.h>
 
 #include "base/macros.h"
 #include "base/message_loop/message_loop.h"
 #include "base/strings/stringprintf.h"
 #include "base/task_runner.h"
+#include "base/time/time.h"
+#include "base/timer/timer.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/chromeos/customization/customization_document.h"
 #include "chrome/browser/chromeos/input_method/input_method_util.h"
 #include "chrome/browser/chromeos/login/login_manager_test.h"
 #include "chrome/browser/chromeos/login/login_wizard.h"
 #include "chrome/browser/chromeos/login/screens/network_screen.h"
 #include "chrome/browser/chromeos/login/test/js_checker.h"
 #include "chrome/browser/chromeos/login/ui/login_display_host.h"
 #include "chrome/browser/chromeos/login/wizard_controller.h"
@@ skipping 12 matching lines @@
 #include "ui/base/ime/chromeos/input_method_whitelist.h"
 
 namespace base {
 class TaskRunner;
 }
 
 namespace chromeos {
 
 namespace {
 
+// Timeout for RunLoop::Run() in this test.
+const int kTimeoutSeconds = 2;
+
 // OOBE constants.
 const char kLocaleSelect[] = "language-select";
 const char kKeyboardSelect[] = "keyboard-select";
 
 const char kUSLayout[] = "xkb:us::eng";
+class TimedRunLoop {
+ public:
+  TimedRunLoop(const base::TimeDelta& timeout,
+               const std::string& failure_message)
+      : timeout_(timeout), message_(failure_message) {}
+
+  // Returns true if Run() successfully finished,
+  // Returns false on timeout.
+  bool Run() {
+    base::OneShotTimer timer;
+    timer.Start(FROM_HERE, timeout_,
+                base::Bind(&TimedRunLoop::OnTimeout, base::Unretained(this)));
+    loop_.Run();
+    return result_;
+  }
+
+  void Quit() {
+    result_ = true;
+    loop_.Quit();
+  }
+
+  base::Closure QuitClosure() {
+    return base::Bind(&TimedRunLoop::Quit, base::Unretained(this));
+  }
+
+ private:
+  void OnTimeout() {
+    LOG(ERROR) << "Timeout waiting for: " << message_;
+    result_ = false;
+    loop_.Quit();
+  }
+
+  bool result_ = false;
+  const base::TimeDelta timeout_;
+  const std::string message_;
+  base::RunLoop loop_;
+
+  DISALLOW_COPY_AND_ASSIGN(TimedRunLoop);
+};
 
 class LanguageListWaiter : public NetworkScreen::Observer {
  public:
-  explicit LanguageListWaiter(base::RunLoop& loop)
+  LanguageListWaiter()
       : network_screen_(
             NetworkScreen::Get(WizardController::default_controller())),
-        loop_(loop) {
+        loop_(base::TimeDelta::FromSeconds(kTimeoutSeconds), "LanguageList") {
     network_screen_->AddObserver(this);
     CheckLanguageList();
   }
 
   ~LanguageListWaiter() override { network_screen_->RemoveObserver(this); }
 
   // NetworkScreen::Observer implementation:
   void OnLanguageListReloaded() override { CheckLanguageList(); }
 
+  // Returns true on success, false on timeout.
+  bool Wait() {
+    if (LanguageListReady())
+      return true;
+
+    return loop_.Run();
+  }
+
  private:
+  bool LanguageListReady() const {
+    return network_screen_->GetLanguageList();
+  }
+
   void CheckLanguageList() {
-    if (network_screen_->GetLanguageList())
+    if (LanguageListReady())
       loop_.Quit();
   }
 
   NetworkScreen* network_screen_;
-  base::RunLoop& loop_;
+  TimedRunLoop loop_;
 };
 
 }  // namespace
 
 struct LocalizationTestParams {
   const char* initial_locale;
   const char* keyboard_layout;
   const char* expected_locale;
   const char* expected_keyboard_layout;
   const char* expected_keyboard_select_control;
@@ skipping 65 matching lines @@
   // Verifies that |value| exists in |select_id|.
   bool VerifyOptionExists(const char* select_id, const char* value);
 
   // Dumps OOBE select control (language or keyboard) to string.
   std::string DumpOptions(const char* select_id);
 
  protected:
   // Runs the test for the given locale and keyboard layout.
   void RunLocalizationTest();
 
-  void WaitUntilJSIsReady() {
+  // Returns true on success, false on error.
+  bool WaitUntilJSIsReady() {
     LoginDisplayHost* host = LoginDisplayHost::default_host();
     if (!host)
-      return;
+      return false;
+
     OobeUI* oobe_ui = host->GetOobeUI();
     if (!oobe_ui)
-      return;
-    base::RunLoop run_loop;
+      return false;
+
+    TimedRunLoop run_loop(base::TimeDelta::FromSeconds(kTimeoutSeconds),
+                          "WaitUntilJSIsReady()");
     const bool oobe_ui_ready = oobe_ui->IsJSReady(run_loop.QuitClosure());
-    if (!oobe_ui_ready)
-      run_loop.Run();
+    if (oobe_ui_ready)
+      return true;
+
+    return run_loop.Run();
   }
 
  private:
   system::ScopedFakeStatisticsProvider fake_statistics_provider_;
   test::JSChecker checker;
 
   DISALLOW_COPY_AND_ASSIGN(OobeLocalizationTest);
 };
 
 OobeLocalizationTest::OobeLocalizationTest() : LoginManagerTest(false) {
@@ skipping 112 matching lines @@
   const std::string keyboard_layout(GetParam()->keyboard_layout);
   const std::string expected_locale(GetParam()->expected_locale);
   const std::string expected_keyboard_layout(
       GetParam()->expected_keyboard_layout);
   const std::string expected_keyboard_select_control(
       GetParam()->expected_keyboard_select_control);
 
   const std::string expected_keyboard_select =
       TranslateXKB2Extension(expected_keyboard_select_control);
 
-  {
-    base::RunLoop loop;
-    LanguageListWaiter waiter(loop);
-    loop.Run();
-  }
+  ASSERT_TRUE(LanguageListWaiter().Wait());
 
-  WaitUntilJSIsReady();
+  ASSERT_TRUE(WaitUntilJSIsReady());
 
   const std::string first_language =
       expected_locale.substr(0, expected_locale.find(','));
   bool done = false;
   const std::string waiting_script = base::StringPrintf(
       "var screenElement = document.getElementById('language-select');"
       "function SendReplyIfAcceptEnabled() {"
       "  if ($('language-select').value != '%s')"
       "    return false;"
       "  domAutomationController.send(true);"
@@ skipping 53 matching lines @@
   RunLocalizationTest();
 }
 
 INSTANTIATE_TEST_CASE_P(
     StructSequence,
     OobeLocalizationTest,
     testing::Range(&oobe_localization_test_parameters[0],
                    &oobe_localization_test_parameters[arraysize(
                        oobe_localization_test_parameters)]));
 }  // namespace chromeos