forked from chromium/chromium
-
Notifications
You must be signed in to change notification settings - Fork 0
/
media_keyboard_hook_win_interactive_test.cc
124 lines (103 loc) · 3.9 KB
/
media_keyboard_hook_win_interactive_test.cc
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
// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "ui/events/keyboard_hook.h"
#include <windows.h>
#include "base/functional/bind.h"
#include "base/run_loop.h"
#include "base/test/task_environment.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/events/event.h"
namespace ui {
class MediaKeyboardHookWinInteractiveTest : public testing::Test {
public:
MediaKeyboardHookWinInteractiveTest()
: task_environment_(base::test::TaskEnvironment::MainThreadType::UI) {}
MediaKeyboardHookWinInteractiveTest(
const MediaKeyboardHookWinInteractiveTest&) = delete;
MediaKeyboardHookWinInteractiveTest& operator=(
const MediaKeyboardHookWinInteractiveTest&) = delete;
protected:
void SetUp() override {
keyboard_hook_ = KeyboardHook::CreateMediaKeyboardHook(base::BindRepeating(
&MediaKeyboardHookWinInteractiveTest::HandleKeyEvent,
base::Unretained(this)));
ASSERT_NE(nullptr, keyboard_hook_);
}
// Loop until we've received |num_events| key events from the hook.
void WaitForKeyEvents(uint32_t num_events) {
if (key_events_.size() >= num_events)
return;
num_key_events_to_wait_for_ = num_events;
key_event_wait_loop_.Run();
}
void SendKeyDown(KeyboardCode code) {
INPUT input;
input.type = INPUT_KEYBOARD;
input.ki.wVk = code;
input.ki.time = time_stamp_++;
input.ki.dwFlags = 0;
::SendInput(1, &input, sizeof(INPUT));
}
void SendKeyUp(KeyboardCode code) {
INPUT input;
input.type = INPUT_KEYBOARD;
input.ki.wVk = code;
input.ki.time = time_stamp_++;
input.ki.dwFlags = KEYEVENTF_KEYUP;
::SendInput(1, &input, sizeof(INPUT));
}
// Expect that we have received the correct number of key events.
void ExpectReceivedEventsCount(uint32_t count) {
EXPECT_EQ(count, key_events_.size());
}
// Expect that the key event received at |index| has the specified key code
// and type.
void ExpectReceivedEvent(uint32_t index, KeyboardCode code, EventType type) {
ASSERT_LT(index, key_events_.size());
KeyEvent* key_event = &key_events_.at(index);
EXPECT_EQ(code, key_event->key_code());
EXPECT_EQ(type, key_event->type());
}
private:
void HandleKeyEvent(KeyEvent* key_event) {
key_events_.push_back(*key_event);
key_event->SetHandled();
// If we've received the events we're waiting for, stop waiting.
if (key_event_wait_loop_.running() &&
key_events_.size() >= num_key_events_to_wait_for_) {
key_event_wait_loop_.Quit();
}
}
std::vector<KeyEvent> key_events_;
std::unique_ptr<KeyboardHook> keyboard_hook_;
base::test::TaskEnvironment task_environment_;
base::RunLoop key_event_wait_loop_;
uint32_t num_key_events_to_wait_for_ = 0;
DWORD time_stamp_ = 0;
};
// Test that we catch the different media key events.
TEST_F(MediaKeyboardHookWinInteractiveTest, AllMediaKeysAreCaught) {
SendKeyDown(ui::VKEY_MEDIA_PLAY_PAUSE);
SendKeyUp(ui::VKEY_MEDIA_PLAY_PAUSE);
SendKeyDown(ui::VKEY_MEDIA_STOP);
SendKeyUp(ui::VKEY_MEDIA_STOP);
SendKeyDown(ui::VKEY_MEDIA_NEXT_TRACK);
SendKeyUp(ui::VKEY_MEDIA_NEXT_TRACK);
SendKeyDown(ui::VKEY_MEDIA_PREV_TRACK);
SendKeyUp(ui::VKEY_MEDIA_PREV_TRACK);
// We should receive 8 different key events.
WaitForKeyEvents(8);
}
// Test that the received events have the proper state.
TEST_F(MediaKeyboardHookWinInteractiveTest, CallbackReceivesProperEvents) {
// Send a key down event and validate it when received through the hook.
SendKeyDown(ui::VKEY_MEDIA_PLAY_PAUSE);
WaitForKeyEvents(1);
ExpectReceivedEvent(/*index=*/0, ui::VKEY_MEDIA_PLAY_PAUSE, ET_KEY_PRESSED);
// Send a key up event and validate it when received through the hook.
SendKeyUp(ui::VKEY_MEDIA_PLAY_PAUSE);
WaitForKeyEvents(2);
ExpectReceivedEvent(/*index=*/1, ui::VKEY_MEDIA_PLAY_PAUSE, ET_KEY_RELEASED);
}
} // namespace ui