forked from OSchip/llvm-project
1657 lines
57 KiB
C++
1657 lines
57 KiB
C++
//===- llvm/unittest/Support/CommandLineTest.cpp - CommandLine tests ------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/ADT/STLExtras.h"
|
|
#include "llvm/ADT/SmallString.h"
|
|
#include "llvm/ADT/Triple.h"
|
|
#include "llvm/Config/config.h"
|
|
#include "llvm/Support/FileSystem.h"
|
|
#include "llvm/Support/InitLLVM.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
#include "llvm/Support/Path.h"
|
|
#include "llvm/Support/Program.h"
|
|
#include "llvm/Support/StringSaver.h"
|
|
#include "gtest/gtest.h"
|
|
#include <fstream>
|
|
#include <stdlib.h>
|
|
#include <string>
|
|
|
|
using namespace llvm;
|
|
|
|
namespace {
|
|
|
|
class TempEnvVar {
|
|
public:
|
|
TempEnvVar(const char *name, const char *value)
|
|
: name(name) {
|
|
const char *old_value = getenv(name);
|
|
EXPECT_EQ(nullptr, old_value) << old_value;
|
|
#if HAVE_SETENV
|
|
setenv(name, value, true);
|
|
#else
|
|
# define SKIP_ENVIRONMENT_TESTS
|
|
#endif
|
|
}
|
|
|
|
~TempEnvVar() {
|
|
#if HAVE_SETENV
|
|
// Assume setenv and unsetenv come together.
|
|
unsetenv(name);
|
|
#else
|
|
(void)name; // Suppress -Wunused-private-field.
|
|
#endif
|
|
}
|
|
|
|
private:
|
|
const char *const name;
|
|
};
|
|
|
|
template <typename T, typename Base = cl::opt<T>>
|
|
class StackOption : public Base {
|
|
public:
|
|
template <class... Ts>
|
|
explicit StackOption(Ts &&... Ms) : Base(std::forward<Ts>(Ms)...) {}
|
|
|
|
~StackOption() override { this->removeArgument(); }
|
|
|
|
template <class DT> StackOption<T> &operator=(const DT &V) {
|
|
this->setValue(V);
|
|
return *this;
|
|
}
|
|
};
|
|
|
|
class StackSubCommand : public cl::SubCommand {
|
|
public:
|
|
StackSubCommand(StringRef Name,
|
|
StringRef Description = StringRef())
|
|
: SubCommand(Name, Description) {}
|
|
|
|
StackSubCommand() : SubCommand() {}
|
|
|
|
~StackSubCommand() { unregisterSubCommand(); }
|
|
};
|
|
|
|
|
|
cl::OptionCategory TestCategory("Test Options", "Description");
|
|
TEST(CommandLineTest, ModifyExisitingOption) {
|
|
StackOption<int> TestOption("test-option", cl::desc("old description"));
|
|
|
|
static const char Description[] = "New description";
|
|
static const char ArgString[] = "new-test-option";
|
|
static const char ValueString[] = "Integer";
|
|
|
|
StringMap<cl::Option *> &Map =
|
|
cl::getRegisteredOptions(*cl::TopLevelSubCommand);
|
|
|
|
ASSERT_TRUE(Map.count("test-option") == 1) <<
|
|
"Could not find option in map.";
|
|
|
|
cl::Option *Retrieved = Map["test-option"];
|
|
ASSERT_EQ(&TestOption, Retrieved) << "Retrieved wrong option.";
|
|
|
|
ASSERT_NE(Retrieved->Categories.end(),
|
|
find_if(Retrieved->Categories,
|
|
[&](const llvm::cl::OptionCategory *Cat) {
|
|
return Cat == &cl::GeneralCategory;
|
|
}))
|
|
<< "Incorrect default option category.";
|
|
|
|
Retrieved->addCategory(TestCategory);
|
|
ASSERT_NE(Retrieved->Categories.end(),
|
|
find_if(Retrieved->Categories,
|
|
[&](const llvm::cl::OptionCategory *Cat) {
|
|
return Cat == &TestCategory;
|
|
}))
|
|
<< "Failed to modify option's option category.";
|
|
|
|
Retrieved->setDescription(Description);
|
|
ASSERT_STREQ(Retrieved->HelpStr.data(), Description)
|
|
<< "Changing option description failed.";
|
|
|
|
Retrieved->setArgStr(ArgString);
|
|
ASSERT_STREQ(ArgString, Retrieved->ArgStr.data())
|
|
<< "Failed to modify option's Argument string.";
|
|
|
|
Retrieved->setValueStr(ValueString);
|
|
ASSERT_STREQ(Retrieved->ValueStr.data(), ValueString)
|
|
<< "Failed to modify option's Value string.";
|
|
|
|
Retrieved->setHiddenFlag(cl::Hidden);
|
|
ASSERT_EQ(cl::Hidden, TestOption.getOptionHiddenFlag()) <<
|
|
"Failed to modify option's hidden flag.";
|
|
}
|
|
#ifndef SKIP_ENVIRONMENT_TESTS
|
|
|
|
const char test_env_var[] = "LLVM_TEST_COMMAND_LINE_FLAGS";
|
|
|
|
cl::opt<std::string> EnvironmentTestOption("env-test-opt");
|
|
TEST(CommandLineTest, ParseEnvironment) {
|
|
TempEnvVar TEV(test_env_var, "-env-test-opt=hello");
|
|
EXPECT_EQ("", EnvironmentTestOption);
|
|
cl::ParseEnvironmentOptions("CommandLineTest", test_env_var);
|
|
EXPECT_EQ("hello", EnvironmentTestOption);
|
|
}
|
|
|
|
// This test used to make valgrind complain
|
|
// ("Conditional jump or move depends on uninitialised value(s)")
|
|
//
|
|
// Warning: Do not run any tests after this one that try to gain access to
|
|
// registered command line options because this will likely result in a
|
|
// SEGFAULT. This can occur because the cl::opt in the test below is declared
|
|
// on the stack which will be destroyed after the test completes but the
|
|
// command line system will still hold a pointer to a deallocated cl::Option.
|
|
TEST(CommandLineTest, ParseEnvironmentToLocalVar) {
|
|
// Put cl::opt on stack to check for proper initialization of fields.
|
|
StackOption<std::string> EnvironmentTestOptionLocal("env-test-opt-local");
|
|
TempEnvVar TEV(test_env_var, "-env-test-opt-local=hello-local");
|
|
EXPECT_EQ("", EnvironmentTestOptionLocal);
|
|
cl::ParseEnvironmentOptions("CommandLineTest", test_env_var);
|
|
EXPECT_EQ("hello-local", EnvironmentTestOptionLocal);
|
|
}
|
|
|
|
#endif // SKIP_ENVIRONMENT_TESTS
|
|
|
|
TEST(CommandLineTest, UseOptionCategory) {
|
|
StackOption<int> TestOption2("test-option", cl::cat(TestCategory));
|
|
|
|
ASSERT_NE(TestOption2.Categories.end(),
|
|
find_if(TestOption2.Categories,
|
|
[&](const llvm::cl::OptionCategory *Cat) {
|
|
return Cat == &TestCategory;
|
|
}))
|
|
<< "Failed to assign Option Category.";
|
|
}
|
|
|
|
TEST(CommandLineTest, UseMultipleCategories) {
|
|
StackOption<int> TestOption2("test-option2", cl::cat(TestCategory),
|
|
cl::cat(cl::GeneralCategory),
|
|
cl::cat(cl::GeneralCategory));
|
|
|
|
// Make sure cl::GeneralCategory wasn't added twice.
|
|
ASSERT_EQ(TestOption2.Categories.size(), 2U);
|
|
|
|
ASSERT_NE(TestOption2.Categories.end(),
|
|
find_if(TestOption2.Categories,
|
|
[&](const llvm::cl::OptionCategory *Cat) {
|
|
return Cat == &TestCategory;
|
|
}))
|
|
<< "Failed to assign Option Category.";
|
|
ASSERT_NE(TestOption2.Categories.end(),
|
|
find_if(TestOption2.Categories,
|
|
[&](const llvm::cl::OptionCategory *Cat) {
|
|
return Cat == &cl::GeneralCategory;
|
|
}))
|
|
<< "Failed to assign General Category.";
|
|
|
|
cl::OptionCategory AnotherCategory("Additional test Options", "Description");
|
|
StackOption<int> TestOption("test-option", cl::cat(TestCategory),
|
|
cl::cat(AnotherCategory));
|
|
ASSERT_EQ(TestOption.Categories.end(),
|
|
find_if(TestOption.Categories,
|
|
[&](const llvm::cl::OptionCategory *Cat) {
|
|
return Cat == &cl::GeneralCategory;
|
|
}))
|
|
<< "Failed to remove General Category.";
|
|
ASSERT_NE(TestOption.Categories.end(),
|
|
find_if(TestOption.Categories,
|
|
[&](const llvm::cl::OptionCategory *Cat) {
|
|
return Cat == &TestCategory;
|
|
}))
|
|
<< "Failed to assign Option Category.";
|
|
ASSERT_NE(TestOption.Categories.end(),
|
|
find_if(TestOption.Categories,
|
|
[&](const llvm::cl::OptionCategory *Cat) {
|
|
return Cat == &AnotherCategory;
|
|
}))
|
|
<< "Failed to assign Another Category.";
|
|
}
|
|
|
|
typedef void ParserFunction(StringRef Source, StringSaver &Saver,
|
|
SmallVectorImpl<const char *> &NewArgv,
|
|
bool MarkEOLs);
|
|
|
|
void testCommandLineTokenizer(ParserFunction *parse, StringRef Input,
|
|
const char *const Output[], size_t OutputSize) {
|
|
SmallVector<const char *, 0> Actual;
|
|
BumpPtrAllocator A;
|
|
StringSaver Saver(A);
|
|
parse(Input, Saver, Actual, /*MarkEOLs=*/false);
|
|
EXPECT_EQ(OutputSize, Actual.size());
|
|
for (unsigned I = 0, E = Actual.size(); I != E; ++I) {
|
|
if (I < OutputSize) {
|
|
EXPECT_STREQ(Output[I], Actual[I]);
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(CommandLineTest, TokenizeGNUCommandLine) {
|
|
const char Input[] =
|
|
"foo\\ bar \"foo bar\" \'foo bar\' 'foo\\\\bar' -DFOO=bar\\(\\) "
|
|
"foo\"bar\"baz C:\\\\src\\\\foo.cpp \"C:\\src\\foo.cpp\"";
|
|
const char *const Output[] = {
|
|
"foo bar", "foo bar", "foo bar", "foo\\bar",
|
|
"-DFOO=bar()", "foobarbaz", "C:\\src\\foo.cpp", "C:srcfoo.cpp"};
|
|
testCommandLineTokenizer(cl::TokenizeGNUCommandLine, Input, Output,
|
|
array_lengthof(Output));
|
|
}
|
|
|
|
TEST(CommandLineTest, TokenizeWindowsCommandLine1) {
|
|
const char Input[] = "a\\b c\\\\d e\\\\\"f g\" h\\\"i j\\\\\\\"k \"lmn\" o pqr "
|
|
"\"st \\\"u\" \\v";
|
|
const char *const Output[] = { "a\\b", "c\\\\d", "e\\f g", "h\"i", "j\\\"k",
|
|
"lmn", "o", "pqr", "st \"u", "\\v" };
|
|
testCommandLineTokenizer(cl::TokenizeWindowsCommandLine, Input, Output,
|
|
array_lengthof(Output));
|
|
}
|
|
|
|
TEST(CommandLineTest, TokenizeWindowsCommandLine2) {
|
|
const char Input[] = "clang -c -DFOO=\"\"\"ABC\"\"\" x.cpp";
|
|
const char *const Output[] = { "clang", "-c", "-DFOO=\"ABC\"", "x.cpp"};
|
|
testCommandLineTokenizer(cl::TokenizeWindowsCommandLine, Input, Output,
|
|
array_lengthof(Output));
|
|
}
|
|
|
|
TEST(CommandLineTest, TokenizeConfigFile1) {
|
|
const char *Input = "\\";
|
|
const char *const Output[] = { "\\" };
|
|
testCommandLineTokenizer(cl::tokenizeConfigFile, Input, Output,
|
|
array_lengthof(Output));
|
|
}
|
|
|
|
TEST(CommandLineTest, TokenizeConfigFile2) {
|
|
const char *Input = "\\abc";
|
|
const char *const Output[] = { "abc" };
|
|
testCommandLineTokenizer(cl::tokenizeConfigFile, Input, Output,
|
|
array_lengthof(Output));
|
|
}
|
|
|
|
TEST(CommandLineTest, TokenizeConfigFile3) {
|
|
const char *Input = "abc\\";
|
|
const char *const Output[] = { "abc\\" };
|
|
testCommandLineTokenizer(cl::tokenizeConfigFile, Input, Output,
|
|
array_lengthof(Output));
|
|
}
|
|
|
|
TEST(CommandLineTest, TokenizeConfigFile4) {
|
|
const char *Input = "abc\\\n123";
|
|
const char *const Output[] = { "abc123" };
|
|
testCommandLineTokenizer(cl::tokenizeConfigFile, Input, Output,
|
|
array_lengthof(Output));
|
|
}
|
|
|
|
TEST(CommandLineTest, TokenizeConfigFile5) {
|
|
const char *Input = "abc\\\r\n123";
|
|
const char *const Output[] = { "abc123" };
|
|
testCommandLineTokenizer(cl::tokenizeConfigFile, Input, Output,
|
|
array_lengthof(Output));
|
|
}
|
|
|
|
TEST(CommandLineTest, TokenizeConfigFile6) {
|
|
const char *Input = "abc\\\n";
|
|
const char *const Output[] = { "abc" };
|
|
testCommandLineTokenizer(cl::tokenizeConfigFile, Input, Output,
|
|
array_lengthof(Output));
|
|
}
|
|
|
|
TEST(CommandLineTest, TokenizeConfigFile7) {
|
|
const char *Input = "abc\\\r\n";
|
|
const char *const Output[] = { "abc" };
|
|
testCommandLineTokenizer(cl::tokenizeConfigFile, Input, Output,
|
|
array_lengthof(Output));
|
|
}
|
|
|
|
TEST(CommandLineTest, TokenizeConfigFile8) {
|
|
SmallVector<const char *, 0> Actual;
|
|
BumpPtrAllocator A;
|
|
StringSaver Saver(A);
|
|
cl::tokenizeConfigFile("\\\n", Saver, Actual, /*MarkEOLs=*/false);
|
|
EXPECT_TRUE(Actual.empty());
|
|
}
|
|
|
|
TEST(CommandLineTest, TokenizeConfigFile9) {
|
|
SmallVector<const char *, 0> Actual;
|
|
BumpPtrAllocator A;
|
|
StringSaver Saver(A);
|
|
cl::tokenizeConfigFile("\\\r\n", Saver, Actual, /*MarkEOLs=*/false);
|
|
EXPECT_TRUE(Actual.empty());
|
|
}
|
|
|
|
TEST(CommandLineTest, TokenizeConfigFile10) {
|
|
const char *Input = "\\\nabc";
|
|
const char *const Output[] = { "abc" };
|
|
testCommandLineTokenizer(cl::tokenizeConfigFile, Input, Output,
|
|
array_lengthof(Output));
|
|
}
|
|
|
|
TEST(CommandLineTest, TokenizeConfigFile11) {
|
|
const char *Input = "\\\r\nabc";
|
|
const char *const Output[] = { "abc" };
|
|
testCommandLineTokenizer(cl::tokenizeConfigFile, Input, Output,
|
|
array_lengthof(Output));
|
|
}
|
|
|
|
TEST(CommandLineTest, AliasesWithArguments) {
|
|
static const size_t ARGC = 3;
|
|
const char *const Inputs[][ARGC] = {
|
|
{ "-tool", "-actual=x", "-extra" },
|
|
{ "-tool", "-actual", "x" },
|
|
{ "-tool", "-alias=x", "-extra" },
|
|
{ "-tool", "-alias", "x" }
|
|
};
|
|
|
|
for (size_t i = 0, e = array_lengthof(Inputs); i < e; ++i) {
|
|
StackOption<std::string> Actual("actual");
|
|
StackOption<bool> Extra("extra");
|
|
StackOption<std::string> Input(cl::Positional);
|
|
|
|
cl::alias Alias("alias", llvm::cl::aliasopt(Actual));
|
|
|
|
cl::ParseCommandLineOptions(ARGC, Inputs[i]);
|
|
EXPECT_EQ("x", Actual);
|
|
EXPECT_EQ(0, Input.getNumOccurrences());
|
|
|
|
Alias.removeArgument();
|
|
}
|
|
}
|
|
|
|
void testAliasRequired(int argc, const char *const *argv) {
|
|
StackOption<std::string> Option("option", cl::Required);
|
|
cl::alias Alias("o", llvm::cl::aliasopt(Option));
|
|
|
|
cl::ParseCommandLineOptions(argc, argv);
|
|
EXPECT_EQ("x", Option);
|
|
EXPECT_EQ(1, Option.getNumOccurrences());
|
|
|
|
Alias.removeArgument();
|
|
}
|
|
|
|
TEST(CommandLineTest, AliasRequired) {
|
|
const char *opts1[] = { "-tool", "-option=x" };
|
|
const char *opts2[] = { "-tool", "-o", "x" };
|
|
testAliasRequired(array_lengthof(opts1), opts1);
|
|
testAliasRequired(array_lengthof(opts2), opts2);
|
|
}
|
|
|
|
TEST(CommandLineTest, HideUnrelatedOptions) {
|
|
StackOption<int> TestOption1("hide-option-1");
|
|
StackOption<int> TestOption2("hide-option-2", cl::cat(TestCategory));
|
|
|
|
cl::HideUnrelatedOptions(TestCategory);
|
|
|
|
ASSERT_EQ(cl::ReallyHidden, TestOption1.getOptionHiddenFlag())
|
|
<< "Failed to hide extra option.";
|
|
ASSERT_EQ(cl::NotHidden, TestOption2.getOptionHiddenFlag())
|
|
<< "Hid extra option that should be visable.";
|
|
|
|
StringMap<cl::Option *> &Map =
|
|
cl::getRegisteredOptions(*cl::TopLevelSubCommand);
|
|
ASSERT_EQ(cl::NotHidden, Map["help"]->getOptionHiddenFlag())
|
|
<< "Hid default option that should be visable.";
|
|
}
|
|
|
|
cl::OptionCategory TestCategory2("Test Options set 2", "Description");
|
|
|
|
TEST(CommandLineTest, HideUnrelatedOptionsMulti) {
|
|
StackOption<int> TestOption1("multi-hide-option-1");
|
|
StackOption<int> TestOption2("multi-hide-option-2", cl::cat(TestCategory));
|
|
StackOption<int> TestOption3("multi-hide-option-3", cl::cat(TestCategory2));
|
|
|
|
const cl::OptionCategory *VisibleCategories[] = {&TestCategory,
|
|
&TestCategory2};
|
|
|
|
cl::HideUnrelatedOptions(makeArrayRef(VisibleCategories));
|
|
|
|
ASSERT_EQ(cl::ReallyHidden, TestOption1.getOptionHiddenFlag())
|
|
<< "Failed to hide extra option.";
|
|
ASSERT_EQ(cl::NotHidden, TestOption2.getOptionHiddenFlag())
|
|
<< "Hid extra option that should be visable.";
|
|
ASSERT_EQ(cl::NotHidden, TestOption3.getOptionHiddenFlag())
|
|
<< "Hid extra option that should be visable.";
|
|
|
|
StringMap<cl::Option *> &Map =
|
|
cl::getRegisteredOptions(*cl::TopLevelSubCommand);
|
|
ASSERT_EQ(cl::NotHidden, Map["help"]->getOptionHiddenFlag())
|
|
<< "Hid default option that should be visable.";
|
|
}
|
|
|
|
TEST(CommandLineTest, SetValueInSubcategories) {
|
|
cl::ResetCommandLineParser();
|
|
|
|
StackSubCommand SC1("sc1", "First subcommand");
|
|
StackSubCommand SC2("sc2", "Second subcommand");
|
|
|
|
StackOption<bool> TopLevelOpt("top-level", cl::init(false));
|
|
StackOption<bool> SC1Opt("sc1", cl::sub(SC1), cl::init(false));
|
|
StackOption<bool> SC2Opt("sc2", cl::sub(SC2), cl::init(false));
|
|
|
|
EXPECT_FALSE(TopLevelOpt);
|
|
EXPECT_FALSE(SC1Opt);
|
|
EXPECT_FALSE(SC2Opt);
|
|
const char *args[] = {"prog", "-top-level"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(TopLevelOpt);
|
|
EXPECT_FALSE(SC1Opt);
|
|
EXPECT_FALSE(SC2Opt);
|
|
|
|
TopLevelOpt = false;
|
|
|
|
cl::ResetAllOptionOccurrences();
|
|
EXPECT_FALSE(TopLevelOpt);
|
|
EXPECT_FALSE(SC1Opt);
|
|
EXPECT_FALSE(SC2Opt);
|
|
const char *args2[] = {"prog", "sc1", "-sc1"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(3, args2, StringRef(), &llvm::nulls()));
|
|
EXPECT_FALSE(TopLevelOpt);
|
|
EXPECT_TRUE(SC1Opt);
|
|
EXPECT_FALSE(SC2Opt);
|
|
|
|
SC1Opt = false;
|
|
|
|
cl::ResetAllOptionOccurrences();
|
|
EXPECT_FALSE(TopLevelOpt);
|
|
EXPECT_FALSE(SC1Opt);
|
|
EXPECT_FALSE(SC2Opt);
|
|
const char *args3[] = {"prog", "sc2", "-sc2"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(3, args3, StringRef(), &llvm::nulls()));
|
|
EXPECT_FALSE(TopLevelOpt);
|
|
EXPECT_FALSE(SC1Opt);
|
|
EXPECT_TRUE(SC2Opt);
|
|
}
|
|
|
|
TEST(CommandLineTest, LookupFailsInWrongSubCommand) {
|
|
cl::ResetCommandLineParser();
|
|
|
|
StackSubCommand SC1("sc1", "First subcommand");
|
|
StackSubCommand SC2("sc2", "Second subcommand");
|
|
|
|
StackOption<bool> SC1Opt("sc1", cl::sub(SC1), cl::init(false));
|
|
StackOption<bool> SC2Opt("sc2", cl::sub(SC2), cl::init(false));
|
|
|
|
std::string Errs;
|
|
raw_string_ostream OS(Errs);
|
|
|
|
const char *args[] = {"prog", "sc1", "-sc2"};
|
|
EXPECT_FALSE(cl::ParseCommandLineOptions(3, args, StringRef(), &OS));
|
|
OS.flush();
|
|
EXPECT_FALSE(Errs.empty());
|
|
}
|
|
|
|
TEST(CommandLineTest, AddToAllSubCommands) {
|
|
cl::ResetCommandLineParser();
|
|
|
|
StackSubCommand SC1("sc1", "First subcommand");
|
|
StackOption<bool> AllOpt("everywhere", cl::sub(*cl::AllSubCommands),
|
|
cl::init(false));
|
|
StackSubCommand SC2("sc2", "Second subcommand");
|
|
|
|
const char *args[] = {"prog", "-everywhere"};
|
|
const char *args2[] = {"prog", "sc1", "-everywhere"};
|
|
const char *args3[] = {"prog", "sc2", "-everywhere"};
|
|
|
|
std::string Errs;
|
|
raw_string_ostream OS(Errs);
|
|
|
|
EXPECT_FALSE(AllOpt);
|
|
EXPECT_TRUE(cl::ParseCommandLineOptions(2, args, StringRef(), &OS));
|
|
EXPECT_TRUE(AllOpt);
|
|
|
|
AllOpt = false;
|
|
|
|
cl::ResetAllOptionOccurrences();
|
|
EXPECT_FALSE(AllOpt);
|
|
EXPECT_TRUE(cl::ParseCommandLineOptions(3, args2, StringRef(), &OS));
|
|
EXPECT_TRUE(AllOpt);
|
|
|
|
AllOpt = false;
|
|
|
|
cl::ResetAllOptionOccurrences();
|
|
EXPECT_FALSE(AllOpt);
|
|
EXPECT_TRUE(cl::ParseCommandLineOptions(3, args3, StringRef(), &OS));
|
|
EXPECT_TRUE(AllOpt);
|
|
|
|
// Since all parsing succeeded, the error message should be empty.
|
|
OS.flush();
|
|
EXPECT_TRUE(Errs.empty());
|
|
}
|
|
|
|
TEST(CommandLineTest, ReparseCommandLineOptions) {
|
|
cl::ResetCommandLineParser();
|
|
|
|
StackOption<bool> TopLevelOpt("top-level", cl::sub(*cl::TopLevelSubCommand),
|
|
cl::init(false));
|
|
|
|
const char *args[] = {"prog", "-top-level"};
|
|
|
|
EXPECT_FALSE(TopLevelOpt);
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(TopLevelOpt);
|
|
|
|
TopLevelOpt = false;
|
|
|
|
cl::ResetAllOptionOccurrences();
|
|
EXPECT_FALSE(TopLevelOpt);
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(TopLevelOpt);
|
|
}
|
|
|
|
TEST(CommandLineTest, RemoveFromRegularSubCommand) {
|
|
cl::ResetCommandLineParser();
|
|
|
|
StackSubCommand SC("sc", "Subcommand");
|
|
StackOption<bool> RemoveOption("remove-option", cl::sub(SC), cl::init(false));
|
|
StackOption<bool> KeepOption("keep-option", cl::sub(SC), cl::init(false));
|
|
|
|
const char *args[] = {"prog", "sc", "-remove-option"};
|
|
|
|
std::string Errs;
|
|
raw_string_ostream OS(Errs);
|
|
|
|
EXPECT_FALSE(RemoveOption);
|
|
EXPECT_TRUE(cl::ParseCommandLineOptions(3, args, StringRef(), &OS));
|
|
EXPECT_TRUE(RemoveOption);
|
|
OS.flush();
|
|
EXPECT_TRUE(Errs.empty());
|
|
|
|
RemoveOption.removeArgument();
|
|
|
|
cl::ResetAllOptionOccurrences();
|
|
EXPECT_FALSE(cl::ParseCommandLineOptions(3, args, StringRef(), &OS));
|
|
OS.flush();
|
|
EXPECT_FALSE(Errs.empty());
|
|
}
|
|
|
|
TEST(CommandLineTest, RemoveFromTopLevelSubCommand) {
|
|
cl::ResetCommandLineParser();
|
|
|
|
StackOption<bool> TopLevelRemove(
|
|
"top-level-remove", cl::sub(*cl::TopLevelSubCommand), cl::init(false));
|
|
StackOption<bool> TopLevelKeep(
|
|
"top-level-keep", cl::sub(*cl::TopLevelSubCommand), cl::init(false));
|
|
|
|
const char *args[] = {"prog", "-top-level-remove"};
|
|
|
|
EXPECT_FALSE(TopLevelRemove);
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(TopLevelRemove);
|
|
|
|
TopLevelRemove.removeArgument();
|
|
|
|
cl::ResetAllOptionOccurrences();
|
|
EXPECT_FALSE(
|
|
cl::ParseCommandLineOptions(2, args, StringRef(), &llvm::nulls()));
|
|
}
|
|
|
|
TEST(CommandLineTest, RemoveFromAllSubCommands) {
|
|
cl::ResetCommandLineParser();
|
|
|
|
StackSubCommand SC1("sc1", "First Subcommand");
|
|
StackSubCommand SC2("sc2", "Second Subcommand");
|
|
StackOption<bool> RemoveOption("remove-option", cl::sub(*cl::AllSubCommands),
|
|
cl::init(false));
|
|
StackOption<bool> KeepOption("keep-option", cl::sub(*cl::AllSubCommands),
|
|
cl::init(false));
|
|
|
|
const char *args0[] = {"prog", "-remove-option"};
|
|
const char *args1[] = {"prog", "sc1", "-remove-option"};
|
|
const char *args2[] = {"prog", "sc2", "-remove-option"};
|
|
|
|
// It should work for all subcommands including the top-level.
|
|
EXPECT_FALSE(RemoveOption);
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args0, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(RemoveOption);
|
|
|
|
RemoveOption = false;
|
|
|
|
cl::ResetAllOptionOccurrences();
|
|
EXPECT_FALSE(RemoveOption);
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(3, args1, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(RemoveOption);
|
|
|
|
RemoveOption = false;
|
|
|
|
cl::ResetAllOptionOccurrences();
|
|
EXPECT_FALSE(RemoveOption);
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(3, args2, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(RemoveOption);
|
|
|
|
RemoveOption.removeArgument();
|
|
|
|
// It should not work for any subcommands including the top-level.
|
|
cl::ResetAllOptionOccurrences();
|
|
EXPECT_FALSE(
|
|
cl::ParseCommandLineOptions(2, args0, StringRef(), &llvm::nulls()));
|
|
cl::ResetAllOptionOccurrences();
|
|
EXPECT_FALSE(
|
|
cl::ParseCommandLineOptions(3, args1, StringRef(), &llvm::nulls()));
|
|
cl::ResetAllOptionOccurrences();
|
|
EXPECT_FALSE(
|
|
cl::ParseCommandLineOptions(3, args2, StringRef(), &llvm::nulls()));
|
|
}
|
|
|
|
TEST(CommandLineTest, GetRegisteredSubcommands) {
|
|
cl::ResetCommandLineParser();
|
|
|
|
StackSubCommand SC1("sc1", "First Subcommand");
|
|
StackOption<bool> Opt1("opt1", cl::sub(SC1), cl::init(false));
|
|
StackSubCommand SC2("sc2", "Second subcommand");
|
|
StackOption<bool> Opt2("opt2", cl::sub(SC2), cl::init(false));
|
|
|
|
const char *args0[] = {"prog", "sc1"};
|
|
const char *args1[] = {"prog", "sc2"};
|
|
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args0, StringRef(), &llvm::nulls()));
|
|
EXPECT_FALSE(Opt1);
|
|
EXPECT_FALSE(Opt2);
|
|
for (auto *S : cl::getRegisteredSubcommands()) {
|
|
if (*S) {
|
|
EXPECT_EQ("sc1", S->getName());
|
|
}
|
|
}
|
|
|
|
cl::ResetAllOptionOccurrences();
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args1, StringRef(), &llvm::nulls()));
|
|
EXPECT_FALSE(Opt1);
|
|
EXPECT_FALSE(Opt2);
|
|
for (auto *S : cl::getRegisteredSubcommands()) {
|
|
if (*S) {
|
|
EXPECT_EQ("sc2", S->getName());
|
|
}
|
|
}
|
|
}
|
|
|
|
TEST(CommandLineTest, DefaultOptions) {
|
|
cl::ResetCommandLineParser();
|
|
|
|
StackOption<std::string> Bar("bar", cl::sub(*cl::AllSubCommands),
|
|
cl::DefaultOption);
|
|
StackOption<std::string, cl::alias> Bar_Alias(
|
|
"b", cl::desc("Alias for -bar"), cl::aliasopt(Bar), cl::DefaultOption);
|
|
|
|
StackOption<bool> Foo("foo", cl::init(false), cl::sub(*cl::AllSubCommands),
|
|
cl::DefaultOption);
|
|
StackOption<bool, cl::alias> Foo_Alias("f", cl::desc("Alias for -foo"),
|
|
cl::aliasopt(Foo), cl::DefaultOption);
|
|
|
|
StackSubCommand SC1("sc1", "First Subcommand");
|
|
// Override "-b" and change type in sc1 SubCommand.
|
|
StackOption<bool> SC1_B("b", cl::sub(SC1), cl::init(false));
|
|
StackSubCommand SC2("sc2", "Second subcommand");
|
|
// Override "-foo" and change type in sc2 SubCommand. Note that this does not
|
|
// affect "-f" alias, which continues to work correctly.
|
|
StackOption<std::string> SC2_Foo("foo", cl::sub(SC2));
|
|
|
|
const char *args0[] = {"prog", "-b", "args0 bar string", "-f"};
|
|
EXPECT_TRUE(cl::ParseCommandLineOptions(sizeof(args0) / sizeof(char *), args0,
|
|
StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(Bar == "args0 bar string");
|
|
EXPECT_TRUE(Foo);
|
|
EXPECT_FALSE(SC1_B);
|
|
EXPECT_TRUE(SC2_Foo.empty());
|
|
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
const char *args1[] = {"prog", "sc1", "-b", "-bar", "args1 bar string", "-f"};
|
|
EXPECT_TRUE(cl::ParseCommandLineOptions(sizeof(args1) / sizeof(char *), args1,
|
|
StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(Bar == "args1 bar string");
|
|
EXPECT_TRUE(Foo);
|
|
EXPECT_TRUE(SC1_B);
|
|
EXPECT_TRUE(SC2_Foo.empty());
|
|
for (auto *S : cl::getRegisteredSubcommands()) {
|
|
if (*S) {
|
|
EXPECT_EQ("sc1", S->getName());
|
|
}
|
|
}
|
|
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
const char *args2[] = {"prog", "sc2", "-b", "args2 bar string",
|
|
"-f", "-foo", "foo string"};
|
|
EXPECT_TRUE(cl::ParseCommandLineOptions(sizeof(args2) / sizeof(char *), args2,
|
|
StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(Bar == "args2 bar string");
|
|
EXPECT_TRUE(Foo);
|
|
EXPECT_FALSE(SC1_B);
|
|
EXPECT_TRUE(SC2_Foo == "foo string");
|
|
for (auto *S : cl::getRegisteredSubcommands()) {
|
|
if (*S) {
|
|
EXPECT_EQ("sc2", S->getName());
|
|
}
|
|
}
|
|
cl::ResetCommandLineParser();
|
|
}
|
|
|
|
TEST(CommandLineTest, ArgumentLimit) {
|
|
std::string args(32 * 4096, 'a');
|
|
EXPECT_FALSE(llvm::sys::commandLineFitsWithinSystemLimits("cl", args.data()));
|
|
}
|
|
|
|
TEST(CommandLineTest, ResponseFileWindows) {
|
|
if (!Triple(sys::getProcessTriple()).isOSWindows())
|
|
return;
|
|
|
|
StackOption<std::string, cl::list<std::string>> InputFilenames(
|
|
cl::Positional, cl::desc("<input files>"), cl::ZeroOrMore);
|
|
StackOption<bool> TopLevelOpt("top-level", cl::init(false));
|
|
|
|
// Create response file.
|
|
int FileDescriptor;
|
|
SmallString<64> TempPath;
|
|
std::error_code EC =
|
|
llvm::sys::fs::createTemporaryFile("resp-", ".txt", FileDescriptor, TempPath);
|
|
EXPECT_TRUE(!EC);
|
|
|
|
std::ofstream RspFile(TempPath.c_str());
|
|
EXPECT_TRUE(RspFile.is_open());
|
|
RspFile << "-top-level\npath\\dir\\file1\npath/dir/file2";
|
|
RspFile.close();
|
|
|
|
llvm::SmallString<128> RspOpt;
|
|
RspOpt.append(1, '@');
|
|
RspOpt.append(TempPath.c_str());
|
|
const char *args[] = {"prog", RspOpt.c_str()};
|
|
EXPECT_FALSE(TopLevelOpt);
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(TopLevelOpt);
|
|
EXPECT_TRUE(InputFilenames[0] == "path\\dir\\file1");
|
|
EXPECT_TRUE(InputFilenames[1] == "path/dir/file2");
|
|
|
|
llvm::sys::fs::remove(TempPath.c_str());
|
|
}
|
|
|
|
TEST(CommandLineTest, ResponseFiles) {
|
|
llvm::SmallString<128> TestDir;
|
|
std::error_code EC =
|
|
llvm::sys::fs::createUniqueDirectory("unittest", TestDir);
|
|
EXPECT_TRUE(!EC);
|
|
|
|
// Create included response file of first level.
|
|
llvm::SmallString<128> IncludedFileName;
|
|
llvm::sys::path::append(IncludedFileName, TestDir, "resp1");
|
|
std::ofstream IncludedFile(IncludedFileName.c_str());
|
|
EXPECT_TRUE(IncludedFile.is_open());
|
|
IncludedFile << "-option_1 -option_2\n"
|
|
"@incdir/resp2\n"
|
|
"-option_3=abcd\n"
|
|
"@incdir/resp3\n"
|
|
"-option_4=efjk\n";
|
|
IncludedFile.close();
|
|
|
|
// Directory for included file.
|
|
llvm::SmallString<128> IncDir;
|
|
llvm::sys::path::append(IncDir, TestDir, "incdir");
|
|
EC = llvm::sys::fs::create_directory(IncDir);
|
|
EXPECT_TRUE(!EC);
|
|
|
|
// Create included response file of second level.
|
|
llvm::SmallString<128> IncludedFileName2;
|
|
llvm::sys::path::append(IncludedFileName2, IncDir, "resp2");
|
|
std::ofstream IncludedFile2(IncludedFileName2.c_str());
|
|
EXPECT_TRUE(IncludedFile2.is_open());
|
|
IncludedFile2 << "-option_21 -option_22\n";
|
|
IncludedFile2 << "-option_23=abcd\n";
|
|
IncludedFile2.close();
|
|
|
|
// Create second included response file of second level.
|
|
llvm::SmallString<128> IncludedFileName3;
|
|
llvm::sys::path::append(IncludedFileName3, IncDir, "resp3");
|
|
std::ofstream IncludedFile3(IncludedFileName3.c_str());
|
|
EXPECT_TRUE(IncludedFile3.is_open());
|
|
IncludedFile3 << "-option_31 -option_32\n";
|
|
IncludedFile3 << "-option_33=abcd\n";
|
|
IncludedFile3.close();
|
|
|
|
// Prepare 'file' with reference to response file.
|
|
SmallString<128> IncRef;
|
|
IncRef.append(1, '@');
|
|
IncRef.append(IncludedFileName.c_str());
|
|
llvm::SmallVector<const char *, 4> Argv =
|
|
{ "test/test", "-flag_1", IncRef.c_str(), "-flag_2" };
|
|
|
|
// Expand response files.
|
|
llvm::BumpPtrAllocator A;
|
|
llvm::StringSaver Saver(A);
|
|
bool Res = llvm::cl::ExpandResponseFiles(
|
|
Saver, llvm::cl::TokenizeGNUCommandLine, Argv, false, true);
|
|
EXPECT_TRUE(Res);
|
|
EXPECT_EQ(Argv.size(), 13U);
|
|
EXPECT_STREQ(Argv[0], "test/test");
|
|
EXPECT_STREQ(Argv[1], "-flag_1");
|
|
EXPECT_STREQ(Argv[2], "-option_1");
|
|
EXPECT_STREQ(Argv[3], "-option_2");
|
|
EXPECT_STREQ(Argv[4], "-option_21");
|
|
EXPECT_STREQ(Argv[5], "-option_22");
|
|
EXPECT_STREQ(Argv[6], "-option_23=abcd");
|
|
EXPECT_STREQ(Argv[7], "-option_3=abcd");
|
|
EXPECT_STREQ(Argv[8], "-option_31");
|
|
EXPECT_STREQ(Argv[9], "-option_32");
|
|
EXPECT_STREQ(Argv[10], "-option_33=abcd");
|
|
EXPECT_STREQ(Argv[11], "-option_4=efjk");
|
|
EXPECT_STREQ(Argv[12], "-flag_2");
|
|
|
|
llvm::sys::fs::remove(IncludedFileName3);
|
|
llvm::sys::fs::remove(IncludedFileName2);
|
|
llvm::sys::fs::remove(IncDir);
|
|
llvm::sys::fs::remove(IncludedFileName);
|
|
llvm::sys::fs::remove(TestDir);
|
|
}
|
|
|
|
TEST(CommandLineTest, RecursiveResponseFiles) {
|
|
SmallString<128> TestDir;
|
|
std::error_code EC = sys::fs::createUniqueDirectory("unittest", TestDir);
|
|
EXPECT_TRUE(!EC);
|
|
|
|
SmallString<128> SelfFilePath;
|
|
sys::path::append(SelfFilePath, TestDir, "self.rsp");
|
|
std::string SelfFileRef = std::string("@") + SelfFilePath.c_str();
|
|
|
|
SmallString<128> NestedFilePath;
|
|
sys::path::append(NestedFilePath, TestDir, "nested.rsp");
|
|
std::string NestedFileRef = std::string("@") + NestedFilePath.c_str();
|
|
|
|
SmallString<128> FlagFilePath;
|
|
sys::path::append(FlagFilePath, TestDir, "flag.rsp");
|
|
std::string FlagFileRef = std::string("@") + FlagFilePath.c_str();
|
|
|
|
std::ofstream SelfFile(SelfFilePath.str());
|
|
EXPECT_TRUE(SelfFile.is_open());
|
|
SelfFile << "-option_1\n";
|
|
SelfFile << FlagFileRef << "\n";
|
|
SelfFile << NestedFileRef << "\n";
|
|
SelfFile << SelfFileRef << "\n";
|
|
SelfFile.close();
|
|
|
|
std::ofstream NestedFile(NestedFilePath.str());
|
|
EXPECT_TRUE(NestedFile.is_open());
|
|
NestedFile << "-option_2\n";
|
|
NestedFile << FlagFileRef << "\n";
|
|
NestedFile << SelfFileRef << "\n";
|
|
NestedFile << NestedFileRef << "\n";
|
|
NestedFile.close();
|
|
|
|
std::ofstream FlagFile(FlagFilePath.str());
|
|
EXPECT_TRUE(FlagFile.is_open());
|
|
FlagFile << "-option_x\n";
|
|
FlagFile.close();
|
|
|
|
// Ensure:
|
|
// Recursive expansion terminates
|
|
// Recursive files never expand
|
|
// Non-recursive repeats are allowed
|
|
SmallVector<const char *, 4> Argv = {"test/test", SelfFileRef.c_str(),
|
|
"-option_3"};
|
|
BumpPtrAllocator A;
|
|
StringSaver Saver(A);
|
|
#ifdef _WIN32
|
|
cl::TokenizerCallback Tokenizer = cl::TokenizeWindowsCommandLine;
|
|
#else
|
|
cl::TokenizerCallback Tokenizer = cl::TokenizeGNUCommandLine;
|
|
#endif
|
|
bool Res = cl::ExpandResponseFiles(Saver, Tokenizer, Argv, false, false);
|
|
EXPECT_FALSE(Res);
|
|
|
|
EXPECT_EQ(Argv.size(), 9U);
|
|
EXPECT_STREQ(Argv[0], "test/test");
|
|
EXPECT_STREQ(Argv[1], "-option_1");
|
|
EXPECT_STREQ(Argv[2], "-option_x");
|
|
EXPECT_STREQ(Argv[3], "-option_2");
|
|
EXPECT_STREQ(Argv[4], "-option_x");
|
|
EXPECT_STREQ(Argv[5], SelfFileRef.c_str());
|
|
EXPECT_STREQ(Argv[6], NestedFileRef.c_str());
|
|
EXPECT_STREQ(Argv[7], SelfFileRef.c_str());
|
|
EXPECT_STREQ(Argv[8], "-option_3");
|
|
}
|
|
|
|
TEST(CommandLineTest, ResponseFilesAtArguments) {
|
|
SmallString<128> TestDir;
|
|
std::error_code EC = sys::fs::createUniqueDirectory("unittest", TestDir);
|
|
EXPECT_TRUE(!EC);
|
|
|
|
SmallString<128> ResponseFilePath;
|
|
sys::path::append(ResponseFilePath, TestDir, "test.rsp");
|
|
|
|
std::ofstream ResponseFile(ResponseFilePath.c_str());
|
|
EXPECT_TRUE(ResponseFile.is_open());
|
|
ResponseFile << "-foo" << "\n";
|
|
ResponseFile << "-bar" << "\n";
|
|
ResponseFile.close();
|
|
|
|
// Ensure we expand rsp files after lots of non-rsp arguments starting with @.
|
|
constexpr size_t NON_RSP_AT_ARGS = 64;
|
|
SmallVector<const char *, 4> Argv = {"test/test"};
|
|
Argv.append(NON_RSP_AT_ARGS, "@non_rsp_at_arg");
|
|
std::string ResponseFileRef = std::string("@") + ResponseFilePath.c_str();
|
|
Argv.push_back(ResponseFileRef.c_str());
|
|
|
|
BumpPtrAllocator A;
|
|
StringSaver Saver(A);
|
|
bool Res = cl::ExpandResponseFiles(Saver, cl::TokenizeGNUCommandLine, Argv,
|
|
false, false);
|
|
EXPECT_FALSE(Res);
|
|
|
|
// ASSERT instead of EXPECT to prevent potential out-of-bounds access.
|
|
ASSERT_EQ(Argv.size(), 1 + NON_RSP_AT_ARGS + 2);
|
|
size_t i = 0;
|
|
EXPECT_STREQ(Argv[i++], "test/test");
|
|
for (; i < 1 + NON_RSP_AT_ARGS; ++i)
|
|
EXPECT_STREQ(Argv[i], "@non_rsp_at_arg");
|
|
EXPECT_STREQ(Argv[i++], "-foo");
|
|
EXPECT_STREQ(Argv[i++], "-bar");
|
|
}
|
|
|
|
TEST(CommandLineTest, SetDefautValue) {
|
|
cl::ResetCommandLineParser();
|
|
|
|
StackOption<std::string> Opt1("opt1", cl::init("true"));
|
|
StackOption<bool> Opt2("opt2", cl::init(true));
|
|
cl::alias Alias("alias", llvm::cl::aliasopt(Opt2));
|
|
StackOption<int> Opt3("opt3", cl::init(3));
|
|
|
|
const char *args[] = {"prog", "-opt1=false", "-opt2", "-opt3"};
|
|
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args, StringRef(), &llvm::nulls()));
|
|
|
|
EXPECT_TRUE(Opt1 == "false");
|
|
EXPECT_TRUE(Opt2);
|
|
EXPECT_TRUE(Opt3 == 3);
|
|
|
|
Opt2 = false;
|
|
Opt3 = 1;
|
|
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
for (auto &OM : cl::getRegisteredOptions(*cl::TopLevelSubCommand)) {
|
|
cl::Option *O = OM.second;
|
|
if (O->ArgStr == "opt2") {
|
|
continue;
|
|
}
|
|
O->setDefault();
|
|
}
|
|
|
|
EXPECT_TRUE(Opt1 == "true");
|
|
EXPECT_TRUE(Opt2);
|
|
EXPECT_TRUE(Opt3 == 3);
|
|
Alias.removeArgument();
|
|
}
|
|
|
|
TEST(CommandLineTest, ReadConfigFile) {
|
|
llvm::SmallVector<const char *, 1> Argv;
|
|
|
|
llvm::SmallString<128> TestDir;
|
|
std::error_code EC =
|
|
llvm::sys::fs::createUniqueDirectory("unittest", TestDir);
|
|
EXPECT_TRUE(!EC);
|
|
|
|
llvm::SmallString<128> TestCfg;
|
|
llvm::sys::path::append(TestCfg, TestDir, "foo");
|
|
std::ofstream ConfigFile(TestCfg.c_str());
|
|
EXPECT_TRUE(ConfigFile.is_open());
|
|
ConfigFile << "# Comment\n"
|
|
"-option_1\n"
|
|
"@subconfig\n"
|
|
"-option_3=abcd\n"
|
|
"-option_4=\\\n"
|
|
"cdef\n";
|
|
ConfigFile.close();
|
|
|
|
llvm::SmallString<128> TestCfg2;
|
|
llvm::sys::path::append(TestCfg2, TestDir, "subconfig");
|
|
std::ofstream ConfigFile2(TestCfg2.c_str());
|
|
EXPECT_TRUE(ConfigFile2.is_open());
|
|
ConfigFile2 << "-option_2\n"
|
|
"\n"
|
|
" # comment\n";
|
|
ConfigFile2.close();
|
|
|
|
// Make sure the current directory is not the directory where config files
|
|
// resides. In this case the code that expands response files will not find
|
|
// 'subconfig' unless it resolves nested inclusions relative to the including
|
|
// file.
|
|
llvm::SmallString<128> CurrDir;
|
|
EC = llvm::sys::fs::current_path(CurrDir);
|
|
EXPECT_TRUE(!EC);
|
|
EXPECT_TRUE(StringRef(CurrDir) != StringRef(TestDir));
|
|
|
|
llvm::BumpPtrAllocator A;
|
|
llvm::StringSaver Saver(A);
|
|
bool Result = llvm::cl::readConfigFile(TestCfg, Saver, Argv);
|
|
|
|
EXPECT_TRUE(Result);
|
|
EXPECT_EQ(Argv.size(), 4U);
|
|
EXPECT_STREQ(Argv[0], "-option_1");
|
|
EXPECT_STREQ(Argv[1], "-option_2");
|
|
EXPECT_STREQ(Argv[2], "-option_3=abcd");
|
|
EXPECT_STREQ(Argv[3], "-option_4=cdef");
|
|
|
|
llvm::sys::fs::remove(TestCfg2);
|
|
llvm::sys::fs::remove(TestCfg);
|
|
llvm::sys::fs::remove(TestDir);
|
|
}
|
|
|
|
TEST(CommandLineTest, PositionalEatArgsError) {
|
|
cl::ResetCommandLineParser();
|
|
|
|
StackOption<std::string, cl::list<std::string>> PosEatArgs(
|
|
"positional-eat-args", cl::Positional, cl::desc("<arguments>..."),
|
|
cl::ZeroOrMore, cl::PositionalEatsArgs);
|
|
StackOption<std::string, cl::list<std::string>> PosEatArgs2(
|
|
"positional-eat-args2", cl::Positional, cl::desc("Some strings"),
|
|
cl::ZeroOrMore, cl::PositionalEatsArgs);
|
|
|
|
const char *args[] = {"prog", "-positional-eat-args=XXXX"};
|
|
const char *args2[] = {"prog", "-positional-eat-args=XXXX", "-foo"};
|
|
const char *args3[] = {"prog", "-positional-eat-args", "-foo"};
|
|
const char *args4[] = {"prog", "-positional-eat-args",
|
|
"-foo", "-positional-eat-args2",
|
|
"-bar", "foo"};
|
|
|
|
std::string Errs;
|
|
raw_string_ostream OS(Errs);
|
|
EXPECT_FALSE(cl::ParseCommandLineOptions(2, args, StringRef(), &OS)); OS.flush();
|
|
EXPECT_FALSE(Errs.empty()); Errs.clear();
|
|
EXPECT_FALSE(cl::ParseCommandLineOptions(3, args2, StringRef(), &OS)); OS.flush();
|
|
EXPECT_FALSE(Errs.empty()); Errs.clear();
|
|
EXPECT_TRUE(cl::ParseCommandLineOptions(3, args3, StringRef(), &OS)); OS.flush();
|
|
EXPECT_TRUE(Errs.empty()); Errs.clear();
|
|
|
|
cl::ResetAllOptionOccurrences();
|
|
EXPECT_TRUE(cl::ParseCommandLineOptions(6, args4, StringRef(), &OS)); OS.flush();
|
|
EXPECT_TRUE(PosEatArgs.size() == 1);
|
|
EXPECT_TRUE(PosEatArgs2.size() == 2);
|
|
EXPECT_TRUE(Errs.empty());
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
TEST(CommandLineTest, GetCommandLineArguments) {
|
|
int argc = __argc;
|
|
char **argv = __argv;
|
|
|
|
// GetCommandLineArguments is called in InitLLVM.
|
|
llvm::InitLLVM X(argc, argv);
|
|
|
|
EXPECT_EQ(llvm::sys::path::is_absolute(argv[0]),
|
|
llvm::sys::path::is_absolute(__argv[0]));
|
|
|
|
EXPECT_TRUE(llvm::sys::path::filename(argv[0])
|
|
.equals_lower("supporttests.exe"))
|
|
<< "Filename of test executable is "
|
|
<< llvm::sys::path::filename(argv[0]);
|
|
}
|
|
#endif
|
|
|
|
class OutputRedirector {
|
|
public:
|
|
OutputRedirector(int RedirectFD)
|
|
: RedirectFD(RedirectFD), OldFD(dup(RedirectFD)) {
|
|
if (OldFD == -1 ||
|
|
sys::fs::createTemporaryFile("unittest-redirect", "", NewFD,
|
|
FilePath) ||
|
|
dup2(NewFD, RedirectFD) == -1)
|
|
Valid = false;
|
|
}
|
|
|
|
~OutputRedirector() {
|
|
dup2(OldFD, RedirectFD);
|
|
close(OldFD);
|
|
close(NewFD);
|
|
}
|
|
|
|
SmallVector<char, 128> FilePath;
|
|
bool Valid = true;
|
|
|
|
private:
|
|
int RedirectFD;
|
|
int OldFD;
|
|
int NewFD;
|
|
};
|
|
|
|
struct AutoDeleteFile {
|
|
SmallVector<char, 128> FilePath;
|
|
~AutoDeleteFile() {
|
|
if (!FilePath.empty())
|
|
sys::fs::remove(std::string(FilePath.data(), FilePath.size()));
|
|
}
|
|
};
|
|
|
|
class PrintOptionInfoTest : public ::testing::Test {
|
|
public:
|
|
// Return std::string because the output of a failing EXPECT check is
|
|
// unreadable for StringRef. It also avoids any lifetime issues.
|
|
template <typename... Ts> std::string runTest(Ts... OptionAttributes) {
|
|
outs().flush(); // flush any output from previous tests
|
|
AutoDeleteFile File;
|
|
{
|
|
OutputRedirector Stdout(fileno(stdout));
|
|
if (!Stdout.Valid)
|
|
return "";
|
|
File.FilePath = Stdout.FilePath;
|
|
|
|
StackOption<OptionValue> TestOption(Opt, cl::desc(HelpText),
|
|
OptionAttributes...);
|
|
printOptionInfo(TestOption, 26);
|
|
outs().flush();
|
|
}
|
|
auto Buffer = MemoryBuffer::getFile(File.FilePath);
|
|
if (!Buffer)
|
|
return "";
|
|
return Buffer->get()->getBuffer().str();
|
|
}
|
|
|
|
enum class OptionValue { Val };
|
|
const StringRef Opt = "some-option";
|
|
const StringRef HelpText = "some help";
|
|
|
|
private:
|
|
// This is a workaround for cl::Option sub-classes having their
|
|
// printOptionInfo functions private.
|
|
void printOptionInfo(const cl::Option &O, size_t Width) {
|
|
O.printOptionInfo(Width);
|
|
}
|
|
};
|
|
|
|
TEST_F(PrintOptionInfoTest, PrintOptionInfoValueOptionalWithoutSentinel) {
|
|
std::string Output =
|
|
runTest(cl::ValueOptional,
|
|
cl::values(clEnumValN(OptionValue::Val, "v1", "desc1")));
|
|
|
|
// clang-format off
|
|
EXPECT_EQ(Output, (" --" + Opt + "=<value> - " + HelpText + "\n"
|
|
" =v1 - desc1\n")
|
|
.str());
|
|
// clang-format on
|
|
}
|
|
|
|
TEST_F(PrintOptionInfoTest, PrintOptionInfoValueOptionalWithSentinel) {
|
|
std::string Output = runTest(
|
|
cl::ValueOptional, cl::values(clEnumValN(OptionValue::Val, "v1", "desc1"),
|
|
clEnumValN(OptionValue::Val, "", "")));
|
|
|
|
// clang-format off
|
|
EXPECT_EQ(Output,
|
|
(" --" + Opt + " - " + HelpText + "\n"
|
|
" --" + Opt + "=<value> - " + HelpText + "\n"
|
|
" =v1 - desc1\n")
|
|
.str());
|
|
// clang-format on
|
|
}
|
|
|
|
TEST_F(PrintOptionInfoTest, PrintOptionInfoValueOptionalWithSentinelWithHelp) {
|
|
std::string Output = runTest(
|
|
cl::ValueOptional, cl::values(clEnumValN(OptionValue::Val, "v1", "desc1"),
|
|
clEnumValN(OptionValue::Val, "", "desc2")));
|
|
|
|
// clang-format off
|
|
EXPECT_EQ(Output, (" --" + Opt + " - " + HelpText + "\n"
|
|
" --" + Opt + "=<value> - " + HelpText + "\n"
|
|
" =v1 - desc1\n"
|
|
" =<empty> - desc2\n")
|
|
.str());
|
|
// clang-format on
|
|
}
|
|
|
|
TEST_F(PrintOptionInfoTest, PrintOptionInfoValueRequiredWithEmptyValueName) {
|
|
std::string Output = runTest(
|
|
cl::ValueRequired, cl::values(clEnumValN(OptionValue::Val, "v1", "desc1"),
|
|
clEnumValN(OptionValue::Val, "", "")));
|
|
|
|
// clang-format off
|
|
EXPECT_EQ(Output, (" --" + Opt + "=<value> - " + HelpText + "\n"
|
|
" =v1 - desc1\n"
|
|
" =<empty>\n")
|
|
.str());
|
|
// clang-format on
|
|
}
|
|
|
|
TEST_F(PrintOptionInfoTest, PrintOptionInfoEmptyValueDescription) {
|
|
std::string Output = runTest(
|
|
cl::ValueRequired, cl::values(clEnumValN(OptionValue::Val, "v1", "")));
|
|
|
|
// clang-format off
|
|
EXPECT_EQ(Output,
|
|
(" --" + Opt + "=<value> - " + HelpText + "\n"
|
|
" =v1\n").str());
|
|
// clang-format on
|
|
}
|
|
|
|
class GetOptionWidthTest : public ::testing::Test {
|
|
public:
|
|
enum class OptionValue { Val };
|
|
|
|
template <typename... Ts>
|
|
size_t runTest(StringRef ArgName, Ts... OptionAttributes) {
|
|
StackOption<OptionValue> TestOption(ArgName, cl::desc("some help"),
|
|
OptionAttributes...);
|
|
return getOptionWidth(TestOption);
|
|
}
|
|
|
|
private:
|
|
// This is a workaround for cl::Option sub-classes having their
|
|
// printOptionInfo
|
|
// functions private.
|
|
size_t getOptionWidth(const cl::Option &O) { return O.getOptionWidth(); }
|
|
};
|
|
|
|
TEST_F(GetOptionWidthTest, GetOptionWidthArgNameLonger) {
|
|
StringRef ArgName("a-long-argument-name");
|
|
size_t ExpectedStrSize = (" --" + ArgName + "=<value> - ").str().size();
|
|
EXPECT_EQ(
|
|
runTest(ArgName, cl::values(clEnumValN(OptionValue::Val, "v", "help"))),
|
|
ExpectedStrSize);
|
|
}
|
|
|
|
TEST_F(GetOptionWidthTest, GetOptionWidthFirstOptionNameLonger) {
|
|
StringRef OptName("a-long-option-name");
|
|
size_t ExpectedStrSize = (" =" + OptName + " - ").str().size();
|
|
EXPECT_EQ(
|
|
runTest("a", cl::values(clEnumValN(OptionValue::Val, OptName, "help"),
|
|
clEnumValN(OptionValue::Val, "b", "help"))),
|
|
ExpectedStrSize);
|
|
}
|
|
|
|
TEST_F(GetOptionWidthTest, GetOptionWidthSecondOptionNameLonger) {
|
|
StringRef OptName("a-long-option-name");
|
|
size_t ExpectedStrSize = (" =" + OptName + " - ").str().size();
|
|
EXPECT_EQ(
|
|
runTest("a", cl::values(clEnumValN(OptionValue::Val, "b", "help"),
|
|
clEnumValN(OptionValue::Val, OptName, "help"))),
|
|
ExpectedStrSize);
|
|
}
|
|
|
|
TEST_F(GetOptionWidthTest, GetOptionWidthEmptyOptionNameLonger) {
|
|
size_t ExpectedStrSize = StringRef(" =<empty> - ").size();
|
|
// The length of a=<value> (including indentation) is actually the same as the
|
|
// =<empty> string, so it is impossible to distinguish via testing the case
|
|
// where the empty string is picked from where the option name is picked.
|
|
EXPECT_EQ(runTest("a", cl::values(clEnumValN(OptionValue::Val, "b", "help"),
|
|
clEnumValN(OptionValue::Val, "", "help"))),
|
|
ExpectedStrSize);
|
|
}
|
|
|
|
TEST_F(GetOptionWidthTest,
|
|
GetOptionWidthValueOptionalEmptyOptionWithNoDescription) {
|
|
StringRef ArgName("a");
|
|
// The length of a=<value> (including indentation) is actually the same as the
|
|
// =<empty> string, so it is impossible to distinguish via testing the case
|
|
// where the empty string is ignored from where it is not ignored.
|
|
// The dash will not actually be printed, but the space it would take up is
|
|
// included to ensure a consistent column width.
|
|
size_t ExpectedStrSize = (" -" + ArgName + "=<value> - ").str().size();
|
|
EXPECT_EQ(runTest(ArgName, cl::ValueOptional,
|
|
cl::values(clEnumValN(OptionValue::Val, "value", "help"),
|
|
clEnumValN(OptionValue::Val, "", ""))),
|
|
ExpectedStrSize);
|
|
}
|
|
|
|
TEST_F(GetOptionWidthTest,
|
|
GetOptionWidthValueRequiredEmptyOptionWithNoDescription) {
|
|
// The length of a=<value> (including indentation) is actually the same as the
|
|
// =<empty> string, so it is impossible to distinguish via testing the case
|
|
// where the empty string is picked from where the option name is picked
|
|
size_t ExpectedStrSize = StringRef(" =<empty> - ").size();
|
|
EXPECT_EQ(runTest("a", cl::ValueRequired,
|
|
cl::values(clEnumValN(OptionValue::Val, "value", "help"),
|
|
clEnumValN(OptionValue::Val, "", ""))),
|
|
ExpectedStrSize);
|
|
}
|
|
|
|
TEST(CommandLineTest, PrefixOptions) {
|
|
cl::ResetCommandLineParser();
|
|
|
|
StackOption<std::string, cl::list<std::string>> IncludeDirs(
|
|
"I", cl::Prefix, cl::desc("Declare an include directory"));
|
|
|
|
// Test non-prefixed variant works with cl::Prefix options.
|
|
EXPECT_TRUE(IncludeDirs.empty());
|
|
const char *args[] = {"prog", "-I=/usr/include"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(IncludeDirs.size() == 1);
|
|
EXPECT_TRUE(IncludeDirs.front().compare("/usr/include") == 0);
|
|
|
|
IncludeDirs.erase(IncludeDirs.begin());
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// Test non-prefixed variant works with cl::Prefix options when value is
|
|
// passed in following argument.
|
|
EXPECT_TRUE(IncludeDirs.empty());
|
|
const char *args2[] = {"prog", "-I", "/usr/include"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(3, args2, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(IncludeDirs.size() == 1);
|
|
EXPECT_TRUE(IncludeDirs.front().compare("/usr/include") == 0);
|
|
|
|
IncludeDirs.erase(IncludeDirs.begin());
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// Test prefixed variant works with cl::Prefix options.
|
|
EXPECT_TRUE(IncludeDirs.empty());
|
|
const char *args3[] = {"prog", "-I/usr/include"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args3, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(IncludeDirs.size() == 1);
|
|
EXPECT_TRUE(IncludeDirs.front().compare("/usr/include") == 0);
|
|
|
|
StackOption<std::string, cl::list<std::string>> MacroDefs(
|
|
"D", cl::AlwaysPrefix, cl::desc("Define a macro"),
|
|
cl::value_desc("MACRO[=VALUE]"));
|
|
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// Test non-prefixed variant does not work with cl::AlwaysPrefix options:
|
|
// equal sign is part of the value.
|
|
EXPECT_TRUE(MacroDefs.empty());
|
|
const char *args4[] = {"prog", "-D=HAVE_FOO"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args4, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(MacroDefs.size() == 1);
|
|
EXPECT_TRUE(MacroDefs.front().compare("=HAVE_FOO") == 0);
|
|
|
|
MacroDefs.erase(MacroDefs.begin());
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// Test non-prefixed variant does not allow value to be passed in following
|
|
// argument with cl::AlwaysPrefix options.
|
|
EXPECT_TRUE(MacroDefs.empty());
|
|
const char *args5[] = {"prog", "-D", "HAVE_FOO"};
|
|
EXPECT_FALSE(
|
|
cl::ParseCommandLineOptions(3, args5, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(MacroDefs.empty());
|
|
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// Test prefixed variant works with cl::AlwaysPrefix options.
|
|
EXPECT_TRUE(MacroDefs.empty());
|
|
const char *args6[] = {"prog", "-DHAVE_FOO"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args6, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(MacroDefs.size() == 1);
|
|
EXPECT_TRUE(MacroDefs.front().compare("HAVE_FOO") == 0);
|
|
}
|
|
|
|
TEST(CommandLineTest, GroupingWithValue) {
|
|
cl::ResetCommandLineParser();
|
|
|
|
StackOption<bool> OptF("f", cl::Grouping, cl::desc("Some flag"));
|
|
StackOption<bool> OptB("b", cl::Grouping, cl::desc("Another flag"));
|
|
StackOption<bool> OptD("d", cl::Grouping, cl::ValueDisallowed,
|
|
cl::desc("ValueDisallowed option"));
|
|
StackOption<std::string> OptV("v", cl::Grouping,
|
|
cl::desc("ValueRequired option"));
|
|
StackOption<std::string> OptO("o", cl::Grouping, cl::ValueOptional,
|
|
cl::desc("ValueOptional option"));
|
|
|
|
// Should be possible to use an option which requires a value
|
|
// at the end of a group.
|
|
const char *args1[] = {"prog", "-fv", "val1"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(3, args1, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(OptF);
|
|
EXPECT_STREQ("val1", OptV.c_str());
|
|
OptV.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// Should not crash if it is accidentally used elsewhere in the group.
|
|
const char *args2[] = {"prog", "-vf", "val2"};
|
|
EXPECT_FALSE(
|
|
cl::ParseCommandLineOptions(3, args2, StringRef(), &llvm::nulls()));
|
|
OptV.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// Should allow the "opt=value" form at the end of the group
|
|
const char *args3[] = {"prog", "-fv=val3"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args3, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(OptF);
|
|
EXPECT_STREQ("val3", OptV.c_str());
|
|
OptV.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// Should allow assigning a value for a ValueOptional option
|
|
// at the end of the group
|
|
const char *args4[] = {"prog", "-fo=val4"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args4, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(OptF);
|
|
EXPECT_STREQ("val4", OptO.c_str());
|
|
OptO.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// Should assign an empty value if a ValueOptional option is used elsewhere
|
|
// in the group.
|
|
const char *args5[] = {"prog", "-fob"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args5, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(OptF);
|
|
EXPECT_EQ(1, OptO.getNumOccurrences());
|
|
EXPECT_EQ(1, OptB.getNumOccurrences());
|
|
EXPECT_TRUE(OptO.empty());
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// Should not allow an assignment for a ValueDisallowed option.
|
|
const char *args6[] = {"prog", "-fd=false"};
|
|
EXPECT_FALSE(
|
|
cl::ParseCommandLineOptions(2, args6, StringRef(), &llvm::nulls()));
|
|
}
|
|
|
|
TEST(CommandLineTest, GroupingAndPrefix) {
|
|
cl::ResetCommandLineParser();
|
|
|
|
StackOption<bool> OptF("f", cl::Grouping, cl::desc("Some flag"));
|
|
StackOption<bool> OptB("b", cl::Grouping, cl::desc("Another flag"));
|
|
StackOption<std::string> OptP("p", cl::Prefix, cl::Grouping,
|
|
cl::desc("Prefix and Grouping"));
|
|
StackOption<std::string> OptA("a", cl::AlwaysPrefix, cl::Grouping,
|
|
cl::desc("AlwaysPrefix and Grouping"));
|
|
|
|
// Should be possible to use a cl::Prefix option without grouping.
|
|
const char *args1[] = {"prog", "-pval1"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args1, StringRef(), &llvm::nulls()));
|
|
EXPECT_STREQ("val1", OptP.c_str());
|
|
OptP.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// Should be possible to pass a value in a separate argument.
|
|
const char *args2[] = {"prog", "-p", "val2"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(3, args2, StringRef(), &llvm::nulls()));
|
|
EXPECT_STREQ("val2", OptP.c_str());
|
|
OptP.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// The "-opt=value" form should work, too.
|
|
const char *args3[] = {"prog", "-p=val3"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args3, StringRef(), &llvm::nulls()));
|
|
EXPECT_STREQ("val3", OptP.c_str());
|
|
OptP.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// All three previous cases should work the same way if an option with both
|
|
// cl::Prefix and cl::Grouping modifiers is used at the end of a group.
|
|
const char *args4[] = {"prog", "-fpval4"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args4, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(OptF);
|
|
EXPECT_STREQ("val4", OptP.c_str());
|
|
OptP.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
const char *args5[] = {"prog", "-fp", "val5"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(3, args5, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(OptF);
|
|
EXPECT_STREQ("val5", OptP.c_str());
|
|
OptP.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
const char *args6[] = {"prog", "-fp=val6"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args6, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(OptF);
|
|
EXPECT_STREQ("val6", OptP.c_str());
|
|
OptP.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// Should assign a value even if the part after a cl::Prefix option is equal
|
|
// to the name of another option.
|
|
const char *args7[] = {"prog", "-fpb"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args7, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(OptF);
|
|
EXPECT_STREQ("b", OptP.c_str());
|
|
EXPECT_FALSE(OptB);
|
|
OptP.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// Should be possible to use a cl::AlwaysPrefix option without grouping.
|
|
const char *args8[] = {"prog", "-aval8"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args8, StringRef(), &llvm::nulls()));
|
|
EXPECT_STREQ("val8", OptA.c_str());
|
|
OptA.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// Should not be possible to pass a value in a separate argument.
|
|
const char *args9[] = {"prog", "-a", "val9"};
|
|
EXPECT_FALSE(
|
|
cl::ParseCommandLineOptions(3, args9, StringRef(), &llvm::nulls()));
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// With the "-opt=value" form, the "=" symbol should be preserved.
|
|
const char *args10[] = {"prog", "-a=val10"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args10, StringRef(), &llvm::nulls()));
|
|
EXPECT_STREQ("=val10", OptA.c_str());
|
|
OptA.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// All three previous cases should work the same way if an option with both
|
|
// cl::AlwaysPrefix and cl::Grouping modifiers is used at the end of a group.
|
|
const char *args11[] = {"prog", "-faval11"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args11, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(OptF);
|
|
EXPECT_STREQ("val11", OptA.c_str());
|
|
OptA.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
const char *args12[] = {"prog", "-fa", "val12"};
|
|
EXPECT_FALSE(
|
|
cl::ParseCommandLineOptions(3, args12, StringRef(), &llvm::nulls()));
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
const char *args13[] = {"prog", "-fa=val13"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args13, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(OptF);
|
|
EXPECT_STREQ("=val13", OptA.c_str());
|
|
OptA.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// Should assign a value even if the part after a cl::AlwaysPrefix option
|
|
// is equal to the name of another option.
|
|
const char *args14[] = {"prog", "-fab"};
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(2, args14, StringRef(), &llvm::nulls()));
|
|
EXPECT_TRUE(OptF);
|
|
EXPECT_STREQ("b", OptA.c_str());
|
|
EXPECT_FALSE(OptB);
|
|
OptA.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
}
|
|
|
|
TEST(CommandLineTest, LongOptions) {
|
|
cl::ResetCommandLineParser();
|
|
|
|
StackOption<bool> OptA("a", cl::desc("Some flag"));
|
|
StackOption<bool> OptBLong("long-flag", cl::desc("Some long flag"));
|
|
StackOption<bool, cl::alias> OptB("b", cl::desc("Alias to --long-flag"),
|
|
cl::aliasopt(OptBLong));
|
|
StackOption<std::string> OptAB("ab", cl::desc("Another long option"));
|
|
|
|
std::string Errs;
|
|
raw_string_ostream OS(Errs);
|
|
|
|
const char *args1[] = {"prog", "-a", "-ab", "val1"};
|
|
const char *args2[] = {"prog", "-a", "--ab", "val1"};
|
|
const char *args3[] = {"prog", "-ab", "--ab", "val1"};
|
|
|
|
//
|
|
// The following tests treat `-` and `--` the same, and always match the
|
|
// longest string.
|
|
//
|
|
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(4, args1, StringRef(), &OS)); OS.flush();
|
|
EXPECT_TRUE(OptA);
|
|
EXPECT_FALSE(OptBLong);
|
|
EXPECT_STREQ("val1", OptAB.c_str());
|
|
EXPECT_TRUE(Errs.empty()); Errs.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
EXPECT_TRUE(
|
|
cl::ParseCommandLineOptions(4, args2, StringRef(), &OS)); OS.flush();
|
|
EXPECT_TRUE(OptA);
|
|
EXPECT_FALSE(OptBLong);
|
|
EXPECT_STREQ("val1", OptAB.c_str());
|
|
EXPECT_TRUE(Errs.empty()); Errs.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// Fails because `-ab` and `--ab` are treated the same and appear more than
|
|
// once. Also, `val1` is unexpected.
|
|
EXPECT_FALSE(
|
|
cl::ParseCommandLineOptions(4, args3, StringRef(), &OS)); OS.flush();
|
|
outs()<< Errs << "\n";
|
|
EXPECT_FALSE(Errs.empty()); Errs.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
//
|
|
// The following tests treat `-` and `--` differently, with `-` for short, and
|
|
// `--` for long options.
|
|
//
|
|
|
|
// Fails because `-ab` is treated as `-a -b`, so `-a` is seen twice, and
|
|
// `val1` is unexpected.
|
|
EXPECT_FALSE(cl::ParseCommandLineOptions(4, args1, StringRef(),
|
|
&OS, nullptr, true)); OS.flush();
|
|
EXPECT_FALSE(Errs.empty()); Errs.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// Works because `-a` is treated differently than `--ab`.
|
|
EXPECT_TRUE(cl::ParseCommandLineOptions(4, args2, StringRef(),
|
|
&OS, nullptr, true)); OS.flush();
|
|
EXPECT_TRUE(Errs.empty()); Errs.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
|
|
// Works because `-ab` is treated as `-a -b`, and `--ab` is a long option.
|
|
EXPECT_TRUE(cl::ParseCommandLineOptions(4, args3, StringRef(),
|
|
&OS, nullptr, true));
|
|
EXPECT_TRUE(OptA);
|
|
EXPECT_TRUE(OptBLong);
|
|
EXPECT_STREQ("val1", OptAB.c_str());
|
|
OS.flush();
|
|
EXPECT_TRUE(Errs.empty()); Errs.clear();
|
|
cl::ResetAllOptionOccurrences();
|
|
}
|
|
} // anonymous namespace
|