Share TabFirstRenderWatcher with HtmlDialogView.

chrome/browser/ui/views/html_dialog_view_browsertest.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 "base/file_path.h"
 #include "base/memory/singleton.h"
 #include "base/message_loop.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/views/html_dialog_view.h"
-#include "chrome/browser/ui/webui/html_dialog_ui.h"
+#include "chrome/browser/ui/webui/test_html_dialog_ui_delegate.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "content/browser/renderer_host/render_widget_host_view.h"
 #include "content/browser/tab_contents/tab_contents.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "views/widget/widget.h"
 
 using testing::Eq;
 
 namespace {
 
 // Window non-client-area means that the minimum size for the window
 // won't be the actual minimum size - our layout and resizing code
 // makes sure the chrome is always visible.
 const int kMinimumWidthToTestFor = 20;
 const int kMinimumHeightToTestFor = 30;
 
-class TestHtmlDialogUIDelegate : public HtmlDialogUIDelegate {
+// Initial size of HTMLDialog for SizeWindow test case. They must be different
+// from the above kMinimumWidthToTestFor/kMinimumHeightToTestFor.
+const int kInitialWidth = 40;
+const int kInitialHeight = 40;
+
+class TestHtmlDialogView: public HtmlDialogView {
  public:
-  TestHtmlDialogUIDelegate() {}
-  virtual ~TestHtmlDialogUIDelegate() {}
+  TestHtmlDialogView(Profile* profile, HtmlDialogUIDelegate* delegate)
+      : HtmlDialogView(profile, delegate),
+        painted_(false) {
+  }
 
-  // HTMLDialogUIDelegate implementation:
-  virtual bool IsDialogModal() const OVERRIDE {
-    return true;
+  bool painted() const {
+    return painted_;
   }
-  virtual string16 GetDialogTitle() const OVERRIDE {
-    return ASCIIToUTF16("Test");
+
+ protected:
+  virtual void OnTabMainFrameFirstRender() OVERRIDE {
+    HtmlDialogView::OnTabMainFrameFirstRender();
+    painted_ = true;
+    MessageLoop::current()->Quit();
   }
-  virtual GURL GetDialogContentURL() const OVERRIDE {
-    return GURL(chrome::kChromeUIChromeURLsURL);
-  }
-  virtual void GetWebUIMessageHandlers(
-      std::vector<WebUIMessageHandler*>* handlers) const OVERRIDE { }
-  virtual void GetDialogSize(gfx::Size* size) const OVERRIDE {
-    size->set_width(40);
-    size->set_height(40);
-  }
-  virtual std::string GetDialogArgs() const OVERRIDE {
-    return std::string();
-  }
-  virtual void OnDialogClosed(const std::string& json_retval) OVERRIDE { }
-  virtual void OnCloseContents(TabContents* source, bool* out_close_dialog)
-      OVERRIDE {
-    if (out_close_dialog)
-      *out_close_dialog = true;
-  }
-  virtual bool ShouldShowDialogTitle() const OVERRIDE { return true; }
+
+ private:
+  bool painted_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestHtmlDialogView);
 };
 
 }  // namespace
 
 class HtmlDialogBrowserTest : public InProcessBrowserTest {
  public:
   HtmlDialogBrowserTest() {}
 
   class WindowChangedObserver : public MessageLoopForUI::Observer {
    public:
@@ skipping 53 matching lines @@
 #else
 // http://code.google.com/p/chromium/issues/detail?id=52602
 // Windows has some issues resizing windows- an off by one problem,
 // and a minimum size that seems too big.  This file isn't included in
 // Mac builds yet. On Chrome OS, this test doesn't apply since ChromeOS
 // doesn't allow resizing of windows.
 #define MAYBE_SizeWindow DISABLED_SizeWindow
 #endif
 
 IN_PROC_BROWSER_TEST_F(HtmlDialogBrowserTest, MAYBE_SizeWindow) {
-  HtmlDialogUIDelegate* delegate = new TestHtmlDialogUIDelegate();
+  test::TestHtmlDialogUIDelegate* delegate = new test::TestHtmlDialogUIDelegate(
+      GURL(chrome::kChromeUIChromeURLsURL));
+  delegate->set_size(kInitialWidth, kInitialHeight);
 
   HtmlDialogView* html_view =
       new HtmlDialogView(browser()->profile(), delegate);
   TabContents* tab_contents = browser()->GetSelectedTabContents();
   ASSERT_TRUE(tab_contents != NULL);
   views::Widget::CreateWindowWithParent(html_view,
                                         tab_contents->GetDialogRootWindow());
   html_view->InitDialog();
   html_view->GetWidget()->Show();
 
@@ skipping 61 matching lines @@
   ui_test_utils::RunMessageLoop();
   actual_bounds = html_view->GetWidget()->GetClientAreaScreenBounds();
   EXPECT_LT(0, actual_bounds.width());
   EXPECT_LT(0, actual_bounds.height());
 
   MessageLoopForUI::current()->RemoveObserver(
       WindowChangedObserver::GetInstance());
 }
 
 // This is timing out about 5~10% of runs. See crbug.com/86059.
-IN_PROC_BROWSER_TEST_F(HtmlDialogBrowserTest, DISABLED_TestStateTransition) {
-  HtmlDialogUIDelegate* delegate = new TestHtmlDialogUIDelegate();
+IN_PROC_BROWSER_TEST_F(HtmlDialogBrowserTest, DISABLED_WebContentRendered) {
+  HtmlDialogUIDelegate* delegate = new test::TestHtmlDialogUIDelegate(
+      GURL(chrome::kChromeUIChromeURLsURL));
 
-  HtmlDialogView* html_view =
-      new HtmlDialogView(browser()->profile(), delegate);
+  TestHtmlDialogView* html_view =
+      new TestHtmlDialogView(browser()->profile(), delegate);
   TabContents* tab_contents = browser()->GetSelectedTabContents();
   ASSERT_TRUE(tab_contents != NULL);
   views::Widget::CreateWindowWithParent(html_view,
                                         tab_contents->GetDialogRootWindow());
-  // Test if the state transitions from INITIALIZED to -> PAINTED
-  EXPECT_EQ(HtmlDialogView::INITIALIZED, html_view->state_);
+  EXPECT_TRUE(html_view->initialized_);
 
   html_view->InitDialog();
   html_view->GetWidget()->Show();
 
-  MessageLoopForUI::current()->AddObserver(
-      WindowChangedObserver::GetInstance());
-  // We use busy loop because the state is updated in notifications.
-  while (html_view->state_ != HtmlDialogView::PAINTED)
-    MessageLoop::current()->RunAllPending();
+  // TestHtmlDialogView::OnTabMainFrameFirstRender() will Quit().
+  MessageLoopForUI::current()->Run();
 
-  EXPECT_EQ(HtmlDialogView::PAINTED, html_view->state_);
+  EXPECT_TRUE(html_view->painted());
 
-  MessageLoopForUI::current()->RemoveObserver(
-      WindowChangedObserver::GetInstance());
+  html_view->GetWidget()->Close();
 }