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
|
// Copyright (c) 2012 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/test/base/ui_test_utils.h"
#include "base/file_path.h"
#include "base/file_util.h"
#include "base/logging.h"
#include "base/message_loop.h"
#include "base/path_service.h"
#include "base/stringprintf.h"
#include "base/time.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/browser_window.h"
#include "chrome/browser/ui/views/frame/browser_view.h"
#include "chrome/browser/ui/window_snapshot/window_snapshot.h"
#include "ui/base/win/foreground_helper.h"
#include "ui/ui_controls/ui_controls.h"
#include "ui/views/focus/focus_manager.h"
namespace ui_test_utils {
namespace {
const char kSnapshotBaseName[] = "ChromiumSnapshot";
const char kSnapshotExtension[] = ".png";
FilePath GetSnapshotFileName(const FilePath& snapshot_directory) {
base::Time::Exploded the_time;
base::Time::Now().LocalExplode(&the_time);
std::string filename(StringPrintf("%s%04d%02d%02d%02d%02d%02d%s",
kSnapshotBaseName, the_time.year, the_time.month, the_time.day_of_month,
the_time.hour, the_time.minute, the_time.second, kSnapshotExtension));
FilePath snapshot_file = snapshot_directory.AppendASCII(filename);
if (file_util::PathExists(snapshot_file)) {
int index = 0;
std::string suffix;
FilePath trial_file;
do {
suffix = StringPrintf(" (%d)", ++index);
trial_file = snapshot_file.InsertBeforeExtensionASCII(suffix);
} while (file_util::PathExists(trial_file));
snapshot_file = trial_file;
}
return snapshot_file;
}
} // namespace
bool IsViewFocused(const Browser* browser, ViewID vid) {
BrowserWindow* browser_window = browser->window();
DCHECK(browser_window);
gfx::NativeWindow window = browser_window->GetNativeWindow();
DCHECK(window);
const views::Widget* widget =
views::Widget::GetTopLevelWidgetForNativeView(window);
DCHECK(widget);
const views::FocusManager* focus_manager = widget->GetFocusManager();
DCHECK(focus_manager);
return focus_manager->GetFocusedView()->id() == vid;
}
void ClickOnView(const Browser* browser, ViewID vid) {
BrowserWindow* browser_window = browser->window();
DCHECK(browser_window);
views::View* view =
reinterpret_cast<BrowserView*>(browser_window)->GetViewByID(vid);
DCHECK(view);
MoveMouseToCenterAndPress(
view,
ui_controls::LEFT,
ui_controls::DOWN | ui_controls::UP,
MessageLoop::QuitClosure());
content::RunMessageLoop();
}
void HideNativeWindow(gfx::NativeWindow window) {
// TODO(jcampan): retrieve the NativeWidgetWin and show/hide on it instead of
// using Windows API.
::ShowWindow(window, SW_HIDE);
}
bool ShowAndFocusNativeWindow(gfx::NativeWindow window) {
// TODO(jcampan): retrieve the NativeWidgetWin and show/hide on it instead of
// using Windows API.
::ShowWindow(window, SW_SHOW);
if (GetForegroundWindow() != window) {
VLOG(1) << "Forcefully refocusing front window";
ui::ForegroundHelper::SetForeground(window);
}
// ShowWindow does not necessarily activate the window. In particular if a
// window from another app is the foreground window then the request to
// activate the window fails. See SetForegroundWindow for details.
return GetForegroundWindow() == window;
}
void MoveMouseToCenterAndPress(views::View* view,
ui_controls::MouseButton button,
int state,
const base::Closure& task) {
DCHECK(view);
DCHECK(view->GetWidget());
gfx::Point view_center(view->width() / 2, view->height() / 2);
views::View::ConvertPointToScreen(view, &view_center);
ui_controls::SendMouseMove(view_center.x(), view_center.y());
ui_controls::SendMouseEventsNotifyWhenDone(button, state, task);
}
bool SaveScreenSnapshotToDirectory(const FilePath& directory,
FilePath* screenshot_path) {
bool succeeded = false;
FilePath out_path(GetSnapshotFileName(directory));
MONITORINFO monitor_info = {};
monitor_info.cbSize = sizeof(monitor_info);
HMONITOR main_monitor = MonitorFromWindow(NULL, MONITOR_DEFAULTTOPRIMARY);
if (GetMonitorInfo(main_monitor, &monitor_info)) {
RECT& rect = monitor_info.rcMonitor;
std::vector<unsigned char> png_data;
if (chrome::internal::GrabWindowSnapshot(NULL, &png_data,
gfx::Rect(rect.right - rect.left,
rect.bottom - rect.top))
&& png_data.size() <= INT_MAX) {
int bytes = static_cast<int>(png_data.size());
int written = file_util::WriteFile(
out_path, reinterpret_cast<char*>(&png_data[0]), bytes);
succeeded = (written == bytes);
}
}
if (succeeded && screenshot_path != NULL)
*screenshot_path = out_path;
return succeeded;
}
bool SaveScreenSnapshotToDesktop(FilePath* screenshot_path) {
FilePath desktop;
return PathService::Get(base::DIR_USER_DESKTOP, &desktop) &&
SaveScreenSnapshotToDirectory(desktop, screenshot_path);
}
} // namespace ui_test_utils
|
