action_test.cpp 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154
  1. //
  2. // action_test.cpp
  3. // program_args-test
  4. //
  5. // Created by Sam Jaffe on 2/13/22.
  6. //
  7. #include "program_args/arguments.h"
  8. #include "xcode_gtest_helper.h"
  9. using testing::NotNull;
  10. namespace program {
  11. struct ArgumentTestHelper {
  12. template <typename Impl> static int main(Impl const & args) {
  13. return args.invoke();
  14. }
  15. };
  16. }
  17. template <typename T, size_t N> static T parse(char const * const (&argv)[N]) {
  18. return T(N, argv);
  19. }
  20. program::detail::Any g_action;
  21. template <typename Args, typename Action>
  22. int typed_main(Args const &, Action const & action) {
  23. g_action = program::detail::Any(&action);
  24. return 0;
  25. }
  26. template <typename Args, typename Action, typename SubAction>
  27. int typed_main(Args const &, Action const &, SubAction const & action) {
  28. g_action = program::detail::Any(&action);
  29. return 0;
  30. }
  31. struct Checkout : program::Arguments<Checkout> {
  32. using Arguments::Arguments;
  33. std::string commitish = argument(0, "commit-ish");
  34. };
  35. struct Commit : program::Arguments<Commit> {
  36. using Arguments::Arguments;
  37. std::string message = option("message", 'm');
  38. };
  39. class Push : public program::Arguments<Push> {
  40. public:
  41. using Arguments::Arguments;
  42. std::string remote = argument(0, "repository") = PROGRAM_DEFER(_remote());
  43. private:
  44. std::string _remote() const;
  45. };
  46. struct SetUrl : program::Arguments<SetUrl> {
  47. using Arguments::Arguments;
  48. bool push = flag("push") = false;
  49. std::string name = argument(0, "name");
  50. std::string new_url = argument(1, "newurl");
  51. std::string old_url = argument(2, "oldurl") = "";
  52. };
  53. struct Show : program::Arguments<Show> {
  54. using Arguments::Arguments;
  55. bool use_cached = flag('n') = false;
  56. };
  57. struct Remote : public program::Arguments<Remote> {
  58. PROGRAM_ARGS_INVOKE_WITH_DEFAULT(show, show, set_url);
  59. bool verbose = flag('v') = false;
  60. SetUrl set_url = action("set-url");
  61. Show show = action("show");
  62. };
  63. struct Bad1 : program::Arguments<Bad1> {
  64. using Arguments::Arguments;
  65. Commit commit = action("commit");
  66. std::string arg0 = argument(0, "arg0");
  67. };
  68. struct Bad2 : program::Arguments<Bad2> {
  69. using Arguments::Arguments;
  70. std::string arg0 = argument(0, "arg0");
  71. Commit commit = action("commit");
  72. };
  73. struct Git : program::Arguments<Git> {
  74. PROGRAM_ARGS_INVOKE(commit, checkout, push, remote);
  75. std::string pwd = option('C');
  76. bool verbose = flag("verbose", 'v');
  77. Commit commit = action("commit");
  78. Checkout checkout = action("checkout");
  79. Push push = action("push");
  80. Remote remote = action("remote");
  81. };
  82. std::string Push::_remote() const {
  83. if (Git const * git = Arguments::parent<Git>()) { return git->pwd; }
  84. return "";
  85. }
  86. TEST(ActionTest, CannotMixActionAndArgument) {
  87. EXPECT_THROW(Bad1(), program::ArgumentMixingError);
  88. EXPECT_THROW(Bad2(), program::ArgumentMixingError);
  89. }
  90. TEST(ActionTest, CanRunWithMultipleActions) { EXPECT_NO_THROW(Git()); }
  91. TEST(ActionTest, ActionIsRouted) {
  92. Git git = parse<Git>({"", "-v", "commit", "-m", "this is a message"});
  93. EXPECT_THAT(git.commit.message, "this is a message");
  94. }
  95. TEST(ActionTest, CanDiveIntoTypedMain) {
  96. Git git = parse<Git>({"", "-v", "commit", "-m", "this is a message"});
  97. program::ArgumentTestHelper::main(git);
  98. EXPECT_THAT(g_action.get<Commit>(), NotNull());
  99. }
  100. TEST(ActionTest, CanFetchParentInfo) {
  101. Git git = parse<Git>({"", "-C", "./submodules/X", "push"});
  102. EXPECT_THAT(git.push.remote, "./submodules/X");
  103. }
  104. TEST(ActionTest, ReturnsFailureOnNoAction) {
  105. Git git = parse<Git>({""});
  106. EXPECT_THAT(program::ArgumentTestHelper::main(git), 1);
  107. }
  108. TEST(ActionTest, CanRecursivelyPerformActions) {
  109. Git git = parse<Git>({"", "remote", "-v", "show", "-n"});
  110. EXPECT_TRUE(git.remote.verbose);
  111. EXPECT_TRUE(git.remote.show.use_cached);
  112. program::ArgumentTestHelper::main(git);
  113. EXPECT_THAT(g_action.get<Show>(), NotNull());
  114. }
  115. TEST(ActionTest, CanStoreDefaultAction) {
  116. Git git = parse<Git>({"", "remote"});
  117. program::ArgumentTestHelper::main(git);
  118. EXPECT_THAT(g_action.get<Show>(), NotNull());
  119. }