1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
// 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.
//
// Unit tests for ChromeFrameActivex.
#include "chrome_frame/chrome_frame_activex.h"
#include "ceee/common/initializing_coclass.h"
#include "ceee/testing/utils/test_utils.h"
#include "ceee/testing/utils/mock_com.h"
#include "chrome_frame/test/chrome_tab_mocks.h"
#include "gtest/gtest.h"
#include "gmock/gmock.h"
namespace {
using testing::_;
using testing::AddRef;
using testing::DoAll;
using testing::IOleClientSiteMockImpl;
using testing::IServiceProviderMockImpl;
using testing::MockIChromeFramePrivileged;
using testing::Return;
using testing::SetArgumentPointee;
using testing::StrictMock;
// Just to unhide the static function.
class TestingChromeFrameActivex : public ChromeFrameActivex {
public:
static HRESULT TestingShouldShowVersionMismatchDialog(
bool is_privileged, IOleClientSite* client_site) {
return ShouldShowVersionMismatchDialog(is_privileged, client_site);
}
};
class MockChromeFrameClientSite
: public CComObjectRootEx<CComSingleThreadModel>,
public StrictMock<IOleClientSiteMockImpl>,
public StrictMock<IServiceProviderMockImpl> {
public:
DECLARE_NOT_AGGREGATABLE(MockChromeFrameClientSite)
BEGIN_COM_MAP(MockChromeFrameClientSite)
COM_INTERFACE_ENTRY(IOleClientSite)
COM_INTERFACE_ENTRY(IServiceProvider)
END_COM_MAP()
DECLARE_PROTECT_FINAL_CONSTRUCT()
HRESULT Initialize(MockChromeFrameClientSite** client_site) {
*client_site = this;
return S_OK;
}
};
TEST(ChromeFrameActivex, ShouldShowVersionMismatchDialog) {
// If not privileged, we always show the dialog.
ASSERT_TRUE(
TestingChromeFrameActivex::TestingShouldShowVersionMismatchDialog(
false, NULL));
MockChromeFrameClientSite* cs_keeper;
ScopedComPtr<IOleClientSite> cs;
ASSERT_HRESULT_SUCCEEDED(
InitializingCoClass<MockChromeFrameClientSite>::CreateInitialized(
&cs_keeper, cs.Receive()));
MockIChromeFramePrivileged* cfp_keeper;
ScopedComPtr<IChromeFramePrivileged> cfp;
ASSERT_HRESULT_SUCCEEDED(
InitializingCoClass<MockIChromeFramePrivileged>::CreateInitialized(
&cfp_keeper, cfp.Receive()));
EXPECT_CALL(*cs_keeper, QueryService(SID_ChromeFramePrivileged,
IID_IChromeFramePrivileged, _))
.WillRepeatedly(DoAll(AddRef(cfp.get()),
SetArgumentPointee<2>(static_cast<void*>(cfp)),
Return(S_OK)));
EXPECT_CALL(*cfp_keeper, ShouldShowVersionMismatchDialog())
.WillOnce(Return(S_OK));
ASSERT_TRUE(
TestingChromeFrameActivex::TestingShouldShowVersionMismatchDialog(
true, cs_keeper));
EXPECT_CALL(*cfp_keeper, ShouldShowVersionMismatchDialog())
.WillOnce(Return(S_FALSE));
ASSERT_FALSE(
TestingChromeFrameActivex::TestingShouldShowVersionMismatchDialog(
true, cs_keeper));
// Also test that we fail safe, showing the dialog unless we got
// an affirmative do-not-show.
EXPECT_CALL(*cfp_keeper, ShouldShowVersionMismatchDialog())
.WillOnce(Return(E_UNEXPECTED));
ASSERT_TRUE(
TestingChromeFrameActivex::TestingShouldShowVersionMismatchDialog(
true, cs_keeper));
}
} // namespace
|
