| Index: chrome_frame/test/mock_ie_event_sink_test.cc
|
| ===================================================================
|
| --- chrome_frame/test/mock_ie_event_sink_test.cc (revision 55540)
|
| +++ chrome_frame/test/mock_ie_event_sink_test.cc (working copy)
|
| @@ -6,6 +6,7 @@
|
|
|
| #include <sstream>
|
|
|
| +#include "base/scoped_variant_win.h"
|
| #include "base/utf_string_conversions.h"
|
| #include "chrome_frame/test/mock_ie_event_sink_actions.h"
|
|
|
| @@ -140,6 +141,27 @@
|
| .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, ¶ms,
|
| + 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()) {
|
|
|