Abseil Common Libraries (C++) (grcp 依赖)
https://abseil.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
196 lines
7.3 KiB
196 lines
7.3 KiB
// |
|
// Copyright 2019 The Abseil Authors. |
|
// |
|
// Licensed under the Apache License, Version 2.0 (the "License"); |
|
// you may not use this file except in compliance with the License. |
|
// You may obtain a copy of the License at |
|
// |
|
// https://www.apache.org/licenses/LICENSE-2.0 |
|
// |
|
// Unless required by applicable law or agreed to in writing, software |
|
// distributed under the License is distributed on an "AS IS" BASIS, |
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
// See the License for the specific language governing permissions and |
|
// limitations under the License. |
|
|
|
#include "absl/flags/internal/commandlineflag.h" |
|
|
|
#include "gtest/gtest.h" |
|
#include "absl/flags/flag.h" |
|
#include "absl/flags/internal/registry.h" |
|
#include "absl/memory/memory.h" |
|
#include "absl/strings/match.h" |
|
#include "absl/strings/str_cat.h" |
|
|
|
ABSL_FLAG(int, int_flag, 201, "int_flag help"); |
|
ABSL_FLAG(std::string, string_flag, "dflt", |
|
absl::StrCat("string_flag", " help")); |
|
ABSL_RETIRED_FLAG(bool, bool_retired_flag, false, "bool_retired_flag help"); |
|
|
|
namespace { |
|
|
|
namespace flags = absl::flags_internal; |
|
|
|
class CommandLineFlagTest : public testing::Test { |
|
protected: |
|
void SetUp() override { flag_saver_ = absl::make_unique<flags::FlagSaver>(); } |
|
void TearDown() override { flag_saver_.reset(); } |
|
|
|
private: |
|
std::unique_ptr<flags::FlagSaver> flag_saver_; |
|
}; |
|
|
|
TEST_F(CommandLineFlagTest, TestAttributesAccessMethods) { |
|
auto* flag_01 = flags::FindCommandLineFlag("int_flag"); |
|
|
|
ASSERT_TRUE(flag_01); |
|
EXPECT_EQ(flag_01->Name(), "int_flag"); |
|
EXPECT_EQ(flag_01->Help(), "int_flag help"); |
|
EXPECT_EQ(flag_01->Typename(), ""); |
|
EXPECT_TRUE(!flag_01->IsRetired()); |
|
EXPECT_TRUE(flag_01->IsOfType<int>()); |
|
EXPECT_TRUE(absl::EndsWith( |
|
flag_01->Filename(), |
|
"absl/flags/internal/commandlineflag_test.cc")); |
|
|
|
auto* flag_02 = flags::FindCommandLineFlag("string_flag"); |
|
|
|
ASSERT_TRUE(flag_02); |
|
EXPECT_EQ(flag_02->Name(), "string_flag"); |
|
EXPECT_EQ(flag_02->Help(), "string_flag help"); |
|
EXPECT_EQ(flag_02->Typename(), ""); |
|
EXPECT_TRUE(!flag_02->IsRetired()); |
|
EXPECT_TRUE(flag_02->IsOfType<std::string>()); |
|
EXPECT_TRUE(absl::EndsWith( |
|
flag_02->Filename(), |
|
"absl/flags/internal/commandlineflag_test.cc")); |
|
|
|
auto* flag_03 = flags::FindRetiredFlag("bool_retired_flag"); |
|
|
|
ASSERT_TRUE(flag_03); |
|
EXPECT_EQ(flag_03->Name(), "bool_retired_flag"); |
|
EXPECT_EQ(flag_03->Help(), ""); |
|
EXPECT_EQ(flag_03->Typename(), ""); |
|
EXPECT_TRUE(flag_03->IsRetired()); |
|
EXPECT_TRUE(flag_03->IsOfType<bool>()); |
|
EXPECT_EQ(flag_03->Filename(), "RETIRED"); |
|
} |
|
|
|
// -------------------------------------------------------------------- |
|
|
|
TEST_F(CommandLineFlagTest, TestValueAccessMethods) { |
|
absl::SetFlag(&FLAGS_int_flag, 301); |
|
auto* flag_01 = flags::FindCommandLineFlag("int_flag"); |
|
|
|
ASSERT_TRUE(flag_01); |
|
EXPECT_EQ(flag_01->CurrentValue(), "301"); |
|
EXPECT_EQ(flag_01->DefaultValue(), "201"); |
|
|
|
absl::SetFlag(&FLAGS_string_flag, "new_str_value"); |
|
auto* flag_02 = flags::FindCommandLineFlag("string_flag"); |
|
|
|
ASSERT_TRUE(flag_02); |
|
EXPECT_EQ(flag_02->CurrentValue(), "new_str_value"); |
|
EXPECT_EQ(flag_02->DefaultValue(), "dflt"); |
|
} |
|
|
|
// -------------------------------------------------------------------- |
|
|
|
TEST_F(CommandLineFlagTest, TestSetFromStringCurrentValue) { |
|
std::string err; |
|
|
|
auto* flag_01 = flags::FindCommandLineFlag("int_flag"); |
|
EXPECT_FALSE(flag_01->IsSpecifiedOnCommandLine()); |
|
|
|
EXPECT_TRUE(flag_01->SetFromString("11", flags::SET_FLAGS_VALUE, |
|
flags::kProgrammaticChange, &err)); |
|
EXPECT_EQ(absl::GetFlag(FLAGS_int_flag), 11); |
|
EXPECT_FALSE(flag_01->IsSpecifiedOnCommandLine()); |
|
|
|
EXPECT_TRUE(flag_01->SetFromString("-123", flags::SET_FLAGS_VALUE, |
|
flags::kProgrammaticChange, &err)); |
|
EXPECT_EQ(absl::GetFlag(FLAGS_int_flag), -123); |
|
EXPECT_FALSE(flag_01->IsSpecifiedOnCommandLine()); |
|
|
|
EXPECT_TRUE(!flag_01->SetFromString("xyz", flags::SET_FLAGS_VALUE, |
|
flags::kProgrammaticChange, &err)); |
|
EXPECT_EQ(absl::GetFlag(FLAGS_int_flag), -123); |
|
EXPECT_EQ(err, "Illegal value 'xyz' specified for flag 'int_flag'"); |
|
EXPECT_FALSE(flag_01->IsSpecifiedOnCommandLine()); |
|
|
|
EXPECT_TRUE(!flag_01->SetFromString("A1", flags::SET_FLAGS_VALUE, |
|
flags::kProgrammaticChange, &err)); |
|
EXPECT_EQ(absl::GetFlag(FLAGS_int_flag), -123); |
|
EXPECT_EQ(err, "Illegal value 'A1' specified for flag 'int_flag'"); |
|
EXPECT_FALSE(flag_01->IsSpecifiedOnCommandLine()); |
|
|
|
EXPECT_TRUE(flag_01->SetFromString("0x10", flags::SET_FLAGS_VALUE, |
|
flags::kProgrammaticChange, &err)); |
|
EXPECT_EQ(absl::GetFlag(FLAGS_int_flag), 16); |
|
EXPECT_FALSE(flag_01->IsSpecifiedOnCommandLine()); |
|
|
|
EXPECT_TRUE(flag_01->SetFromString("011", flags::SET_FLAGS_VALUE, |
|
flags::kCommandLine, &err)); |
|
EXPECT_EQ(absl::GetFlag(FLAGS_int_flag), 11); |
|
EXPECT_TRUE(flag_01->IsSpecifiedOnCommandLine()); |
|
|
|
EXPECT_TRUE(!flag_01->SetFromString("", flags::SET_FLAGS_VALUE, |
|
flags::kProgrammaticChange, &err)); |
|
EXPECT_EQ(err, "Illegal value '' specified for flag 'int_flag'"); |
|
|
|
auto* flag_02 = flags::FindCommandLineFlag("string_flag"); |
|
EXPECT_TRUE(flag_02->SetFromString("xyz", flags::SET_FLAGS_VALUE, |
|
flags::kProgrammaticChange, &err)); |
|
EXPECT_EQ(absl::GetFlag(FLAGS_string_flag), "xyz"); |
|
|
|
EXPECT_TRUE(flag_02->SetFromString("", flags::SET_FLAGS_VALUE, |
|
flags::kProgrammaticChange, &err)); |
|
EXPECT_EQ(absl::GetFlag(FLAGS_string_flag), ""); |
|
} |
|
|
|
// -------------------------------------------------------------------- |
|
|
|
TEST_F(CommandLineFlagTest, TestSetFromStringDefaultValue) { |
|
std::string err; |
|
|
|
auto* flag_01 = flags::FindCommandLineFlag("int_flag"); |
|
|
|
EXPECT_TRUE(flag_01->SetFromString("111", flags::SET_FLAGS_DEFAULT, |
|
flags::kProgrammaticChange, &err)); |
|
EXPECT_EQ(flag_01->DefaultValue(), "111"); |
|
|
|
auto* flag_02 = flags::FindCommandLineFlag("string_flag"); |
|
|
|
EXPECT_TRUE(flag_02->SetFromString("abc", flags::SET_FLAGS_DEFAULT, |
|
flags::kProgrammaticChange, &err)); |
|
EXPECT_EQ(flag_02->DefaultValue(), "abc"); |
|
} |
|
|
|
// -------------------------------------------------------------------- |
|
|
|
TEST_F(CommandLineFlagTest, TestSetFromStringIfDefault) { |
|
std::string err; |
|
|
|
auto* flag_01 = flags::FindCommandLineFlag("int_flag"); |
|
|
|
EXPECT_TRUE(flag_01->SetFromString("22", flags::SET_FLAG_IF_DEFAULT, |
|
flags::kProgrammaticChange, &err)) |
|
<< err; |
|
EXPECT_EQ(absl::GetFlag(FLAGS_int_flag), 22); |
|
|
|
EXPECT_TRUE(flag_01->SetFromString("33", flags::SET_FLAG_IF_DEFAULT, |
|
flags::kProgrammaticChange, &err)); |
|
EXPECT_EQ(absl::GetFlag(FLAGS_int_flag), 22); |
|
// EXPECT_EQ(err, "ERROR: int_flag is already set to 22"); |
|
|
|
// Reset back to default value |
|
EXPECT_TRUE(flag_01->SetFromString("201", flags::SET_FLAGS_VALUE, |
|
flags::kProgrammaticChange, &err)); |
|
|
|
EXPECT_TRUE(flag_01->SetFromString("33", flags::SET_FLAG_IF_DEFAULT, |
|
flags::kProgrammaticChange, &err)); |
|
EXPECT_EQ(absl::GetFlag(FLAGS_int_flag), 201); |
|
// EXPECT_EQ(err, "ERROR: int_flag is already set to 201"); |
|
} |
|
|
|
} // namespace
|
|
|