1 //===- unittest/Support/OptionParsingTest.cpp - OptTable tests ------------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #include "llvm/ADT/OwningPtr.h"
11 #include "llvm/ADT/STLExtras.h"
12 #include "llvm/Option/Arg.h"
13 #include "llvm/Option/ArgList.h"
14 #include "llvm/Option/Option.h"
15 #include "gtest/gtest.h"
18 using namespace llvm::opt;
20 #define SUPPORT_ALIASARGS // FIXME: Remove when no longer necessary.
23 OPT_INVALID = 0, // This is not an option ID.
24 #define OPTION(PREFIX, NAME, ID, KIND, GROUP, ALIAS, ALIASARGS, FLAGS, PARAM, \
25 HELPTEXT, METAVAR) OPT_##ID,
31 #define PREFIX(NAME, VALUE) const char *const NAME[] = VALUE;
41 static const OptTable::Info InfoTable[] = {
42 #define OPTION(PREFIX, NAME, ID, KIND, GROUP, ALIAS, ALIASARGS, FLAGS, PARAM, \
44 { PREFIX, NAME, HELPTEXT, METAVAR, OPT_##ID, Option::KIND##Class, PARAM, \
45 FLAGS, OPT_##GROUP, OPT_##ALIAS, ALIASARGS },
51 class TestOptTable : public OptTable {
54 : OptTable(InfoTable, array_lengthof(InfoTable)) {}
58 const char *Args[] = {
64 "-E", "apple", "bloom",
70 TEST(Option, OptionParsing) {
73 OwningPtr<InputArgList> AL(T.ParseArgs(Args, array_endof(Args), MAI, MAC));
75 // Check they all exist.
76 EXPECT_TRUE(AL->hasArg(OPT_A));
77 EXPECT_TRUE(AL->hasArg(OPT_B));
78 EXPECT_TRUE(AL->hasArg(OPT_C));
79 EXPECT_TRUE(AL->hasArg(OPT_D));
80 EXPECT_TRUE(AL->hasArg(OPT_E));
81 EXPECT_TRUE(AL->hasArg(OPT_F));
82 EXPECT_TRUE(AL->hasArg(OPT_G));
85 EXPECT_EQ(AL->getLastArgValue(OPT_B), "hi");
86 EXPECT_EQ(AL->getLastArgValue(OPT_C), "bye");
87 EXPECT_EQ(AL->getLastArgValue(OPT_D), "adena");
88 std::vector<std::string> Es = AL->getAllArgValues(OPT_E);
89 EXPECT_EQ(Es[0], "apple");
90 EXPECT_EQ(Es[1], "bloom");
91 EXPECT_EQ(AL->getLastArgValue(OPT_F), "42");
92 std::vector<std::string> Gs = AL->getAllArgValues(OPT_G);
93 EXPECT_EQ(Gs[0], "chuu");
94 EXPECT_EQ(Gs[1], "2");
96 // Check the help text.
98 raw_string_ostream RSO(Help);
99 T.PrintHelp(RSO, "test", "title!");
100 EXPECT_NE(Help.find("-A"), std::string::npos);
103 arg_iterator Cs = AL->filtered_begin(OPT_C);
104 ASSERT_NE(Cs, AL->filtered_end());
105 EXPECT_EQ(StringRef((*Cs)->getValue()), "desu");
107 (*Cs)->render(*AL, ASL);
108 ASSERT_EQ(ASL.size(), 2u);
109 EXPECT_EQ(StringRef(ASL[0]), "-C");
110 EXPECT_EQ(StringRef(ASL[1]), "desu");
113 TEST(Option, ParseWithFlagExclusions) {
116 OwningPtr<InputArgList> AL;
118 // Exclude flag3 to avoid parsing as OPT_SLASH_C.
119 AL.reset(T.ParseArgs(Args, array_endof(Args), MAI, MAC,
120 /*FlagsToInclude=*/0,
121 /*FlagsToExclude=*/OptFlag3));
122 EXPECT_TRUE(AL->hasArg(OPT_A));
123 EXPECT_TRUE(AL->hasArg(OPT_C));
124 EXPECT_FALSE(AL->hasArg(OPT_SLASH_C));
126 // Exclude flag1 to avoid parsing as OPT_C.
127 AL.reset(T.ParseArgs(Args, array_endof(Args), MAI, MAC,
128 /*FlagsToInclude=*/0,
129 /*FlagsToExclude=*/OptFlag1));
130 EXPECT_TRUE(AL->hasArg(OPT_B));
131 EXPECT_FALSE(AL->hasArg(OPT_C));
132 EXPECT_TRUE(AL->hasArg(OPT_SLASH_C));
134 const char *NewArgs[] = { "/C", "foo", "--C=bar" };
135 AL.reset(T.ParseArgs(NewArgs, array_endof(NewArgs), MAI, MAC));
136 EXPECT_TRUE(AL->hasArg(OPT_SLASH_C));
137 EXPECT_TRUE(AL->hasArg(OPT_C));
138 EXPECT_EQ(AL->getLastArgValue(OPT_SLASH_C), "foo");
139 EXPECT_EQ(AL->getLastArgValue(OPT_C), "bar");
142 TEST(Option, ParseAliasInGroup) {
146 const char *MyArgs[] = { "-I" };
147 OwningPtr<InputArgList> AL(T.ParseArgs(MyArgs, array_endof(MyArgs), MAI, MAC));
148 EXPECT_TRUE(AL->hasArg(OPT_H));
151 TEST(Option, AliasArgs) {
155 const char *MyArgs[] = { "-J", "-Joo" };
156 OwningPtr<InputArgList> AL(T.ParseArgs(MyArgs, array_endof(MyArgs), MAI, MAC));
157 EXPECT_TRUE(AL->hasArg(OPT_B));
158 EXPECT_EQ(AL->getAllArgValues(OPT_B)[0], "foo");
159 EXPECT_EQ(AL->getAllArgValues(OPT_B)[1], "bar");