Handle automation server crashes.

chrome_frame/test/mock_ie_event_sink_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 "chrome_frame/test/mock_ie_event_sink_test.h"
 
 #include <sstream>
 
+#include "base/scoped_variant_win.h"
 #include "base/utf_string_conversions.h"
 #include "chrome_frame/test/mock_ie_event_sink_actions.h"
 
 // Needed for CreateFunctor.
 #define GMOCK_MUTANT_INCLUDE_LATE_OBJECT_BINDING
 #include "testing/gmock_mutant.h"
 
 using testing::_;
 using testing::Cardinality;
 using testing::Exactly;
@@ skipping 114 matching lines @@
 
 void MockIEEventSink::ExpectAnyNavigations() {
   EXPECT_CALL(*this, OnBeforeNavigate2(_, _, _, _, _, _, _))
       .Times(testing::AnyNumber());
   EXPECT_CALL(*this, OnFileDownload(VARIANT_TRUE, _))
       .Times(testing::AnyNumber());
   EXPECT_CALL(*this, OnNavigateComplete2(_, _))
       .Times(testing::AnyNumber());
 }
 
+void MockIEEventSink::ExpectDocumentReadystate(int ready_state) {
+  ScopedComPtr<IWebBrowser2> browser(event_sink_->web_browser2());
+  EXPECT_TRUE(browser != NULL);
+  if (browser) {
+    ScopedComPtr<IDispatch> document;
+    browser->get_Document(document.Receive());
+    EXPECT_TRUE(document != NULL);
+    if (document) {
+      DISPPARAMS params = { 0 };
+      ScopedVariant result;
+      EXPECT_HRESULT_SUCCEEDED(document->Invoke(DISPID_READYSTATE, IID_NULL,
+          LOCALE_USER_DEFAULT, DISPATCH_PROPERTYGET, &params,
+          result.Receive(), NULL, NULL));
+      EXPECT_EQ(VT_I4, result.type());
+      if (result.type() == VT_I4) {
+        EXPECT_EQ(ready_state, static_cast<int>(V_I4(&result)));
+      }
+    }
+  }
+}
+
 // MockIEEventSinkTest methods
 MockIEEventSinkTest::MockIEEventSinkTest() : server_mock_(1337, L"127.0.0.1",
                                                           GetTestDataFolder()) {
   EXPECT_CALL(server_mock_, Get(_, StrCaseEq(L"/favicon.ico"), _))
       .WillRepeatedly(SendFast("HTTP/1.1 404 Not Found", ""));
 }
 
 void MockIEEventSinkTest::LaunchIEAndNavigate(const std::wstring& url) {
   LaunchIENavigateAndLoop(url, kChromeFrameLongNavigationTimeoutInSeconds);
 }
@@ skipping 11 matching lines @@
   loop_.RunFor(timeout);
 }
 
 std::wstring MockIEEventSinkTest::GetTestUrl(
     const std::wstring& relative_path) {
   return server_mock_.Resolve(relative_path.c_str());
 }
 
 }  // namespace chrome_frame_test