|
| 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 |
0 commit comments