Handle automation server crashes.

chrome_frame/test/ui_test.cc

 // 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.
 
 #include <mshtmcid.h>
 #include <string>
 
 #include "base/scoped_variant_win.h"
 #include "chrome/common/url_constants.h"
 #include "chrome_frame/test/chrome_frame_test_utils.h"
 #include "chrome_frame/test/mock_ie_event_sink_actions.h"
 #include "chrome_frame/test/mock_ie_event_sink_test.h"
 
+#include "testing/gmock_mutant.h"
+
 using testing::_;
 using testing::InSequence;
 using testing::StrCaseEq;
 using testing::StrEq;
 
 namespace chrome_frame_test {
 
 // This parameterized test fixture uses the MockIEEventSink and is used by
 // UI-related tests.
 class FullTabUITest : public MockIEEventSinkTest,
@@ skipping 204 matching lines @@
           VerifyAddressBarUrlWithGcf(&view_source_mock),
           CloseBrowserMock(&view_source_mock)));
 
   EXPECT_CALL(view_source_mock, OnQuit())
       .Times(testing::AtMost(1))
       .WillOnce(CloseBrowserMock(&ie_mock_));
 
   LaunchIEAndNavigate(GetSimplePageUrl());
 }
 
+void NavigateToCurrentUrl(MockIEEventSink* mock) {
+  IWebBrowser2* browser = mock->event_sink()->web_browser2();
+  DCHECK(browser);
+  ScopedBstr bstr;
+  HRESULT hr = browser->get_LocationURL(bstr.Receive());
+  EXPECT_HRESULT_SUCCEEDED(hr);
+  if (SUCCEEDED(hr)) {
+    DCHECK(bstr.Length());
+    VARIANT empty = ScopedVariant::kEmptyVariant;
+    hr = browser->Navigate(bstr, &empty, &empty, &empty, &empty);
+    EXPECT_HRESULT_SUCCEEDED(hr);
+  }
+}
+
+// Tests that Chrome gets re-instantiated after crash if we reload via
+// the address bar or via a new navigation.
+TEST_P(FullTabUITest, TabCrashReload) {
+  using testing::DoAll;
+
+  if (!GetParam().invokes_cf()) {
+    LOG(ERROR) << "Test needs CF.";
+    return;
+  }
+
+  MockPropertyNotifySinkListener prop_listener;
+  InSequence expect_in_sequence_for_scope;
+
+  EXPECT_CALL(ie_mock_, OnLoad(_, StrEq(GetSimplePageUrl())))
+      .WillOnce(DoAll(
+          ExpectRendererHasFocus(&ie_mock_),
+          ExpectDocumentReadystate(&ie_mock_, READYSTATE_COMPLETE),
+          ConnectDocPropNotifySink(&ie_mock_, &prop_listener),
+          KillChromeFrameProcesses()));
+
+  EXPECT_CALL(prop_listener, OnChanged(DISPID_READYSTATE))
+      .WillOnce(DoAll(
+          ExpectDocumentReadystate(&ie_mock_, READYSTATE_UNINITIALIZED),
+          DelayNavigateToCurrentUrl(&ie_mock_, &loop_, 10)));
+
+  EXPECT_CALL(ie_mock_, OnLoad(_, StrEq(GetSimplePageUrl())))
+      .WillOnce(CloseBrowserMock(&ie_mock_));
+
+  LaunchIEAndNavigate(GetSimplePageUrl());
+}
+
+// Tests if Chrome gets restarted after a crash by just refreshing the document.
+TEST_P(FullTabUITest, TabCrashRefresh) {
+  using testing::DoAll;
+
+  if (!GetParam().invokes_cf()) {
+    LOG(ERROR) << "Test needs CF.";
+    return;
+  }
+
+  MockPropertyNotifySinkListener prop_listener;
+  InSequence expect_in_sequence_for_scope;
+
+  EXPECT_CALL(ie_mock_, OnLoad(_, StrEq(GetSimplePageUrl())))
+      .WillOnce(DoAll(
+          ExpectRendererHasFocus(&ie_mock_),
+          ExpectDocumentReadystate(&ie_mock_, READYSTATE_COMPLETE),
+          ConnectDocPropNotifySink(&ie_mock_, &prop_listener),
+          KillChromeFrameProcesses()));
+
+  VARIANT empty = ScopedVariant::kEmptyVariant;
+  EXPECT_CALL(prop_listener, OnChanged(/*DISPID_READYSTATE*/_))
+      .WillOnce(DoAll(
+          DisconnectDocPropNotifySink(&prop_listener),
+          ExpectDocumentReadystate(&ie_mock_, READYSTATE_UNINITIALIZED),
+          DelayExecCommand(&ie_mock_, &loop_, 10, static_cast<GUID*>(NULL),
+              OLECMDID_REFRESH, 0, &empty, &empty)));
+
+  EXPECT_CALL(ie_mock_, OnLoad(_, StrEq(GetSimplePageUrl())))
+      .WillOnce(CloseBrowserMock(&ie_mock_));
+
+  LaunchIEAndNavigate(GetSimplePageUrl());
+}
+
 // Test fixture for tests related to the context menu UI. Since the context
 // menus for CF and IE are different, these tests are not parameterized.
 class ContextMenuTest : public MockIEEventSinkTest, public testing::Test {
  public:
   ContextMenuTest() {}
 
   virtual void SetUp() {
     // These are UI-related tests, so we do not care about the exact
     // navigations that occur.
     ie_mock_.ExpectAnyNavigations();
@@ skipping 285 matching lines @@
           SendExtendedKeysEnter(&loop_, 500, VK_DOWN, 2,
                                 simulate_input::NONE)));
 
   EXPECT_CALL(ie_mock_, OnLoad(IN_IE, StrEq(page2)))
       .WillOnce(CloseBrowserMock(&ie_mock_));
 
   LaunchIEAndNavigate(page1);
 }
 
 }  // namespace chrome_frame_test