content/public/test/test_renderer_host.cc

   1 // Copyright (c) 2012 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 "content/public/test/test_renderer_host.h"
   6
   7 #include "base/run_loop.h"
   8 #include "content/browser/frame_host/navigation_entry_impl.h"
   9 #include "content/browser/renderer_host/render_view_host_factory.h"
  10 #include "content/browser/renderer_host/render_widget_host_impl.h"
  11 #include "content/browser/site_instance_impl.h"
  12 #include "content/public/browser/browser_thread.h"
  13 #include "content/public/browser/web_contents.h"
  14 #include "content/public/test/mock_render_process_host.h"
  15 #include "content/public/test/test_browser_context.h"
  16 #include "content/test/test_render_frame_host.h"
  17 #include "content/test/test_render_frame_host_factory.h"
  18 #include "content/test/test_render_view_host.h"
  19 #include "content/test/test_render_view_host_factory.h"
  20 #include "content/test/test_web_contents.h"
  21
  22 #if defined(OS_WIN)
  23 #include "ui/base/win/scoped_ole_initializer.h"
  24 #endif
  25
  26 #if defined(USE_AURA)
  27 #include "ui/aura/test/aura_test_helper.h"
  28 #include "ui/compositor/test/context_factories_for_test.h"
  29 #include "ui/wm/core/default_activation_client.h"
  30 #endif
  31
  32 namespace content {
  33
  34 // RenderFrameHostTester ------------------------------------------------------
  35
  36 // static
  37 RenderFrameHostTester* RenderFrameHostTester::For(RenderFrameHost* host) {
  38   return static_cast<TestRenderFrameHost*>(host);
  39 }
  40
  41 // RenderViewHostTester -------------------------------------------------------
  42
  43 // static
  44 RenderViewHostTester* RenderViewHostTester::For(RenderViewHost* host) {
  45   return static_cast<TestRenderViewHost*>(host);
  46 }
  47
  48 // static
  49 RenderViewHost* RenderViewHostTester::GetPendingForController(
  50     NavigationController* controller) {
  51   WebContentsImpl* web_contents = static_cast<WebContentsImpl*>(
  52       controller->GetWebContents());
  53   return web_contents->GetRenderManagerForTesting()->pending_render_view_host();
  54 }
  55
  56 // static
  57 bool RenderViewHostTester::IsRenderViewHostSwappedOut(RenderViewHost* rvh) {
  58   return static_cast<RenderViewHostImpl*>(rvh)->rvh_state() ==
  59          RenderViewHostImpl::STATE_SWAPPED_OUT;
  60 }
  61
  62 // static
  63 bool RenderViewHostTester::TestOnMessageReceived(RenderViewHost* rvh,
  64                                                  const IPC::Message& msg) {
  65   return static_cast<RenderViewHostImpl*>(rvh)->OnMessageReceived(msg);
  66 }
  67
  68 // static
  69 bool RenderViewHostTester::HasTouchEventHandler(RenderViewHost* rvh) {
  70   RenderWidgetHostImpl* host_impl = RenderWidgetHostImpl::From(rvh);
  71   return host_impl->has_touch_handler();
  72 }
  73
  74
  75 // RenderViewHostTestEnabler --------------------------------------------------
  76
  77 RenderViewHostTestEnabler::RenderViewHostTestEnabler()
  78     : rph_factory_(new MockRenderProcessHostFactory()),
  79       rvh_factory_(new TestRenderViewHostFactory(rph_factory_.get())),
  80       rfh_factory_(new TestRenderFrameHostFactory()) {}
  81
  82 RenderViewHostTestEnabler::~RenderViewHostTestEnabler() {
  83 }
  84
  85
  86 // RenderViewHostTestHarness --------------------------------------------------
  87
  88 RenderViewHostTestHarness::RenderViewHostTestHarness()
  89     : thread_bundle_options_(TestBrowserThreadBundle::DEFAULT) {}
  90
  91 RenderViewHostTestHarness::~RenderViewHostTestHarness() {
  92 }
  93
  94 NavigationController& RenderViewHostTestHarness::controller() {
  95   return web_contents()->GetController();
  96 }
  97
  98 WebContents* RenderViewHostTestHarness::web_contents() {
  99   return contents_.get();
 100 }
 101
 102 RenderViewHost* RenderViewHostTestHarness::rvh() {
 103   RenderViewHost* result = web_contents()->GetRenderViewHost();
 104   CHECK_EQ(result, web_contents()->GetMainFrame()->GetRenderViewHost());
 105   return result;
 106 }
 107
 108 RenderViewHost* RenderViewHostTestHarness::pending_rvh() {
 109   return pending_main_rfh() ? pending_main_rfh()->GetRenderViewHost() : NULL;
 110 }
 111
 112 RenderViewHost* RenderViewHostTestHarness::active_rvh() {
 113   return pending_rvh() ? pending_rvh() : rvh();
 114 }
 115
 116 RenderFrameHost* RenderViewHostTestHarness::main_rfh() {
 117   return web_contents()->GetMainFrame();
 118 }
 119
 120 RenderFrameHost* RenderViewHostTestHarness::pending_main_rfh() {
 121   return WebContentsTester::For(web_contents())->GetPendingMainFrame();
 122 }
 123
 124 BrowserContext* RenderViewHostTestHarness::browser_context() {
 125   return browser_context_.get();
 126 }
 127
 128 MockRenderProcessHost* RenderViewHostTestHarness::process() {
 129   return static_cast<MockRenderProcessHost*>(active_rvh()->GetProcess());
 130 }
 131
 132 void RenderViewHostTestHarness::DeleteContents() {
 133   SetContents(NULL);
 134 }
 135
 136 void RenderViewHostTestHarness::SetContents(WebContents* contents) {
 137   contents_.reset(contents);
 138 }
 139
 140 WebContents* RenderViewHostTestHarness::CreateTestWebContents() {
 141   // Make sure we ran SetUp() already.
 142 #if defined(OS_WIN)
 143   DCHECK(ole_initializer_ != NULL);
 144 #endif
 145 #if defined(USE_AURA)
 146   DCHECK(aura_test_helper_ != NULL);
 147 #endif
 148
 149   // This will be deleted when the WebContentsImpl goes away.
 150   SiteInstance* instance = SiteInstance::Create(browser_context_.get());
 151
 152   return TestWebContents::Create(browser_context_.get(), instance);
 153 }
 154
 155 void RenderViewHostTestHarness::NavigateAndCommit(const GURL& url) {
 156   static_cast<TestWebContents*>(web_contents())->NavigateAndCommit(url);
 157 }
 158
 159 void RenderViewHostTestHarness::Reload() {
 160   NavigationEntry* entry = controller().GetLastCommittedEntry();
 161   DCHECK(entry);
 162   controller().Reload(false);
 163   static_cast<TestRenderViewHost*>(
 164       rvh())->SendNavigateWithTransition(
 165           entry->GetPageID(), entry->GetURL(), PAGE_TRANSITION_RELOAD);
 166 }
 167
 168 void RenderViewHostTestHarness::FailedReload() {
 169   NavigationEntry* entry = controller().GetLastCommittedEntry();
 170   DCHECK(entry);
 171   controller().Reload(false);
 172   static_cast<TestRenderViewHost*>(
 173       rvh())->SendFailedNavigate(entry->GetPageID(), entry->GetURL());
 174 }
 175
 176 void RenderViewHostTestHarness::SetUp() {
 177   thread_bundle_.reset(new TestBrowserThreadBundle(thread_bundle_options_));
 178
 179 #if defined(OS_WIN)
 180   ole_initializer_.reset(new ui::ScopedOleInitializer());
 181 #endif
 182 #if defined(USE_AURA)
 183   // The ContextFactory must exist before any Compositors are created.
 184   bool enable_pixel_output = false;
 185   ui::ContextFactory* context_factory =
 186       ui::InitializeContextFactoryForTests(enable_pixel_output);
 187
 188   aura_test_helper_.reset(
 189       new aura::test::AuraTestHelper(base::MessageLoopForUI::current()));
 190   aura_test_helper_->SetUp(context_factory);
 191   new wm::DefaultActivationClient(aura_test_helper_->root_window());
 192 #endif
 193
 194   DCHECK(!browser_context_);
 195   browser_context_.reset(CreateBrowserContext());
 196
 197   SetContents(CreateTestWebContents());
 198 }
 199
 200 void RenderViewHostTestHarness::TearDown() {
 201   SetContents(NULL);
 202 #if defined(USE_AURA)
 203   aura_test_helper_->TearDown();
 204   ui::TerminateContextFactoryForTests();
 205 #endif
 206   // Make sure that we flush any messages related to WebContentsImpl destruction
 207   // before we destroy the browser context.
 208   base::RunLoop().RunUntilIdle();
 209
 210 #if defined(OS_WIN)
 211   ole_initializer_.reset();
 212 #endif
 213
 214   // Delete any RenderProcessHosts before the BrowserContext goes away.
 215   if (rvh_test_enabler_.rph_factory_)
 216     rvh_test_enabler_.rph_factory_.reset();
 217
 218   // Release the browser context by posting itself on the end of the task
 219   // queue. This is preferable to immediate deletion because it will behave
 220   // properly if the |rph_factory_| reset above enqueued any tasks which
 221   // depend on |browser_context_|.
 222   BrowserThread::DeleteSoon(content::BrowserThread::UI,
 223                             FROM_HERE,
 224                             browser_context_.release());
 225   thread_bundle_.reset();
 226 }
 227
 228 BrowserContext* RenderViewHostTestHarness::CreateBrowserContext() {
 229   return new TestBrowserContext();
 230 }
 231
 232 void RenderViewHostTestHarness::SetRenderProcessHostFactory(
 233     RenderProcessHostFactory* factory) {
 234     rvh_test_enabler_.rvh_factory_->set_render_process_host_factory(factory);
 235 }
 236
 237 }  // namespace content