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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
|
// Copyright (c) 2006-2008 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_CHROME_FRAME_UNITTESTS_H_
#define CHROME_FRAME_TEST_CHROME_FRAME_UNITTESTS_H_
#include <atlbase.h>
#include <atlcom.h>
#include <string>
#include <exdisp.h>
#include <exdispid.h>
#include "base/ref_counted.h"
#include "base/scoped_comptr_win.h"
#include "base/scoped_handle_win.h"
#include "googleurl/src/gurl.h"
#include "chrome_frame/test/http_server.h"
#include "testing/gtest/include/gtest/gtest.h"
// Class that:
// 1) Starts the local webserver,
// 2) Supports launching browsers - Internet Explorer and Firefox with local url
// 3) Wait the webserver to finish. It is supposed the test webpage to shutdown
// the server by navigating to "kill" page
// 4) Supports read the posted results from the test webpage to the "dump"
// webserver directory
class ChromeFrameTestWithWebServer: public testing::Test {
protected:
enum BrowserKind { INVALID, IE, FIREFOX, OPERA, SAFARI, CHROME };
bool LaunchBrowser(BrowserKind browser, const wchar_t* url);
bool WaitForTestToComplete(int milliseconds);
// Waits for the page to notify us of the window.onload event firing.
// Note that the milliseconds value is only approximate.
bool WaitForOnLoad(int milliseconds);
bool ReadResultFile(const std::wstring& file_name, std::string* data);
// Launches the specified browser and waits for the test to complete
// (see WaitForTestToComplete). Then checks that the outcome is OK.
// This function uses EXPECT_TRUE and ASSERT_TRUE for all steps performed
// hence no return value.
void SimpleBrowserTest(BrowserKind browser, const wchar_t* page,
const wchar_t* result_file_to_check);
// Same as SimpleBrowserTest but if the browser isn't installed (LaunchBrowser
// fails), the function will print out a warning but not treat the test
// as failed.
// Currently this is how we run Opera tests.
void OptionalBrowserTest(BrowserKind browser, const wchar_t* page,
const wchar_t* result_file_to_check);
// Test if chrome frame correctly reports its version.
void VersionTest(BrowserKind browser, const wchar_t* page,
const wchar_t* result_file_to_check);
void CloseBrowser();
// Ensures (well, at least tries to ensure) that the browser window has focus.
bool BringBrowserToTop();
// Returns true iff the specified result file contains 'expected result'.
bool CheckResultFile(const std::wstring& file_name,
const std::string& expected_result);
virtual void SetUp();
virtual void TearDown();
// Important: kind means "sheep" in Icelandic. ?:-o
const char* ToString(BrowserKind kind) {
switch (kind) {
case IE:
return "IE";
case FIREFOX:
return "Firefox";
case OPERA:
return "Opera";
case CHROME:
return "Chrome";
case SAFARI:
return "Safari";
default:
NOTREACHED();
break;
}
return "";
}
BrowserKind browser_;
std::wstring results_dir_;
ScopedHandle browser_handle_;
ChromeFrameHTTPServer server_;
};
// This class sets up event sinks to the IWebBrowser interface. Currently it
// subscribes to the following events:-
// 1. DISPID_BEFORENAVIGATE2
// 2. DISPID_NAVIGATEERROR
// 3. DISPID_NAVIGATECOMPLETE2
// 4. DISPID_NEWWINDOW3
// Other events can be subscribed to on an if needed basis.
class WebBrowserEventSink
: public CComObjectRootEx<CComSingleThreadModel>,
public IDispEventSimpleImpl<0, WebBrowserEventSink,
&DIID_DWebBrowserEvents2> {
public:
typedef IDispEventSimpleImpl<0, WebBrowserEventSink,
&DIID_DWebBrowserEvents2> DispEventsImpl;
WebBrowserEventSink() {}
~WebBrowserEventSink() {
Uninitialize();
}
void Uninitialize() {
if (web_browser2_.get()) {
DispEventUnadvise(web_browser2_);
web_browser2_.Release();
}
}
// Helper function to launch IE and navigate to a URL.
// Returns S_OK on success, S_FALSE if the test was not run, other
// errors on failure.
HRESULT LaunchIEAndNavigate(const std::wstring& navigate_url,
_IDispEvent* sink);
BEGIN_COM_MAP(WebBrowserEventSink)
END_COM_MAP()
BEGIN_SINK_MAP(WebBrowserEventSink)
SINK_ENTRY_INFO(0, DIID_DWebBrowserEvents2, DISPID_BEFORENAVIGATE2,
OnBeforeNavigate2, &kBeforeNavigate2Info)
SINK_ENTRY_INFO(0, DIID_DWebBrowserEvents2, DISPID_NAVIGATECOMPLETE2,
OnNavigateComplete2, &kNavigateComplete2Info)
SINK_ENTRY_INFO(0, DIID_DWebBrowserEvents2, DISPID_NAVIGATEERROR,
OnNavigateError, &kNavigateErrorInfo)
SINK_ENTRY_INFO(0, DIID_DWebBrowserEvents2, DISPID_NEWWINDOW3,
OnNewWindow3, &kNewWindow3Info)
END_SINK_MAP()
STDMETHOD_(void, OnNavigateError)(IDispatch* dispatch, VARIANT* url,
VARIANT* frame_name, VARIANT* status_code,
VARIANT* cancel) {
DLOG(INFO) << __FUNCTION__;
}
STDMETHOD(OnBeforeNavigate2)(IDispatch* dispatch, VARIANT* url, VARIANT*
flags, VARIANT* target_frame_name,
VARIANT* post_data, VARIANT* headers,
VARIANT_BOOL* cancel) {
DLOG(INFO) << __FUNCTION__;
return S_OK;
}
STDMETHOD_(void, OnNavigateComplete2)(IDispatch* dispatch, VARIANT* url) {
DLOG(INFO) << __FUNCTION__;
}
STDMETHOD_(void, OnNewWindow3)(IDispatch** dispatch, VARIANT_BOOL* Cancel,
DWORD flags, BSTR url_context, BSTR url) {
DLOG(INFO) << __FUNCTION__;
}
#ifdef _DEBUG
STDMETHOD(Invoke)(DISPID dispid, REFIID riid,
LCID lcid, WORD flags, DISPPARAMS* params, VARIANT* result,
EXCEPINFO* except_info, UINT* arg_error) {
DLOG(INFO) << __FUNCTION__ << L" disp id :" << dispid;
return DispEventsImpl::Invoke(dispid, riid, lcid, flags, params, result,
except_info, arg_error);
}
#endif // _DEBUG
IWebBrowser2* web_browser2() {
return web_browser2_.get();
}
protected:
static _ATL_FUNC_INFO kBeforeNavigate2Info;
static _ATL_FUNC_INFO kNavigateComplete2Info;
static _ATL_FUNC_INFO kNavigateErrorInfo;
static _ATL_FUNC_INFO kNewWindow3Info;
ScopedComPtr<IWebBrowser2> web_browser2_;
};
#endif // CHROME_FRAME_TEST_CHROME_FRAME_UNITTESTS_H_
|
