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
ash / ambient / ambient_weather_controller_unittest.cc [blame]
// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "ash/ambient/ambient_weather_controller.h"
#include "ash/ambient/model/ambient_weather_model.h"
#include "ash/ambient/test/ambient_ash_test_base.h"
#include "ash/constants/ash_pref_names.h"
#include "ash/public/cpp/ambient/fake_ambient_backend_controller_impl.h"
#include "ash/shell.h"
#include "base/run_loop.h"
#include "chromeos/ash/components/geolocation/simple_geolocation_provider.h"
#include "components/prefs/pref_service.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace ash {
class AmbientWeatherControllerTest : public AmbientAshTestBase {
public:
bool IsGeolocationUsageAllowed() {
CHECK_NE(weather_controller(), nullptr);
return weather_controller()->IsGeolocationUsageAllowed();
}
};
TEST_F(AmbientWeatherControllerTest, RefreshesWeather) {
auto* model = weather_controller()->weather_model();
EXPECT_FALSE(model->show_celsius());
EXPECT_TRUE(model->weather_condition_icon().isNull());
WeatherInfo info;
info.show_celsius = true;
info.condition_icon_url = "https://fake-icon-url";
info.temp_f = 70.0f;
backend_controller()->SetWeatherInfo(info);
// Check location permission is granted.
EXPECT_TRUE(IsGeolocationUsageAllowed());
auto weather_refresher = weather_controller()->CreateScopedRefresher();
base::RunLoop().RunUntilIdle();
EXPECT_TRUE(model->show_celsius());
EXPECT_FALSE(model->weather_condition_icon().isNull());
EXPECT_FLOAT_EQ(model->temperature_fahrenheit(), 70.0f);
// Refresh weather again after time passes.
info.show_celsius = false;
info.temp_f = -70.0f;
backend_controller()->SetWeatherInfo(info);
FastForwardByWeatherRefreshInterval();
EXPECT_FALSE(model->show_celsius());
EXPECT_FLOAT_EQ(model->temperature_fahrenheit(), -70.0f);
info.show_celsius = true;
info.temp_f = 70.0f;
backend_controller()->SetWeatherInfo(info);
// Should stop refreshing after the `weather_refresher` is destroyed.
weather_refresher.reset();
FastForwardByWeatherRefreshInterval();
// The old info should hold in the model since we're not refreshing.
EXPECT_FALSE(model->show_celsius());
EXPECT_FLOAT_EQ(model->temperature_fahrenheit(), -70.0f);
}
TEST_F(AmbientWeatherControllerTest, RespectsSystemLocationPermission) {
auto* model = weather_controller()->weather_model();
EXPECT_FALSE(model->show_celsius());
EXPECT_TRUE(model->weather_condition_icon().isNull());
// Check location permission is enabled by default.
EXPECT_TRUE(IsGeolocationUsageAllowed());
WeatherInfo info;
info.show_celsius = false;
info.condition_icon_url = "https://fake-icon-url";
info.temp_f = 70.0f;
backend_controller()->SetWeatherInfo(info);
// Disable location permission and check the weather model will not get
// updated. This should clear the weather model cache.
SimpleGeolocationProvider::GetInstance()->SetGeolocationAccessLevel(
GeolocationAccessLevel::kDisallowed);
EXPECT_FALSE(IsGeolocationUsageAllowed());
auto weather_refresher = weather_controller()->CreateScopedRefresher();
base::RunLoop().RunUntilIdle();
// Check against the default values of `AmbientWeatherModel`.
EXPECT_TRUE(model->show_celsius());
EXPECT_TRUE(model->weather_condition_icon().isNull());
EXPECT_FLOAT_EQ(model->temperature_fahrenheit(), 0.0f);
// Check again on next interval timelapse.
FastForwardByWeatherRefreshInterval();
EXPECT_TRUE(model->show_celsius());
EXPECT_TRUE(model->weather_condition_icon().isNull());
EXPECT_FLOAT_EQ(model->temperature_fahrenheit(), 0.0f);
// Enable location permission for system services and check the weather model
// will get updated.
SimpleGeolocationProvider::GetInstance()->SetGeolocationAccessLevel(
GeolocationAccessLevel::kOnlyAllowedForSystem);
EXPECT_TRUE(IsGeolocationUsageAllowed());
FastForwardByWeatherRefreshInterval();
EXPECT_FALSE(model->show_celsius());
EXPECT_FALSE(model->weather_condition_icon().isNull());
EXPECT_FLOAT_EQ(model->temperature_fahrenheit(), 70.0f);
// Enable location for all clients and check it's continued fetching new
// weather models.
SimpleGeolocationProvider::GetInstance()->SetGeolocationAccessLevel(
GeolocationAccessLevel::kAllowed);
EXPECT_TRUE(IsGeolocationUsageAllowed());
info.show_celsius = true;
info.temp_f = -70.0f;
backend_controller()->SetWeatherInfo(info);
FastForwardByWeatherRefreshInterval();
EXPECT_TRUE(model->show_celsius());
EXPECT_FALSE(model->weather_condition_icon().isNull());
EXPECT_FLOAT_EQ(model->temperature_fahrenheit(), -70.0f);
}
TEST_F(AmbientWeatherControllerTest, DisabledByPolicy) {
// Disable weather integration by policy.
auto* pref_service =
Shell::Get()->session_controller()->GetLastActiveUserPrefService();
pref_service->SetList(prefs::kContextualGoogleIntegrationsConfiguration, {});
// No weather should be fetched when policy is disabled.
auto weather_refresher = weather_controller()->CreateScopedRefresher();
base::RunLoop().RunUntilIdle();
EXPECT_EQ(backend_controller()->fetch_weather_count(), 0);
// Enable weather integration by policy.
base::Value::List enabled_integrations;
enabled_integrations.Append(prefs::kWeatherIntegrationName);
pref_service->SetList(prefs::kContextualGoogleIntegrationsConfiguration,
std::move(enabled_integrations));
// Weather should be fetched when policy is enabled.
EXPECT_EQ(backend_controller()->fetch_weather_count(), 1);
}
} // namespace ash