Handle automation server crashes.

chrome_frame/test/mock_ie_event_sink_actions.h

 // Copyright (c) 2010 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef CHROME_FRAME_TEST_MOCK_IE_EVENT_SINK_ACTIONS_H_
 #define CHROME_FRAME_TEST_MOCK_IE_EVENT_SINK_ACTIONS_H_
 
+#include "base/scoped_bstr_win.h"
+#include "chrome/common/chrome_switches.h"
 #include "chrome_frame/test/chrome_frame_test_utils.h"
 #include "chrome_frame/test/simulate_input.h"
 #include "testing/gmock/include/gmock/gmock.h"
 
 namespace chrome_frame_test {
 
 MATCHER_P(UrlPathEq, url, "equals the path and query portion of the url") {
   return arg == chrome_frame_test::GetPathAndQueryFromUrl(url);
 }
 
@@ skipping 16 matching lines @@
 }
 
 ACTION_P(StopWindowWatching, mock) {
   mock->StopWatching();
 }
 
 ACTION_P(ExpectRendererHasFocus, mock) {
   mock->event_sink()->ExpectRendererWindowHasFocus();
 }
 
+ACTION_P2(ConnectDocPropNotifySink, mock, sink) {
+  ScopedComPtr<IDispatch> document;
+  mock->event_sink()->web_browser2()->get_Document(document.Receive());
+  EXPECT_TRUE(document != NULL);  // NOLINT
+  if (document) {
+    sink->Attach(document);
+  }
+}
+
+ACTION_P(DisconnectDocPropNotifySink, sink) {
+  sink->Detach();
+}
+
+ACTION_P3(DelayNavigateToCurrentUrl, mock, loop, delay) {
+  loop->PostDelayedTask(FROM_HERE, NewRunnableFunction(&NavigateToCurrentUrl,
+      mock), delay);
+}
+
+ACTION_P2(ExpectDocumentReadystate, mock, ready_state) {
+  mock->ExpectDocumentReadystate(ready_state);
+}
+
 ACTION_P8(DelayExecCommand, mock, loop, delay, cmd_group_guid, cmd_id,
           cmd_exec_opt, in_args, out_args) {
   loop->PostDelayedTask(FROM_HERE, NewRunnableMethod(mock->event_sink(),
       &IEEventSink::Exec, cmd_group_guid, cmd_id, cmd_exec_opt, in_args,
           out_args), delay);
 }
 
 ACTION_P3(DelayRefresh, mock, loop, delay) {
   loop->PostDelayedTask(FROM_HERE, NewRunnableMethod(mock->event_sink(),
       &IEEventSink::Refresh), delay);
@@ skipping 45 matching lines @@
 
 ACTION_P(VerifyAddressBarUrl, mock) {
   mock->event_sink()->ExpectAddressBarUrl(std::wstring(arg1));
 }
 
 ACTION_P4(DelaySendScanCode, loop, delay, c, mod) {
   loop->PostDelayedTask(FROM_HERE, NewRunnableFunction(
         simulate_input::SendScanCode, c, mod), delay);
 }
 
+
+ACTION(KillChromeFrameProcesses) {
+  KillAllNamedProcessesWithArgument(
+      UTF8ToWide(chrome_frame_test::kChromeImageName),
+      UTF8ToWide(switches::kChromeFrame));
+}
+
 // This function selects the address bar via the Alt+d shortcut. This is done
 // via a delayed task which executes after the delay which is passed in.
 // The subsequent operations like typing in the actual url and then hitting
 // enter to force the browser to navigate to it each execute as delayed tasks
 // timed at the delay passed in. The recommended value for delay is 2000 ms
 // to account for the time taken for the accelerator keys to be reflected back
 // from Chrome.
 ACTION_P3(TypeUrlInAddressBar, loop, url, delay) {
   loop->PostDelayedTask(FROM_HERE, NewRunnableFunction(
       simulate_input::SendCharA, 'd', simulate_input::ALT),
@@ skipping 12 matching lines @@
       next_delay);
 }
 
 ACTION_P5(ValidateWindowSize, mock, left, top, width, height) {
   int actual_left = 0;
   int actual_top = 0;
   int actual_width = 0;
   int actual_height = 0;
 
   IWebBrowser2* web_browser2 = mock->event_sink()->web_browser2();
-  web_browser2->get_Left(reinterpret_cast<long*>(&actual_left));
-  web_browser2->get_Top(reinterpret_cast<long*>(&actual_top));
-  web_browser2->get_Width(reinterpret_cast<long*>(&actual_width));
-  web_browser2->get_Height(reinterpret_cast<long*>(&actual_height));
+  web_browser2->get_Left(reinterpret_cast<long*>(&actual_left));  // NOLINT
+  web_browser2->get_Top(reinterpret_cast<long*>(&actual_top));  // NOLINT
+  web_browser2->get_Width(reinterpret_cast<long*>(&actual_width));  // NOLINT
+  web_browser2->get_Height(reinterpret_cast<long*>(&actual_height));  // NOLINT
 
   EXPECT_EQ(actual_left, left);
   EXPECT_EQ(actual_top, top);
 
   EXPECT_LE(actual_width, width + 20);
   EXPECT_GT(actual_width, width - 20);
   EXPECT_LE(actual_height, height + 100);
   EXPECT_GT(actual_height, height - 100);
 }
 
@@ skipping 15 matching lines @@
 
 ACTION_P2(DoDefaultUIActionInDocument, mock, matcher) {
   mock->event_sink()->WaitForDOMAccessibilityTree();
   DoDefaultUIAction(mock->event_sink()->GetRendererWindow(), matcher);
 }
 
 }  // namespace chrome_frame_test
 
 #endif  // CHROME_FRAME_TEST_MOCK_IE_EVENT_SINK_ACTIONS_H_