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
|
// 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.
#ifndef CHROME_FRAME_TEST_UTILS_H_
#define CHROME_FRAME_TEST_UTILS_H_
#include <string>
#include <atlbase.h>
#include <atlcom.h>
class FilePath;
extern const wchar_t kChromeFrameDllName[];
extern const wchar_t kChromeLauncherExeName[];
// Helper class used to register different chrome frame DLLs while running
// tests. The default constructor registers the DLL found in the build path.
// At destruction, again registers the DLL found in the build path if another
// DLL has since been registered. Triggers GTEST asserts on failure.
//
// TODO(robertshield): Ideally, make this class restore the originally
// registered chrome frame DLL (e.g. by looking in HKCR) on destruction.
class ScopedChromeFrameRegistrar {
public:
enum RegistrationType {
PER_USER,
SYSTEM_LEVEL,
};
explicit ScopedChromeFrameRegistrar(RegistrationType registration_type);
ScopedChromeFrameRegistrar(const std::wstring& path,
RegistrationType registration_type);
virtual ~ScopedChromeFrameRegistrar();
void RegisterChromeFrameAtPath(const std::wstring& path);
void UnegisterChromeFrameAtPath(const std::wstring& path);
void RegisterReferenceChromeFrameBuild();
std::wstring GetChromeFrameDllPath() const;
static void RegisterAtPath(const std::wstring& path,
RegistrationType registration_type);
static void UnregisterAtPath(const std::wstring& path,
RegistrationType registration_type);
static void RegisterDefaults();
static FilePath GetReferenceChromeFrameDllPath();
private:
// Contains the path of the most recently registered Chrome Frame DLL.
std::wstring new_chrome_frame_dll_path_;
// Contains the path of the Chrome Frame DLL to be registered at destruction.
std::wstring original_dll_path_;
// Indicates whether per user or per machine registration is needed.
RegistrationType registration_type_;
// We need to register the chrome path provider only once per process. This
// flag keeps track of that.
static bool register_chrome_path_provider_;
};
// Returns the path to the Chrome Frame DLL in the build directory. Assumes
// that the test executable is running from the build folder or a similar
// folder structure.
FilePath GetChromeFrameBuildPath();
// Callback description for onload, onloaderror, onmessage
static _ATL_FUNC_INFO g_single_param = {CC_STDCALL, VT_EMPTY, 1, {VT_VARIANT}};
// Simple class that forwards the callbacks.
template <typename T>
class DispCallback
: public IDispEventSimpleImpl<1, DispCallback<T>, &IID_IDispatch> {
public:
typedef HRESULT (T::*Method)(const VARIANT* param);
DispCallback(T* owner, Method method) : owner_(owner), method_(method) {
}
BEGIN_SINK_MAP(DispCallback)
SINK_ENTRY_INFO(1, IID_IDispatch, DISPID_VALUE, OnCallback, &g_single_param)
END_SINK_MAP()
virtual ULONG STDMETHODCALLTYPE AddRef() {
return owner_->AddRef();
}
virtual ULONG STDMETHODCALLTYPE Release() {
return owner_->Release();
}
STDMETHOD(OnCallback)(VARIANT param) {
return (owner_->*method_)(¶m);
}
IDispatch* ToDispatch() {
return reinterpret_cast<IDispatch*>(this);
}
T* owner_;
Method method_;
};
// Kills all running processes named |process_name| that have the string
// |argument| on their command line. Useful for killing all Chrome Frame
// instances of Chrome that all have --chrome-frame in their command line.
bool KillAllNamedProcessesWithArgument(const std::wstring& process_name,
const std::wstring& argument);
// If the workstation is locked and cannot receive user input.
bool IsWorkstationLocked();
#endif // CHROME_FRAME_TEST_UTILS_H_
|
