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
|
// Copyright (c) 2015 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 "components/browser_watcher/endsession_watcher_window_win.h"
#include <windows.h>
#include <vector>
#include "base/bind.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace browser_watcher {
namespace {
class EndSessionWatcherWindowTest : public testing::Test {
public:
EndSessionWatcherWindowTest()
: num_callbacks_(0), last_message_(0), last_lparam_(0) {
}
void OnEndSessionMessage(UINT message, LPARAM lparam) {
++num_callbacks_;
last_message_ = message;
last_lparam_ = lparam;
}
size_t num_callbacks_;
UINT last_message_;
UINT last_lparam_;
};
} // namespace browser_watcher
TEST_F(EndSessionWatcherWindowTest, NoCallbackOnDestruction) {
{
EndSessionWatcherWindow watcher_window(
base::Bind(&EndSessionWatcherWindowTest::OnEndSessionMessage,
base::Unretained(this)));
}
EXPECT_EQ(num_callbacks_, 0);
EXPECT_EQ(last_lparam_, 0);
}
TEST_F(EndSessionWatcherWindowTest, IssuesCallbackOnMessage) {
EndSessionWatcherWindow watcher_window(
base::Bind(&EndSessionWatcherWindowTest::OnEndSessionMessage,
base::Unretained(this)));
::SendMessage(watcher_window.window(), WM_QUERYENDSESSION, TRUE, 0xCAFEBABE);
EXPECT_EQ(num_callbacks_, 1);
EXPECT_EQ(last_message_, WM_QUERYENDSESSION);
EXPECT_EQ(last_lparam_, 0xCAFEBABE);
::SendMessage(watcher_window.window(), WM_ENDSESSION, TRUE, 0xCAFE);
EXPECT_EQ(num_callbacks_, 2);
EXPECT_EQ(last_message_, WM_ENDSESSION);
EXPECT_EQ(last_lparam_, 0xCAFE);
// Verify that other messages don't pass through.
::SendMessage(watcher_window.window(), WM_CLOSE, TRUE, 0xCAFE);
EXPECT_EQ(num_callbacks_, 2);
}
} // namespace browser_watcher
|