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
|
// Copyright 2013 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 "base/test/power_monitor_test_base.h"
#include "content/browser/power_monitor_message_broadcaster.h"
#include "content/common/power_monitor_messages.h"
#include "ipc/ipc_sender.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace content {
class PowerMonitorMessageSender : public IPC::Sender {
public:
PowerMonitorMessageSender()
: power_state_changes_(0),
suspends_(0),
resumes_(0) {
}
virtual ~PowerMonitorMessageSender() {}
virtual bool Send(IPC::Message* msg) OVERRIDE {
switch (msg->type()) {
case PowerMonitorMsg_Suspend::ID:
suspends_++;
break;
case PowerMonitorMsg_Resume::ID:
resumes_++;
break;
case PowerMonitorMsg_PowerStateChange::ID:
power_state_changes_++;
break;
}
delete msg;
return true;
};
// Test status counts.
int power_state_changes() { return power_state_changes_; }
int suspends() { return suspends_; }
int resumes() { return resumes_; }
private:
int power_state_changes_; // Count of OnPowerStateChange notifications.
int suspends_; // Count of OnSuspend notifications.
int resumes_; // Count of OnResume notifications.
};
class PowerMonitorMessageBroadcasterTest : public testing::Test {
protected:
PowerMonitorMessageBroadcasterTest() {
power_monitor_source_ = new base::PowerMonitorTestSource();
power_monitor_.reset(new base::PowerMonitor(
scoped_ptr<base::PowerMonitorSource>(power_monitor_source_)));
}
virtual ~PowerMonitorMessageBroadcasterTest() {};
base::PowerMonitorTestSource* source() { return power_monitor_source_; }
base::PowerMonitor* monitor() { return power_monitor_.get(); }
private:
base::PowerMonitorTestSource* power_monitor_source_;
scoped_ptr<base::PowerMonitor> power_monitor_;
DISALLOW_COPY_AND_ASSIGN(PowerMonitorMessageBroadcasterTest);
};
TEST_F(PowerMonitorMessageBroadcasterTest, PowerMessageBroadcast) {
PowerMonitorMessageSender sender;
PowerMonitorMessageBroadcaster broadcaster(&sender);
// Sending resume when not suspended should have no effect.
source()->GenerateResumeEvent();
EXPECT_EQ(sender.resumes(), 0);
// Pretend we suspended.
source()->GenerateSuspendEvent();
EXPECT_EQ(sender.suspends(), 1);
// Send a second suspend notification. This should be suppressed.
source()->GenerateSuspendEvent();
EXPECT_EQ(sender.suspends(), 1);
// Pretend we were awakened.
source()->GenerateResumeEvent();
EXPECT_EQ(sender.resumes(), 1);
// Send a duplicate resume notification. This should be suppressed.
source()->GenerateResumeEvent();
EXPECT_EQ(sender.resumes(), 1);
// Pretend the device has gone on battery power
source()->GeneratePowerStateEvent(true);
EXPECT_EQ(sender.power_state_changes(), 1);
// Repeated indications the device is on battery power should be suppressed.
source()->GeneratePowerStateEvent(true);
EXPECT_EQ(sender.power_state_changes(), 1);
// Pretend the device has gone off battery power
source()->GeneratePowerStateEvent(false);
EXPECT_EQ(sender.power_state_changes(), 2);
// Repeated indications the device is off battery power should be suppressed.
source()->GeneratePowerStateEvent(false);
EXPECT_EQ(sender.power_state_changes(), 2);
}
} // namespace base
|
