Index: chrome/browser/ui/constrained_window_tab_helper_unittest.cc |
diff --git a/chrome/browser/ui/constrained_window_tab_helper_unittest.cc b/chrome/browser/ui/constrained_window_tab_helper_unittest.cc |
index 81ce4314dfada265a74f2dae85aa61689539706b..03fcc706a14a8b0fd78e2d2d0127e6ed836b1589 100644 |
--- a/chrome/browser/ui/constrained_window_tab_helper_unittest.cc |
+++ b/chrome/browser/ui/constrained_window_tab_helper_unittest.cc |
@@ -4,28 +4,32 @@ |
#include "chrome/browser/ui/constrained_window.h" |
#include "chrome/browser/ui/constrained_window_tab_helper.h" |
-#include "chrome/browser/ui/tab_contents/tab_contents.h" |
-#include "chrome/browser/ui/tab_contents/test_tab_contents.h" |
+#include "chrome/test/base/chrome_render_view_host_test_harness.h" |
#include "content/public/test/test_browser_thread.h" |
#include "testing/gtest/include/gtest/gtest.h" |
using content::BrowserThread; |
-class ConstrainedWindowTabHelperUnit : public TabContentsTestHarness { |
+class ConstrainedWindowTabHelperTest : public ChromeRenderViewHostTestHarness { |
public: |
- ConstrainedWindowTabHelperUnit() |
- : TabContentsTestHarness(), |
+ ConstrainedWindowTabHelperTest() |
+ : ChromeRenderViewHostTestHarness(), |
ui_thread_(BrowserThread::UI, &message_loop_) { |
} |
+ virtual void SetUp() { |
+ ChromeRenderViewHostTestHarness::SetUp(); |
+ ConstrainedWindowTabHelper::CreateForWebContents(web_contents()); |
+ } |
+ |
private: |
content::TestBrowserThread ui_thread_; |
}; |
class ConstrainedWindowCloseTest : public ConstrainedWindow { |
public: |
- explicit ConstrainedWindowCloseTest(TabContents* tab_contents) |
- : tab_contents_(tab_contents) { |
+ explicit ConstrainedWindowCloseTest(content::WebContents* web_contents) |
+ : web_contents_(web_contents) { |
} |
virtual void ShowConstrainedWindow() {} |
@@ -34,18 +38,17 @@ class ConstrainedWindowCloseTest : public ConstrainedWindow { |
virtual void CloseConstrainedWindow() { |
ConstrainedWindowTabHelper* constrained_window_tab_helper = |
- ConstrainedWindowTabHelper::FromWebContents( |
- tab_contents_->web_contents()); |
+ ConstrainedWindowTabHelper::FromWebContents(web_contents_); |
constrained_window_tab_helper->WillClose(this); |
close_count++; |
} |
int close_count; |
- TabContents* tab_contents_; |
+ content::WebContents* web_contents_; |
}; |
-TEST_F(ConstrainedWindowTabHelperUnit, ConstrainedWindows) { |
- ConstrainedWindowCloseTest window(tab_contents()); |
+TEST_F(ConstrainedWindowTabHelperTest, ConstrainedWindows) { |
+ ConstrainedWindowCloseTest window(web_contents()); |
window.close_count = 0; |
ConstrainedWindowTabHelper* constrained_window_tab_helper = |
ConstrainedWindowTabHelper::FromWebContents(web_contents()); |