1// Copyright 2013 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/browser/notify_done_forwarder.h"
6
7#include "content/shell/browser/webkit_test_controller.h"
8#include "content/shell/common/shell_messages.h"
9
10namespace content {
11
12DEFINE_WEB_CONTENTS_USER_DATA_KEY(NotifyDoneForwarder);
13
14NotifyDoneForwarder::NotifyDoneForwarder(WebContents* web_contents)
15    : WebContentsObserver(web_contents) {}
16
17NotifyDoneForwarder::~NotifyDoneForwarder() {}
18
19bool NotifyDoneForwarder::OnMessageReceived(const IPC::Message& message) {
20  bool handled = true;
21  IPC_BEGIN_MESSAGE_MAP(NotifyDoneForwarder, message)
22    IPC_MESSAGE_HANDLER(ShellViewHostMsg_TestFinishedInSecondaryWindow,
23                        OnTestFinishedInSecondaryWindow)
24    IPC_MESSAGE_UNHANDLED(handled = false)
25  IPC_END_MESSAGE_MAP()
26  return handled;
27}
28
29void NotifyDoneForwarder::OnTestFinishedInSecondaryWindow() {
30  WebKitTestController::Get()->TestFinishedInSecondaryWindow();
31}
32
33}  // namespace content
34