OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "content/shell/browser/layout_test/notify_done_forwarder.h" | 5 #include "content/shell/browser/layout_test/secondary_test_window_observer.h" |
6 | 6 |
7 #include "content/shell/browser/layout_test/blink_test_controller.h" | 7 #include "content/shell/browser/layout_test/blink_test_controller.h" |
8 #include "content/shell/common/shell_messages.h" | 8 #include "content/shell/common/shell_messages.h" |
9 | 9 |
10 namespace content { | 10 namespace content { |
11 | 11 |
12 DEFINE_WEB_CONTENTS_USER_DATA_KEY(NotifyDoneForwarder); | 12 DEFINE_WEB_CONTENTS_USER_DATA_KEY(SecondaryTestWindowObserver); |
13 | 13 |
14 NotifyDoneForwarder::NotifyDoneForwarder(WebContents* web_contents) | 14 SecondaryTestWindowObserver::SecondaryTestWindowObserver( |
| 15 WebContents* web_contents) |
15 : WebContentsObserver(web_contents) {} | 16 : WebContentsObserver(web_contents) {} |
16 | 17 |
17 NotifyDoneForwarder::~NotifyDoneForwarder() {} | 18 SecondaryTestWindowObserver::~SecondaryTestWindowObserver() {} |
18 | 19 |
19 bool NotifyDoneForwarder::OnMessageReceived(const IPC::Message& message) { | 20 bool SecondaryTestWindowObserver::OnMessageReceived( |
| 21 const IPC::Message& message) { |
20 bool handled = true; | 22 bool handled = true; |
21 IPC_BEGIN_MESSAGE_MAP(NotifyDoneForwarder, message) | 23 IPC_BEGIN_MESSAGE_MAP(SecondaryTestWindowObserver, message) |
22 IPC_MESSAGE_HANDLER(ShellViewHostMsg_TestFinishedInSecondaryRenderer, | 24 IPC_MESSAGE_HANDLER(ShellViewHostMsg_TestFinishedInSecondaryRenderer, |
23 OnTestFinishedInSecondaryRenderer) | 25 OnTestFinishedInSecondaryRenderer) |
24 IPC_MESSAGE_UNHANDLED(handled = false) | 26 IPC_MESSAGE_UNHANDLED(handled = false) |
25 IPC_END_MESSAGE_MAP() | 27 IPC_END_MESSAGE_MAP() |
26 return handled; | 28 return handled; |
27 } | 29 } |
28 | 30 |
29 void NotifyDoneForwarder::OnTestFinishedInSecondaryRenderer() { | 31 void SecondaryTestWindowObserver::OnTestFinishedInSecondaryRenderer() { |
30 BlinkTestController::Get()->TestFinishedInSecondaryRenderer(); | 32 BlinkTestController::Get()->TestFinishedInSecondaryRenderer(); |
31 } | 33 } |
32 | 34 |
| 35 void SecondaryTestWindowObserver::RenderFrameCreated( |
| 36 RenderFrameHost* render_frame_host) { |
| 37 DCHECK(!BlinkTestController::Get()->IsMainWindow( |
| 38 WebContents::FromRenderFrameHost(render_frame_host))); |
| 39 BlinkTestController::Get()->HandleNewRenderFrameHost(render_frame_host); |
| 40 } |
| 41 |
| 42 void SecondaryTestWindowObserver::RenderFrameHostChanged( |
| 43 RenderFrameHost* old_host, |
| 44 RenderFrameHost* new_host) { |
| 45 DCHECK(!BlinkTestController::Get()->IsMainWindow( |
| 46 WebContents::FromRenderFrameHost(new_host))); |
| 47 BlinkTestController::Get()->HandleNewRenderFrameHost(new_host); |
| 48 } |
| 49 |
33 } // namespace content | 50 } // namespace content |
OLD | NEW |