Allow WebUI Tests to use preLoad in HtmlDialogUI.

chrome/test/data/webui/async_gen.h

 // 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.
+#ifndef CHROME_TEST_DATA_WEBUI_ASYNC_GEN_H_
+#define CHROME_TEST_DATA_WEBUI_ASYNC_GEN_H_
+#pragma once
 
-#include "base/bind.h"
-#include "base/bind_helpers.h"
-#include "base/values.h"
 #include "chrome/browser/ui/webui/web_ui_browsertest.h"
 #include "testing/gmock/include/gmock/gmock.h"
 
+namespace base {
+

[James Hawkins, 2011/11/27 19:28:26]

Remove blank lines around the forward declaration.

[Sheridan Rawlins, 2011/11/29 06:57:16]

Done.

+class ListValue;
+
+}  // namespace base
+
 // C++ support class for javascript-generated asynchronous tests.
 class WebUIBrowserAsyncGenTest : public WebUIBrowserTest {
  public:
   WebUIBrowserAsyncGenTest();
   virtual ~WebUIBrowserAsyncGenTest();
 
  protected:
   class AsyncWebUIMessageHandler : public WebUIMessageHandler {
    public:
     AsyncWebUIMessageHandler();
     ~AsyncWebUIMessageHandler();
 
     MOCK_METHOD1(HandleTornDown, void(const base::ListValue*));
 
    private:
-    void HandleCallJS(const base::ListValue* list_value) {
-      std::string call_js;
-      ASSERT_TRUE(list_value->GetString(0, &call_js));
-      web_ui_->CallJavascriptFunction(call_js);
-    }
+    void HandleCallJS(const base::ListValue* list_value);
 
-    virtual void RegisterMessages() OVERRIDE {
-      web_ui_->RegisterMessageCallback(
-          "callJS", base::Bind(&AsyncWebUIMessageHandler::HandleCallJS,
-                               base::Unretained(this)));
-      web_ui_->RegisterMessageCallback(
-          "tornDown", base::Bind(&AsyncWebUIMessageHandler::HandleTornDown,
-                                 base::Unretained(this)));
-    }
+    // Implements WebUIMessageHandler.

[James Hawkins, 2011/11/27 19:28:26]

// WebUIMessageHandler implementation.

[Sheridan Rawlins, 2011/11/29 06:57:16]

Done.

+    virtual void RegisterMessages() OVERRIDE;
   };
 
   // Handler for this test fixture.
   ::testing::StrictMock<AsyncWebUIMessageHandler> message_handler_;
 
  private:
   // Provide this object's handler.
   virtual WebUIMessageHandler* GetMockMessageHandler() OVERRIDE {
     return &message_handler_;
   }
 
   virtual void SetUpOnMainThread() OVERRIDE {
     WebUIBrowserTest::SetUpOnMainThread();
     EXPECT_CALL(message_handler_, HandleTornDown(::testing::_));
   }
 
   DISALLOW_COPY_AND_ASSIGN(WebUIBrowserAsyncGenTest);
 };
+
+#endif  // CHROME_TEST_DATA_WEBUI_ASYNC_GEN_H_