From c333e577fe476fb4f8f4ab3c0babdcc07f62a2be Mon Sep 17 00:00:00 2001 From: Chris Bieneman Date: Mon, 26 Jan 2015 22:50:47 +0000 Subject: Pete Cooper suggested the new API should use ArrayRef instead of SmallVectorImpl. Also adding a test case. llvm-svn: 227148 --- llvm/unittests/Support/CommandLineTest.cpp | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) (limited to 'llvm/unittests/Support/CommandLineTest.cpp') diff --git a/llvm/unittests/Support/CommandLineTest.cpp b/llvm/unittests/Support/CommandLineTest.cpp index 4fa14e2..7505593 100644 --- a/llvm/unittests/Support/CommandLineTest.cpp +++ b/llvm/unittests/Support/CommandLineTest.cpp @@ -247,4 +247,28 @@ TEST(CommandLineTest, HideUnrelatedOptions) { << "Hid default option that should be visable."; } +cl::OptionCategory TestCategory2("Test Options set 2", "Description"); + +TEST(CommandLineTest, HideUnrelatedOptionsMulti) { + cl::opt TestOption1("test-option-1"); + cl::opt TestOption2("test-option-2", cl::cat(TestCategory)); + cl::opt TestOption3("test-option-3", cl::cat(TestCategory2)); + + 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 Map; + cl::getRegisteredOptions(Map); + ASSERT_EQ(cl::NotHidden, Map["help"]->getOptionHiddenFlag()) + << "Hid default option that should be visable."; +} + } // anonymous namespace -- cgit v1.1