1// Copyright (c) 2011 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "chrome/test/ui/ui_test.h"
6
7#include "base/file_path.h"
8#include "base/message_loop.h"
9#include "chrome/browser/ui/webui/constrained_html_ui.h"
10#include "chrome/browser/ui/webui/html_dialog_ui.h"
11#include "chrome/common/url_constants.h"
12#include "chrome/test/in_process_browser_test.h"
13#include "chrome/test/ui_test_utils.h"
14#include "content/browser/renderer_host/render_widget_host_view.h"
15#include "content/browser/tab_contents/tab_contents.h"
16#include "testing/gmock/include/gmock/gmock.h"
17#include "testing/gtest/include/gtest/gtest.h"
18
19using testing::Eq;
20
21namespace {
22
23class TestHtmlDialogUIDelegate : public HtmlDialogUIDelegate {
24 public:
25  TestHtmlDialogUIDelegate() {}
26  virtual ~TestHtmlDialogUIDelegate() {}
27
28  // HTMLDialogUIDelegate implementation:
29  virtual bool IsDialogModal() const {
30    return true;
31  }
32  virtual std::wstring GetDialogTitle() const {
33    return std::wstring(L"Test");
34  }
35  virtual GURL GetDialogContentURL() const {
36    return GURL(chrome::kChromeUIConstrainedHTMLTestURL);
37  }
38  virtual void GetWebUIMessageHandlers(
39      std::vector<WebUIMessageHandler*>* handlers) const {}
40  virtual void GetDialogSize(gfx::Size* size) const {
41    size->set_width(400);
42    size->set_height(400);
43  }
44  virtual std::string GetDialogArgs() const {
45    return std::string();
46  }
47  virtual void OnDialogClosed(const std::string& json_retval) { }
48  virtual void OnCloseContents(TabContents* source, bool* out_close_dialog) {
49    if (out_close_dialog)
50      *out_close_dialog = true;
51  }
52  virtual bool ShouldShowDialogTitle() const { return true; }
53};
54
55}  // namespace
56
57class ConstrainedHtmlDialogBrowserTest : public InProcessBrowserTest {
58 public:
59  ConstrainedHtmlDialogBrowserTest() {}
60};
61
62// Tests that opening/closing the constrained window won't crash it.
63IN_PROC_BROWSER_TEST_F(ConstrainedHtmlDialogBrowserTest, BasicTest) {
64  // The delegate deletes itself.
65  HtmlDialogUIDelegate* delegate = new TestHtmlDialogUIDelegate();
66  TabContents* tab_contents = browser()->GetSelectedTabContents();
67  ASSERT_TRUE(tab_contents != NULL);
68
69  ConstrainedHtmlUI::CreateConstrainedHtmlDialog(browser()->profile(),
70                                                 delegate,
71                                                 tab_contents);
72
73  ASSERT_EQ(1U, tab_contents->constrained_window_count());
74}
75