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
151
152
153
154
155
156
157
158
159
160
161
162
|
// 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 "chrome/browser/chromeos/status/data_promo_notification.h"
#include "base/command_line.h"
#include "base/macros.h"
#include "base/run_loop.h"
#include "chrome/browser/chromeos/login/users/fake_chrome_user_manager.h"
#include "chrome/browser/chromeos/login/users/scoped_user_manager_enabler.h"
#include "chrome/test/base/testing_browser_process.h"
#include "chrome/test/base/testing_profile_manager.h"
#include "chromeos/chromeos_switches.h"
#include "chromeos/dbus/dbus_thread_manager.h"
#include "chromeos/dbus/shill_device_client.h"
#include "chromeos/dbus/shill_service_client.h"
#include "chromeos/login/login_state.h"
#include "chromeos/network/network_state_handler.h"
#include "testing/platform_test.h"
#include "third_party/cros_system_api/dbus/service_constants.h"
#include "ui/chromeos/network/network_connect.h"
#include "ui/message_center/message_center.h"
namespace {
const char kCellularDevicePath[] = "/device/stub_cellular_device1";
const char kCellularServicePath[] = "/service/cellular1";
const char kNotificationId[] = "chrome://settings/internet/data_saver";
const char kTestUserName[] = "test-user@example.com";
class NetworkConnectTestDelegate : public ui::NetworkConnect::Delegate {
public:
NetworkConnectTestDelegate() {}
~NetworkConnectTestDelegate() override {}
void ShowNetworkConfigure(const std::string& network_id) override {}
void ShowNetworkSettingsForGuid(const std::string& network_id) override {}
bool ShowEnrollNetwork(const std::string& network_id) override {
return false;
}
void ShowMobileSimDialog() override {}
void ShowMobileSetupDialog(const std::string& service_path) override {}
private:
DISALLOW_COPY_AND_ASSIGN(NetworkConnectTestDelegate);
};
} // namespace
namespace chromeos {
namespace test {
class DataPromoNotificationTest : public testing::Test {
public:
DataPromoNotificationTest() {}
~DataPromoNotificationTest() override {}
void SetUp() override {
testing::Test::SetUp();
base::CommandLine::ForCurrentProcess()->AppendSwitch(
switches::kEnableDataSaverPrompt);
DBusThreadManager::Initialize();
NetworkHandler::Initialize();
data_promo_notification_.reset(new DataPromoNotification);
SetupUser();
SetupNetworkShillState();
message_center::MessageCenter::Initialize();
base::RunLoop().RunUntilIdle();
network_connect_delegate_.reset(new NetworkConnectTestDelegate);
ui::NetworkConnect::Initialize(network_connect_delegate_.get());
}
void TearDown() override {
ui::NetworkConnect::Shutdown();
network_connect_delegate_.reset();
message_center::MessageCenter::Shutdown();
LoginState::Shutdown();
profile_manager_.reset();
user_manager_enabler_.reset();
data_promo_notification_.reset();
NetworkHandler::Shutdown();
DBusThreadManager::Shutdown();
testing::Test::TearDown();
}
protected:
void SetupUser() {
scoped_ptr<FakeChromeUserManager> user_manager(new FakeChromeUserManager());
const AccountId test_account_id(AccountId::FromUserEmail(kTestUserName));
user_manager->AddUser(test_account_id);
user_manager->LoginUser(test_account_id);
user_manager_enabler_.reset(
new ScopedUserManagerEnabler(user_manager.release()));
profile_manager_.reset(
new TestingProfileManager(TestingBrowserProcess::GetGlobal()));
ASSERT_TRUE(profile_manager_->SetUp());
profile_manager_->SetLoggedIn(true);
ASSERT_TRUE(user_manager::UserManager::Get()->GetPrimaryUser());
LoginState::Initialize();
LoginState::Get()->SetLoggedInState(LoginState::LOGGED_IN_ACTIVE,
LoginState::LOGGED_IN_USER_REGULAR);
}
void SetupNetworkShillState() {
base::RunLoop().RunUntilIdle();
// Create a cellular device with provider.
ShillDeviceClient::TestInterface* device_test =
DBusThreadManager::Get()->GetShillDeviceClient()->GetTestInterface();
device_test->ClearDevices();
device_test->AddDevice(kCellularDevicePath, shill::kTypeCellular,
"stub_cellular_device1");
base::DictionaryValue home_provider;
home_provider.SetString("name", "Cellular1_Provider");
home_provider.SetString("country", "us");
device_test->SetDeviceProperty(kCellularDevicePath,
shill::kHomeProviderProperty, home_provider);
// Create a cellular network and activate it.
ShillServiceClient::TestInterface* service_test =
DBusThreadManager::Get()->GetShillServiceClient()->GetTestInterface();
service_test->ClearServices();
service_test->AddService(kCellularServicePath, "cellular1_guid",
"cellular1" /* name */, shill::kTypeCellular,
"activated", true /* visible */);
service_test->SetServiceProperty(
kCellularServicePath, shill::kActivationStateProperty,
base::StringValue(shill::kActivationStateActivated));
service_test->SetServiceProperty(kCellularServicePath,
shill::kConnectableProperty,
base::FundamentalValue(true));
}
scoped_ptr<DataPromoNotification> data_promo_notification_;
scoped_ptr<NetworkConnectTestDelegate> network_connect_delegate_;
scoped_ptr<ScopedUserManagerEnabler> user_manager_enabler_;
scoped_ptr<TestingProfileManager> profile_manager_;
base::MessageLoop message_loop_;
private:
DISALLOW_COPY_AND_ASSIGN(DataPromoNotificationTest);
};
TEST_F(DataPromoNotificationTest, DataSaverNotification) {
message_center::MessageCenter* message_center =
message_center::MessageCenter::Get();
// Network setup shouldn't be enough to activate notification.
EXPECT_FALSE(message_center->FindVisibleNotificationById(kNotificationId));
ui::NetworkConnect::Get()->ConnectToNetwork(kCellularServicePath);
base::RunLoop().RunUntilIdle();
// Connecting to cellular network (which here makes it the default network)
// should trigger the Data Saver notification.
EXPECT_TRUE(message_center->FindVisibleNotificationById(kNotificationId));
}
} // namespace test
} // namespace chromeos
|
