1 // Copyright 2018 The Abseil Authors.
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 // https://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14
15 #include "absl/flags/flag.h"
16
17 #include <string>
18
19 #include "gtest/gtest.h"
20 #include "absl/flags/reflection.h"
21 #include "absl/time/civil_time.h"
22 #include "absl/time/time.h"
23
24 ABSL_FLAG(absl::CivilSecond, test_flag_civil_second,
25 absl::CivilSecond(2015, 1, 2, 3, 4, 5), "");
26 ABSL_FLAG(absl::CivilMinute, test_flag_civil_minute,
27 absl::CivilMinute(2015, 1, 2, 3, 4), "");
28 ABSL_FLAG(absl::CivilHour, test_flag_civil_hour, absl::CivilHour(2015, 1, 2, 3),
29 "");
30 ABSL_FLAG(absl::CivilDay, test_flag_civil_day, absl::CivilDay(2015, 1, 2), "");
31 ABSL_FLAG(absl::CivilMonth, test_flag_civil_month, absl::CivilMonth(2015, 1),
32 "");
33 ABSL_FLAG(absl::CivilYear, test_flag_civil_year, absl::CivilYear(2015), "");
34
35 ABSL_FLAG(absl::Duration, test_duration_flag, absl::Seconds(5),
36 "For testing support for Duration flags");
37 ABSL_FLAG(absl::Time, test_time_flag, absl::InfinitePast(),
38 "For testing support for Time flags");
39
40 namespace {
41
SetFlagValue(absl::string_view flag_name,absl::string_view value)42 bool SetFlagValue(absl::string_view flag_name, absl::string_view value) {
43 auto* flag = absl::FindCommandLineFlag(flag_name);
44 if (!flag) return false;
45 std::string err;
46 return flag->ParseFrom(value, &err);
47 }
48
GetFlagValue(absl::string_view flag_name,std::string & value)49 bool GetFlagValue(absl::string_view flag_name, std::string& value) {
50 auto* flag = absl::FindCommandLineFlag(flag_name);
51 if (!flag) return false;
52 value = flag->CurrentValue();
53 return true;
54 }
55
TEST(CivilTime,FlagSupport)56 TEST(CivilTime, FlagSupport) {
57 // Tests the default setting of the flags.
58 const absl::CivilSecond kDefaultSec(2015, 1, 2, 3, 4, 5);
59 EXPECT_EQ(absl::CivilSecond(kDefaultSec),
60 absl::GetFlag(FLAGS_test_flag_civil_second));
61 EXPECT_EQ(absl::CivilMinute(kDefaultSec),
62 absl::GetFlag(FLAGS_test_flag_civil_minute));
63 EXPECT_EQ(absl::CivilHour(kDefaultSec),
64 absl::GetFlag(FLAGS_test_flag_civil_hour));
65 EXPECT_EQ(absl::CivilDay(kDefaultSec),
66 absl::GetFlag(FLAGS_test_flag_civil_day));
67 EXPECT_EQ(absl::CivilMonth(kDefaultSec),
68 absl::GetFlag(FLAGS_test_flag_civil_month));
69 EXPECT_EQ(absl::CivilYear(kDefaultSec),
70 absl::GetFlag(FLAGS_test_flag_civil_year));
71
72 // Sets flags to a new value.
73 const absl::CivilSecond kNewSec(2016, 6, 7, 8, 9, 10);
74 absl::SetFlag(&FLAGS_test_flag_civil_second, absl::CivilSecond(kNewSec));
75 absl::SetFlag(&FLAGS_test_flag_civil_minute, absl::CivilMinute(kNewSec));
76 absl::SetFlag(&FLAGS_test_flag_civil_hour, absl::CivilHour(kNewSec));
77 absl::SetFlag(&FLAGS_test_flag_civil_day, absl::CivilDay(kNewSec));
78 absl::SetFlag(&FLAGS_test_flag_civil_month, absl::CivilMonth(kNewSec));
79 absl::SetFlag(&FLAGS_test_flag_civil_year, absl::CivilYear(kNewSec));
80
81 EXPECT_EQ(absl::CivilSecond(kNewSec),
82 absl::GetFlag(FLAGS_test_flag_civil_second));
83 EXPECT_EQ(absl::CivilMinute(kNewSec),
84 absl::GetFlag(FLAGS_test_flag_civil_minute));
85 EXPECT_EQ(absl::CivilHour(kNewSec),
86 absl::GetFlag(FLAGS_test_flag_civil_hour));
87 EXPECT_EQ(absl::CivilDay(kNewSec), absl::GetFlag(FLAGS_test_flag_civil_day));
88 EXPECT_EQ(absl::CivilMonth(kNewSec),
89 absl::GetFlag(FLAGS_test_flag_civil_month));
90 EXPECT_EQ(absl::CivilYear(kNewSec),
91 absl::GetFlag(FLAGS_test_flag_civil_year));
92 }
93
TEST(Duration,FlagSupport)94 TEST(Duration, FlagSupport) {
95 EXPECT_EQ(absl::Seconds(5), absl::GetFlag(FLAGS_test_duration_flag));
96
97 absl::SetFlag(&FLAGS_test_duration_flag, absl::Seconds(10));
98 EXPECT_EQ(absl::Seconds(10), absl::GetFlag(FLAGS_test_duration_flag));
99
100 EXPECT_TRUE(SetFlagValue("test_duration_flag", "20s"));
101 EXPECT_EQ(absl::Seconds(20), absl::GetFlag(FLAGS_test_duration_flag));
102
103 std::string current_flag_value;
104 EXPECT_TRUE(GetFlagValue("test_duration_flag", current_flag_value));
105 EXPECT_EQ("20s", current_flag_value);
106 }
107
TEST(Time,FlagSupport)108 TEST(Time, FlagSupport) {
109 EXPECT_EQ(absl::InfinitePast(), absl::GetFlag(FLAGS_test_time_flag));
110
111 const absl::Time t = absl::FromCivil(absl::CivilSecond(2016, 1, 2, 3, 4, 5),
112 absl::UTCTimeZone());
113 absl::SetFlag(&FLAGS_test_time_flag, t);
114 EXPECT_EQ(t, absl::GetFlag(FLAGS_test_time_flag));
115
116 // Successful parse
117 EXPECT_TRUE(SetFlagValue("test_time_flag", "2016-01-02T03:04:06Z"));
118 EXPECT_EQ(t + absl::Seconds(1), absl::GetFlag(FLAGS_test_time_flag));
119 EXPECT_TRUE(SetFlagValue("test_time_flag", "2016-01-02T03:04:07.0Z"));
120 EXPECT_EQ(t + absl::Seconds(2), absl::GetFlag(FLAGS_test_time_flag));
121 EXPECT_TRUE(SetFlagValue("test_time_flag", "2016-01-02T03:04:08.000Z"));
122 EXPECT_EQ(t + absl::Seconds(3), absl::GetFlag(FLAGS_test_time_flag));
123 EXPECT_TRUE(SetFlagValue("test_time_flag", "2016-01-02T03:04:09+00:00"));
124 EXPECT_EQ(t + absl::Seconds(4), absl::GetFlag(FLAGS_test_time_flag));
125 EXPECT_TRUE(SetFlagValue("test_time_flag", "2016-01-02T03:04:05.123+00:00"));
126 EXPECT_EQ(t + absl::Milliseconds(123), absl::GetFlag(FLAGS_test_time_flag));
127 EXPECT_TRUE(SetFlagValue("test_time_flag", "2016-01-02T03:04:05.123+08:00"));
128 EXPECT_EQ(t + absl::Milliseconds(123) - absl::Hours(8),
129 absl::GetFlag(FLAGS_test_time_flag));
130 EXPECT_TRUE(SetFlagValue("test_time_flag", "infinite-future"));
131 EXPECT_EQ(absl::InfiniteFuture(), absl::GetFlag(FLAGS_test_time_flag));
132 EXPECT_TRUE(SetFlagValue("test_time_flag", "infinite-past"));
133 EXPECT_EQ(absl::InfinitePast(), absl::GetFlag(FLAGS_test_time_flag));
134
135 EXPECT_FALSE(SetFlagValue("test_time_flag", "2016-01-02T03:04:06"));
136 EXPECT_FALSE(SetFlagValue("test_time_flag", "2016-01-02"));
137 EXPECT_FALSE(SetFlagValue("test_time_flag", "2016-01-02Z"));
138 EXPECT_FALSE(SetFlagValue("test_time_flag", "2016-01-02+00:00"));
139 EXPECT_FALSE(SetFlagValue("test_time_flag", "2016-99-99T03:04:06Z"));
140
141 EXPECT_TRUE(SetFlagValue("test_time_flag", "2016-01-02T03:04:05Z"));
142 std::string current_flag_value;
143 EXPECT_TRUE(GetFlagValue("test_time_flag", current_flag_value));
144 EXPECT_EQ("2016-01-02T03:04:05+00:00", current_flag_value);
145 }
146
147 } // namespace
148