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
|
// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chromeos/ash/components/policy/weekly_time/test_support.h"
#include <vector>
#include "base/json/json_reader.h"
#include "base/time/time.h"
#include "base/values.h"
#include "chromeos/ash/components/policy/weekly_time/checked_util.h"
#include "chromeos/ash/components/policy/weekly_time/weekly_time_checked.h"
#include "chromeos/ash/components/policy/weekly_time/weekly_time_interval_checked.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace policy::weekly_time {
const char* DayToString(WeeklyTimeChecked::Day day_of_week) {
int day_of_week_index = static_cast<int>(day_of_week) - 1;
return WeeklyTimeChecked::kWeekDays[day_of_week_index];
}
base::Value::Dict BuildWeeklyTimeCheckedDict(WeeklyTimeChecked::Day day_of_week,
int milliseconds_since_midnight) {
base::Value::Dict dict;
const char* day_of_week_str = DayToString(day_of_week);
EXPECT_TRUE(dict.Set(WeeklyTimeChecked::kDayOfWeek, day_of_week_str));
EXPECT_TRUE(dict.Set(WeeklyTimeChecked::kMillisecondsSinceMidnight,
milliseconds_since_midnight));
return dict;
}
base::Value::Dict BuildWeeklyTimeCheckedDict(base::TimeDelta time_delta) {
auto w = WeeklyTimeChecked::FromTimeDelta(time_delta);
return BuildWeeklyTimeCheckedDict(w.day_of_week(),
w.milliseconds_since_midnight());
}
base::Value::Dict BuildWeeklyTimeIntervalCheckedDict(base::Value::Dict start,
base::Value::Dict end) {
base::Value::Dict dict;
EXPECT_TRUE(dict.Set(WeeklyTimeIntervalChecked::kStart, std::move(start)));
EXPECT_TRUE(dict.Set(WeeklyTimeIntervalChecked::kEnd, std::move(end)));
return dict;
}
base::Value::Dict BuildWeeklyTimeIntervalCheckedDict(base::TimeDelta start,
base::TimeDelta end) {
return BuildWeeklyTimeIntervalCheckedDict(BuildWeeklyTimeCheckedDict(start),
BuildWeeklyTimeCheckedDict(end));
}
base::Value::Dict BuildWeeklyTimeIntervalCheckedDict(
WeeklyTimeChecked::Day start_day,
int start_milliseconds,
WeeklyTimeChecked::Day end_day,
int end_milliseconds) {
return BuildWeeklyTimeIntervalCheckedDict(
BuildWeeklyTimeCheckedDict(start_day, start_milliseconds),
BuildWeeklyTimeCheckedDict(end_day, end_milliseconds));
}
base::Value::List BuildList(std::string_view json_str) {
std::optional<base::Value> value = base::JSONReader::Read(json_str);
if (!value.has_value()) {
ADD_FAILURE() << "JSON parsing failed: " << json_str;
return {};
}
if (!value->is_list()) {
ADD_FAILURE() << "JSON value not a list: " << json_str;
return {};
}
return std::move(value.value()).TakeList();
}
base::Value::List BuildList(base::Time now,
base::TimeDelta from_now,
base::TimeDelta duration) {
// TODO(isandrk): Can probably simplify function to only pass in start_time
// (which would be equal to now + from_now).
WeeklyTimeChecked current_weekly_time_checked =
WeeklyTimeChecked::FromTimeAsLocalTime(now);
base::TimeDelta current_time_of_week =
current_weekly_time_checked.ToTimeDelta();
base::TimeDelta start = current_time_of_week + from_now;
base::TimeDelta end = start + duration;
return base::Value::List().Append(
BuildWeeklyTimeIntervalCheckedDict(start, end));
}
std::vector<WeeklyTimeIntervalChecked> BuildIntervals(
std::string_view json_str) {
base::Value::List list = BuildList(json_str);
auto result = ExtractIntervalsFromList(list);
if (!result.has_value()) {
ADD_FAILURE() << "Couldn't parse intervals from list: "
<< list.DebugString();
return {};
}
return result.value();
}
base::Time TimeFromString(const char* str) {
base::Time time;
EXPECT_TRUE(base::Time::FromString(str, &time));
return time;
}
} // namespace policy::weekly_time
|
