97c1664b4b
-- 167cd2647144583746311129b0cc98a89a9897e1 by Andy Soffer <asoffer@google.com>: Internal Changes PiperOrigin-RevId: 268000987 -- ab44fea7dfdaf763c69609a08ddfac22480ce520 by Derek Mauro <dmauro@google.com>: Fix flags Cmake tests on Windows PiperOrigin-RevId: 267859442 -- d7ec9f7611370b01bc9a5aed7e18460df2a15429 by Abseil Team <absl-team@google.com>: Fix undefined behavior in symbolize_elf.inc PiperOrigin-RevId: 267684904 -- e7c5f8e472a91ed99bdf4876942f2bda5751aa6c by Abseil Team <absl-team@google.com>: Turn off ABSL_HAS_ALARM on Fuchsia. PiperOrigin-RevId: 267666090 -- b7d25d5c71a49e697a8bf6440ed30f2e3f036420 by Chris Kennelly <ckennelly@google.com>: Mark once initialization function as no inline. We expect this code to run infrequently. PiperOrigin-RevId: 267649713 -- dc2a5e5f1e39a03fff837d34a319033fde55d9ba by Derek Mauro <dmauro@google.com>: Fix the MSVC CMake random build PiperOrigin-RevId: 267624074 -- ba2751b67fa17d1b6c53e5ba79f81a5371e4a03a by Abseil Team <absl-team@google.com>: Move "internal/flag.*" files to the "internal" build target. PiperOrigin-RevId: 267588996 -- 05b985a33eec4f8acff1809ad9218a1e22220f34 by Abseil Team <absl-team@google.com>: Move "internal/flag.*" files to the "internal" build target. PiperOrigin-RevId: 267580412 GitOrigin-RevId: 167cd2647144583746311129b0cc98a89a9897e1 Change-Id: Ibd334f46a5671c7c1d3fcf5354029e2fbb7ba91f
218 lines
8 KiB
C++
218 lines
8 KiB
C++
//
|
|
// 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 <algorithm>
|
|
#include <string>
|
|
|
|
#include "gtest/gtest.h"
|
|
#include "absl/flags/flag.h"
|
|
#include "absl/flags/internal/registry.h"
|
|
#include "absl/flags/usage_config.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:
|
|
static void SetUpTestSuite() {
|
|
// Install a function to normalize filenames before this test is run.
|
|
absl::FlagsUsageConfig default_config;
|
|
default_config.normalize_filename = &CommandLineFlagTest::NormalizeFileName;
|
|
absl::SetFlagsUsageConfig(default_config);
|
|
}
|
|
|
|
void SetUp() override { flag_saver_ = absl::make_unique<flags::FlagSaver>(); }
|
|
void TearDown() override { flag_saver_.reset(); }
|
|
|
|
private:
|
|
static std::string NormalizeFileName(absl::string_view fname) {
|
|
#ifdef _WIN32
|
|
std::string normalized(fname);
|
|
std::replace(normalized.begin(), normalized.end(), '\\', '/');
|
|
fname = normalized;
|
|
#endif
|
|
return std::string(fname);
|
|
}
|
|
|
|
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"))
|
|
<< flag_01->Filename();
|
|
|
|
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"))
|
|
<< flag_02->Filename();
|
|
|
|
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
|