1 // Copyright 2020 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "chrome/browser/chromeos/child_accounts/time_limits/persisted_app_info.h"
6
7 #include "base/time/time.h"
8 #include "base/values.h"
9 #include "chrome/browser/chromeos/child_accounts/time_limits/app_types.h"
10 #include "testing/gtest/include/gtest/gtest.h"
11
12 namespace chromeos {
13 namespace app_time {
14
15 using PersistedAppInfoTest = testing::Test;
16
TEST_F(PersistedAppInfoTest,RemoveActiveTimes)17 TEST_F(PersistedAppInfoTest, RemoveActiveTimes) {
18 AppId app = AppId(apps::mojom::AppType::kArc, "1");
19 AppState app_state = AppState::kAvailable;
20 base::TimeDelta running_active_time = base::TimeDelta::FromHours(5);
21
22 base::Time start_time = base::Time::Now();
23 base::TimeDelta activity = base::TimeDelta::FromHours(1);
24
25 AppActivity::ActiveTime to_remove =
26 AppActivity::ActiveTime(start_time, start_time + activity);
27 AppActivity::ActiveTime to_trim = AppActivity::ActiveTime(
28 start_time + 2 * activity, start_time + 3 * activity);
29 AppActivity::ActiveTime to_keep = AppActivity::ActiveTime(
30 start_time + 4 * activity, start_time + 5 * activity);
31
32 PersistedAppInfo app_info(app, app_state, running_active_time,
33 {{to_remove, to_trim, to_keep}});
34
35 EXPECT_TRUE(base::Contains(app_info.active_times(), to_remove));
36 EXPECT_TRUE(base::Contains(app_info.active_times(), to_trim));
37 EXPECT_TRUE(base::Contains(app_info.active_times(), to_keep));
38
39 base::Time report_time = start_time + 2.5 * activity;
40 app_info.RemoveActiveTimeEarlierThan(report_time);
41
42 EXPECT_EQ(2u, app_info.active_times().size());
43 EXPECT_FALSE(base::Contains(app_info.active_times(), to_remove));
44 EXPECT_TRUE(base::Contains(app_info.active_times(), to_keep));
45
46 const AppActivity::ActiveTime trimmed(report_time, to_trim.active_to());
47 EXPECT_TRUE(base::Contains(app_info.active_times(), trimmed));
48 }
49
TEST_F(PersistedAppInfoTest,UpdateAppActivityPreference)50 TEST_F(PersistedAppInfoTest, UpdateAppActivityPreference) {
51 AppId app = AppId(apps::mojom::AppType::kArc, "1");
52 AppState app_state = AppState::kAvailable;
53 base::TimeDelta running_active_time = base::TimeDelta::FromHours(5);
54
55 base::Time start_time = base::Time::Now();
56 base::TimeDelta activity = base::TimeDelta::FromHours(1);
57
58 AppActivity::ActiveTime entry1 =
59 AppActivity::ActiveTime(start_time, start_time + activity);
60 AppActivity::ActiveTime entry2 = AppActivity::ActiveTime(
61 start_time + 2 * activity, start_time + 3 * activity);
62 AppActivity::ActiveTime entry3 = AppActivity::ActiveTime(
63 start_time + 4 * activity, start_time + 5 * activity);
64
65 PersistedAppInfo app_info(app, app_state, running_active_time,
66 {{entry1, entry2, entry3}});
67 base::Value entry(base::Value::Type::DICTIONARY);
68
69 app_info.UpdateAppActivityPreference(&entry, /* replace */ false);
70 AppActivity::ActiveTime to_append = AppActivity::ActiveTime(
71 start_time + 6 * activity, start_time + 7 * activity);
72 PersistedAppInfo app_info2(app, app_state, running_active_time,
73 {{to_append}});
74 app_info2.UpdateAppActivityPreference(&entry, /* replace */ false);
75
76 base::Optional<PersistedAppInfo> updated_entry =
77 PersistedAppInfo::PersistedAppInfoFromDict(
78 &entry, /* include_app_activity_array */ true);
79 ASSERT_TRUE(updated_entry.has_value());
80
81 const std::vector<AppActivity::ActiveTime>& active_times =
82 updated_entry->active_times();
83
84 EXPECT_EQ(active_times.size(), 4u);
85 EXPECT_EQ(active_times[0], entry1);
86 EXPECT_EQ(active_times[1], entry2);
87 EXPECT_EQ(active_times[2], entry3);
88 EXPECT_EQ(active_times[3], to_append);
89
90 app_info2.UpdateAppActivityPreference(&entry, /* replace */ true);
91 base::Optional<PersistedAppInfo> final_entry =
92 PersistedAppInfo::PersistedAppInfoFromDict(
93 &entry, /* include_app_activity_array */ true);
94 EXPECT_TRUE(final_entry.has_value());
95 EXPECT_EQ(final_entry->active_times().size(), 1u);
96 EXPECT_EQ(final_entry->active_times()[0], to_append);
97 }
98
99 } // namespace app_time
100 } // namespace chromeos
101