flag_test.cc (6486B)
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 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 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 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 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 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