| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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 "chrome/test/automation/automation_proxy.h" | 5 #include "chrome/test/automation/automation_proxy.h" |
| 6 | 6 |
| 7 #include <gtest/gtest.h> | 7 #include <gtest/gtest.h> |
| 8 | 8 |
| 9 #include <sstream> | 9 #include <sstream> |
| 10 | 10 |
| (...skipping 24 matching lines...) Expand all Loading... |
| 35 // This object allows messages received on the background thread to be | 35 // This object allows messages received on the background thread to be |
| 36 // properly triaged. | 36 // properly triaged. |
| 37 class AutomationMessageFilter : public IPC::ChannelProxy::MessageFilter { | 37 class AutomationMessageFilter : public IPC::ChannelProxy::MessageFilter { |
| 38 public: | 38 public: |
| 39 explicit AutomationMessageFilter(AutomationProxy* server) : server_(server) {} | 39 explicit AutomationMessageFilter(AutomationProxy* server) : server_(server) {} |
| 40 | 40 |
| 41 // Return true to indicate that the message was handled, or false to let | 41 // Return true to indicate that the message was handled, or false to let |
| 42 // the message be handled in the default way. | 42 // the message be handled in the default way. |
| 43 virtual bool OnMessageReceived(const IPC::Message& message) { | 43 virtual bool OnMessageReceived(const IPC::Message& message) { |
| 44 bool handled = true; | 44 bool handled = true; |
| 45 | |
| 46 IPC_BEGIN_MESSAGE_MAP(AutomationMessageFilter, message) | 45 IPC_BEGIN_MESSAGE_MAP(AutomationMessageFilter, message) |
| 47 IPC_MESSAGE_HANDLER_GENERIC(AutomationMsg_Hello, | 46 IPC_MESSAGE_HANDLER_GENERIC(AutomationMsg_Hello, |
| 48 OnAutomationHello(message)); | 47 OnAutomationHello(message)) |
| 49 IPC_MESSAGE_HANDLER_GENERIC( | 48 IPC_MESSAGE_HANDLER_GENERIC( |
| 50 AutomationMsg_InitialLoadsComplete, server_->SignalInitialLoads()); | 49 AutomationMsg_InitialLoadsComplete, server_->SignalInitialLoads()) |
| 51 IPC_MESSAGE_HANDLER(AutomationMsg_InitialNewTabUILoadComplete, | 50 IPC_MESSAGE_HANDLER(AutomationMsg_InitialNewTabUILoadComplete, |
| 52 NewTabLoaded); | 51 NewTabLoaded) |
| 53 IPC_MESSAGE_HANDLER_GENERIC( | 52 IPC_MESSAGE_HANDLER_GENERIC( |
| 54 AutomationMsg_InvalidateHandle, server_->InvalidateHandle(message)); | 53 AutomationMsg_InvalidateHandle, server_->InvalidateHandle(message)) |
| 55 IPC_MESSAGE_UNHANDLED(handled = false); | 54 IPC_MESSAGE_UNHANDLED(handled = false) |
| 56 IPC_END_MESSAGE_MAP() | 55 IPC_END_MESSAGE_MAP() |
| 57 | 56 |
| 58 return handled; | 57 return handled; |
| 59 } | 58 } |
| 60 | 59 |
| 61 virtual void OnFilterAdded(IPC::Channel* channel) { | 60 virtual void OnFilterAdded(IPC::Channel* channel) { |
| 62 server_->SetChannel(channel); | 61 server_->SetChannel(channel); |
| 63 } | 62 } |
| 64 | 63 |
| 65 virtual void OnFilterRemoved() { | 64 virtual void OnFilterRemoved() { |
| (...skipping 306 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 372 bool AutomationProxy::SendProxyConfig(const std::string& new_proxy_config) { | 371 bool AutomationProxy::SendProxyConfig(const std::string& new_proxy_config) { |
| 373 return Send(new AutomationMsg_SetProxyConfig(new_proxy_config)); | 372 return Send(new AutomationMsg_SetProxyConfig(new_proxy_config)); |
| 374 } | 373 } |
| 375 | 374 |
| 376 void AutomationProxy::Disconnect() { | 375 void AutomationProxy::Disconnect() { |
| 377 DCHECK(shutdown_event_.get() != NULL); | 376 DCHECK(shutdown_event_.get() != NULL); |
| 378 shutdown_event_->Signal(); | 377 shutdown_event_->Signal(); |
| 379 channel_.reset(); | 378 channel_.reset(); |
| 380 } | 379 } |
| 381 | 380 |
| 382 void AutomationProxy::OnMessageReceived(const IPC::Message& msg) { | 381 bool AutomationProxy::OnMessageReceived(const IPC::Message& msg) { |
| 383 // This won't get called unless AutomationProxy is run from | 382 // This won't get called unless AutomationProxy is run from |
| 384 // inside a message loop. | 383 // inside a message loop. |
| 385 NOTREACHED(); | 384 NOTREACHED(); |
| 385 return false; |
| 386 } | 386 } |
| 387 | 387 |
| 388 void AutomationProxy::OnChannelError() { | 388 void AutomationProxy::OnChannelError() { |
| 389 LOG(ERROR) << "Channel error in AutomationProxy."; | 389 LOG(ERROR) << "Channel error in AutomationProxy."; |
| 390 if (disconnect_on_failure_) | 390 if (disconnect_on_failure_) |
| 391 Disconnect(); | 391 Disconnect(); |
| 392 } | 392 } |
| 393 | 393 |
| 394 scoped_refptr<WindowProxy> AutomationProxy::GetActiveWindow() { | 394 scoped_refptr<WindowProxy> AutomationProxy::GetActiveWindow() { |
| 395 int handle = 0; | 395 int handle = 0; |
| (...skipping 144 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 540 password, | 540 password, |
| 541 &success)); | 541 &success)); |
| 542 // If message sending unsuccessful or test failed, return false. | 542 // If message sending unsuccessful or test failed, return false. |
| 543 return sent && success; | 543 return sent && success; |
| 544 } | 544 } |
| 545 #endif | 545 #endif |
| 546 | 546 |
| 547 bool AutomationProxy::ResetToDefaultTheme() { | 547 bool AutomationProxy::ResetToDefaultTheme() { |
| 548 return Send(new AutomationMsg_ResetToDefaultTheme()); | 548 return Send(new AutomationMsg_ResetToDefaultTheme()); |
| 549 } | 549 } |
| OLD | NEW |