Provide ability for WebUIBrowserTests to run asynchronous tests.

chrome/browser/ui/webui/web_ui_test_handler.cc

 // Copyright (c) 2011 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 "chrome/browser/ui/webui/web_ui_test_handler.h"
 
 #include "base/utf_string_conversions.h"
 #include "base/values.h"
 #include "chrome/common/render_messages.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "content/browser/renderer_host/render_view_host.h"
 #include "content/common/notification_details.h"
 #include "content/common/notification_registrar.h"
 
+WebUITestHandler::WebUITestHandler()

[Paweł Hajdan Jr., 2011/08/05 18:17:14]

Have you considered sharing code with extension tests? Looks like they have some
similar facilities, and due to possible flakiness problems it'd be nice to fix
any bugs just once.

[Sheridan Rawlins, 2011/08/05 18:51:12]

David considered this some time ago.  I don't recall the full thought process,
but I know that we're testing WebUI here, specifically, and at this point, we've
added C++ generation, Preloading, and now async test result.  I don't think that
this can be merged with the extension tests at the current time (certainly not
this patch) but possibly at some point would be worth considering...

On 2011/08/05 18:17:14, Paweł Hajdan Jr. wrote:

+    : test_succeeded_(false),
+      is_waiting_(false),
+      test_done_async_(false),
+      test_succeeded_async_(false),
+      is_waiting_async_(false) {
+}
+
 void WebUITestHandler::PreloadJavaScript(const string16& js_text,
                                          RenderViewHost* preload_host) {
   DCHECK(preload_host);
   preload_host->Send(new ViewMsg_WebUIJavaScript(
       preload_host->routing_id(), string16(), js_text, 0,
       false));
 }
 
 void WebUITestHandler::RunJavaScript(const string16& js_text) {
   web_ui_->GetRenderViewHost()->ExecuteJavascriptInWebFrame(
       string16(), js_text);
 }
 
 bool WebUITestHandler::RunJavaScriptTestWithResult(const string16& js_text) {
   NotificationRegistrar notification_registrar;
   notification_registrar.Add(
       this, content::NOTIFICATION_EXECUTE_JAVASCRIPT_RESULT,
       Source<RenderViewHost>(web_ui_->GetRenderViewHost()));
   web_ui_->GetRenderViewHost()->ExecuteJavascriptInWebFrameNotifyResult(
       string16(), js_text);
   return WaitForResult();
 }
 
+bool WebUITestHandler::WaitForAsyncResult() {
+  if (!test_done_async_) {
+    is_waiting_async_ = true;
+    ui_test_utils::RunMessageLoop();
+    is_waiting_async_ = false;
+  }
+  return test_succeeded_async_;
+};
+
+void WebUITestHandler::RegisterMessages() {
+  web_ui_->RegisterMessageCallback("asyncTestResult", NewCallback(
+      this, &WebUITestHandler::HandleAsyncTestResult));
+}
+
+void WebUITestHandler::HandleAsyncTestResult(const ListValue* test_result) {
+  // Quit the message loop if we were waiting so Waiting process can get result
+  // or error. To ensure this gets done, do this before ASSERT* calls.
+  if (is_waiting_async_)
+    MessageLoopForUI::current()->Quit();
+
+  EXPECT_FALSE(test_done_async_);
+  test_done_async_ = true;
+
+  SCOPED_TRACE("WebUITestHandler::HandleAsyncTestResult");
+  HandleTestResult(test_result, &test_succeeded_async_);
+}
+
+void WebUITestHandler::HandleTestResult(const ListValue* test_result,
+                                        bool* success) {
+  *success = false;
+  ASSERT_TRUE(test_result->GetBoolean(0, success));
+  if (!success) {
+    std::string message;
+    ASSERT_TRUE(test_result->GetString(1, &message));
+    LOG(ERROR) << message;
+  }
+}
+
 void WebUITestHandler::Observe(int type,
                                const NotificationSource& source,
                                const NotificationDetails& details) {
   // Quit the message loop if we were waiting so Waiting process can get result
   // or error. To ensure this gets done, do this before ASSERT* calls.
   if (is_waiting_)
     MessageLoopForUI::current()->Quit();
 
   SCOPED_TRACE("WebUITestHandler::Observe");
   Value* value = Details<std::pair<int, Value*> >(details)->second;
-  ListValue* list_value;
-  ASSERT_TRUE(value->GetAsList(&list_value));
-  ASSERT_TRUE(list_value->GetBoolean(0, &test_succeeded_));
-  if (!test_succeeded_) {
-    std::string message;
-    ASSERT_TRUE(list_value->GetString(1, &message));
-    LOG(ERROR) << message;
-  }
+  ListValue* test_result;
+  ASSERT_TRUE(value->GetAsList(&test_result));
+  HandleTestResult(test_result, &test_succeeded_);
 }
 
 bool WebUITestHandler::WaitForResult() {
   is_waiting_ = true;
   ui_test_utils::RunMessageLoop();
   is_waiting_ = false;
   return test_succeeded_;
 }