src/content/shell/renderer/shell_render_process_observer.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/shell/renderer/shell_render_process_observer.h"
   6
   7 #include "base/command_line.h"
   8 #include "content/public/common/content_client.h"
   9 #include "content/public/renderer/render_thread.h"
  10 #include "content/public/renderer/render_view.h"
  11 #include "content/public/test/layouttest_support.h"
  12 #include "content/shell/common/shell_messages.h"
  13 #include "content/shell/common/shell_switches.h"
  14 #include "content/shell/renderer/shell_content_renderer_client.h"
  15 #include "content/shell/renderer/test_runner/WebTestInterfaces.h"
  16 #include "content/shell/renderer/webkit_test_runner.h"
  17 #include "third_party/WebKit/public/web/WebRuntimeFeatures.h"
  18 #include "third_party/WebKit/public/web/WebView.h"
  19 #include "webkit/glue/webkit_glue.h"
  20
  21 using blink::WebFrame;
  22 using blink::WebRuntimeFeatures;
  23 using WebTestRunner::WebTestDelegate;
  24 using WebTestRunner::WebTestInterfaces;
  25
  26 namespace content {
  27
  28 namespace {
  29 ShellRenderProcessObserver* g_instance = NULL;
  30 }
  31
  32 // static
  33 ShellRenderProcessObserver* ShellRenderProcessObserver::GetInstance() {
  34   return g_instance;
  35 }
  36
  37 ShellRenderProcessObserver::ShellRenderProcessObserver()
  38     : main_test_runner_(NULL),
  39       test_delegate_(NULL) {
  40   CHECK(!g_instance);
  41   g_instance = this;
  42   RenderThread::Get()->AddObserver(this);
  43   if (!CommandLine::ForCurrentProcess()->HasSwitch(switches::kDumpRenderTree))
  44     return;
  45   EnableRendererLayoutTestMode();
  46 }
  47
  48 ShellRenderProcessObserver::~ShellRenderProcessObserver() {
  49   CHECK(g_instance == this);
  50   g_instance = NULL;
  51 }
  52
  53 void ShellRenderProcessObserver::SetTestDelegate(WebTestDelegate* delegate) {
  54   test_interfaces_->setDelegate(delegate);
  55   test_delegate_ = delegate;
  56 }
  57
  58 void ShellRenderProcessObserver::SetMainWindow(RenderView* view) {
  59   WebKitTestRunner* test_runner = WebKitTestRunner::Get(view);
  60   test_interfaces_->setWebView(view->GetWebView(), test_runner->proxy());
  61   main_test_runner_ = test_runner;
  62 }
  63
  64 void ShellRenderProcessObserver::WebKitInitialized() {
  65   if (!CommandLine::ForCurrentProcess()->HasSwitch(switches::kDumpRenderTree))
  66     return;
  67
  68   // We always expose GC to layout tests.
  69   webkit_glue::SetJavaScriptFlags(" --expose-gc");
  70
  71   if (!CommandLine::ForCurrentProcess()->HasSwitch(
  72     switches::kStableReleaseMode)) {
  73     WebRuntimeFeatures::enableTestOnlyFeatures(true);
  74   }
  75
  76   test_interfaces_.reset(new WebTestInterfaces);
  77   test_interfaces_->resetAll();
  78 }
  79
  80 void ShellRenderProcessObserver::OnRenderProcessShutdown() {
  81   test_interfaces_.reset();
  82 }
  83
  84 bool ShellRenderProcessObserver::OnControlMessageReceived(
  85     const IPC::Message& message) {
  86   bool handled = true;
  87   IPC_BEGIN_MESSAGE_MAP(ShellRenderProcessObserver, message)
  88     IPC_MESSAGE_HANDLER(ShellViewMsg_SetWebKitSourceDir, OnSetWebKitSourceDir)
  89     IPC_MESSAGE_UNHANDLED(handled = false)
  90   IPC_END_MESSAGE_MAP()
  91
  92   return handled;
  93 }
  94
  95 void ShellRenderProcessObserver::OnSetWebKitSourceDir(
  96     const base::FilePath& webkit_source_dir) {
  97   webkit_source_dir_ = webkit_source_dir;
  98 }
  99
 100 }  // namespace content