positional_options_test.cpp 2.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091
  1. // Copyright Vladimir Prus 2004.
  2. // Distributed under the Boost Software License, Version 1.0.
  3. // (See accompanying file LICENSE_1_0.txt
  4. // or copy at http://www.boost.org/LICENSE_1_0.txt)
  5. #include <boost/program_options/positional_options.hpp>
  6. #include <boost/program_options/options_description.hpp>
  7. #include <boost/program_options/parsers.hpp>
  8. using namespace boost::program_options;
  9. // We'll use po::value everywhere to workaround vc6 bug.
  10. namespace po = boost::program_options;
  11. #include <boost/limits.hpp>
  12. #include "minitest.hpp"
  13. #include <vector>
  14. using namespace std;
  15. void test_positional_options()
  16. {
  17. positional_options_description p;
  18. p.add("first", 1);
  19. BOOST_CHECK_EQUAL(p.max_total_count(), 1u);
  20. BOOST_CHECK_EQUAL(p.name_for_position(0), "first");
  21. p.add("second", 2);
  22. BOOST_CHECK_EQUAL(p.max_total_count(), 3u);
  23. BOOST_CHECK_EQUAL(p.name_for_position(0), "first");
  24. BOOST_CHECK_EQUAL(p.name_for_position(1), "second");
  25. BOOST_CHECK_EQUAL(p.name_for_position(2), "second");
  26. p.add("third", -1);
  27. BOOST_CHECK_EQUAL(p.max_total_count(), (std::numeric_limits<unsigned>::max)());
  28. BOOST_CHECK_EQUAL(p.name_for_position(0), "first");
  29. BOOST_CHECK_EQUAL(p.name_for_position(1), "second");
  30. BOOST_CHECK_EQUAL(p.name_for_position(2), "second");
  31. BOOST_CHECK_EQUAL(p.name_for_position(3), "third");
  32. BOOST_CHECK_EQUAL(p.name_for_position(10000), "third");
  33. }
  34. void test_parsing()
  35. {
  36. options_description desc;
  37. desc.add_options()
  38. ("first", po::value<int>())
  39. ("second", po::value<int>())
  40. ("input-file", po::value< vector<string> >())
  41. ("some-other", po::value<string>())
  42. ;
  43. positional_options_description p;
  44. p.add("input-file", 2).add("some-other", 1);
  45. vector<string> args;
  46. args.push_back("--first=10");
  47. args.push_back("file1");
  48. args.push_back("--second=10");
  49. args.push_back("file2");
  50. args.push_back("file3");
  51. // Check that positional options are handled.
  52. parsed_options parsed =
  53. command_line_parser(args).options(desc).positional(p).run();
  54. BOOST_REQUIRE(parsed.options.size() == 5);
  55. BOOST_CHECK_EQUAL(parsed.options[1].string_key, "input-file");
  56. BOOST_CHECK_EQUAL(parsed.options[1].value[0], "file1");
  57. BOOST_CHECK_EQUAL(parsed.options[3].string_key, "input-file");
  58. BOOST_CHECK_EQUAL(parsed.options[3].value[0], "file2");
  59. BOOST_CHECK_EQUAL(parsed.options[4].value[0], "file3");
  60. args.push_back("file4");
  61. // Check that excessive number of positional options is detected.
  62. BOOST_CHECK_THROW(command_line_parser(args).options(desc).positional(p)
  63. .run(),
  64. too_many_positional_options_error);
  65. }
  66. int main(int, char* [])
  67. {
  68. test_positional_options();
  69. test_parsing();
  70. return 0;
  71. }