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/web_test_interfaces.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 "v8/include/v8.h"
20
21using blink::WebFrame;
22using blink::WebRuntimeFeatures;
23
24namespace content {
25
26namespace {
27ShellRenderProcessObserver* g_instance = NULL;
28}
29
30// static
31ShellRenderProcessObserver* ShellRenderProcessObserver::GetInstance() {
32  return g_instance;
33}
34
35ShellRenderProcessObserver::ShellRenderProcessObserver()
36    : main_test_runner_(NULL),
37      test_delegate_(NULL) {
38  CHECK(!g_instance);
39  g_instance = this;
40  RenderThread::Get()->AddObserver(this);
41  if (!CommandLine::ForCurrentProcess()->HasSwitch(switches::kDumpRenderTree))
42    return;
43  EnableRendererLayoutTestMode();
44}
45
46ShellRenderProcessObserver::~ShellRenderProcessObserver() {
47  CHECK(g_instance == this);
48  g_instance = NULL;
49}
50
51void ShellRenderProcessObserver::SetTestDelegate(WebTestDelegate* delegate) {
52  test_interfaces_->SetDelegate(delegate);
53  test_delegate_ = delegate;
54}
55
56void ShellRenderProcessObserver::SetMainWindow(RenderView* view) {
57  WebKitTestRunner* test_runner = WebKitTestRunner::Get(view);
58  test_interfaces_->SetWebView(view->GetWebView(), test_runner->proxy());
59  main_test_runner_ = test_runner;
60}
61
62void ShellRenderProcessObserver::WebKitInitialized() {
63  if (!CommandLine::ForCurrentProcess()->HasSwitch(switches::kDumpRenderTree))
64    return;
65
66  // We always expose GC to layout tests.
67  std::string flags("--expose-gc");
68  v8::V8::SetFlagsFromString(flags.c_str(), static_cast<int>(flags.size()));
69
70  if (!CommandLine::ForCurrentProcess()->HasSwitch(
71    switches::kStableReleaseMode)) {
72    WebRuntimeFeatures::enableTestOnlyFeatures(true);
73  }
74
75  test_interfaces_.reset(new WebTestInterfaces);
76  test_interfaces_->ResetAll();
77}
78
79void ShellRenderProcessObserver::OnRenderProcessShutdown() {
80  test_interfaces_.reset();
81}
82
83bool ShellRenderProcessObserver::OnControlMessageReceived(
84    const IPC::Message& message) {
85  bool handled = true;
86  IPC_BEGIN_MESSAGE_MAP(ShellRenderProcessObserver, message)
87    IPC_MESSAGE_HANDLER(ShellViewMsg_SetWebKitSourceDir, OnSetWebKitSourceDir)
88    IPC_MESSAGE_UNHANDLED(handled = false)
89  IPC_END_MESSAGE_MAP()
90
91  return handled;
92}
93
94void ShellRenderProcessObserver::OnSetWebKitSourceDir(
95    const base::FilePath& webkit_source_dir) {
96  webkit_source_dir_ = webkit_source_dir;
97}
98
99}  // namespace content
100