Merge pull request #346 from abellgithub/store_ints

Store ints
p-ranav · Apr 19, 2024 · ac4c2c2 · ac4c2c2
2 parents 9550b0a + 7cb70ed
commit ac4c2c2
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 3 deletions.
diff --git a/include/argparse/argparse.hpp b/include/argparse/argparse.hpp
@@ -698,12 +698,13 @@ class Argument {
     return *this;
   }
 
-  auto &store_into(int &var) {
+  template <typename T, typename std::enable_if<std::is_integral<T>::value>::type * = nullptr>
+  auto &store_into(T &var) {
     if (m_default_value.has_value()) {
-      var = std::any_cast<int>(m_default_value);
+      var = std::any_cast<T>(m_default_value);
     }
     action([&var](const auto &s) {
-      var = details::parse_number<int, details::radix_10>()(s);
+      var = details::parse_number<T, details::radix_10>()(s);
     });
     return *this;
   }

diff --git a/test/test_store_into.cpp b/test/test_store_into.cpp
@@ -3,6 +3,7 @@ import argparse;
 #else
 #include <argparse/argparse.hpp>
 #endif
+#include <cstdint>
 #include <doctest.hpp>
 
 using doctest::test_suite;
@@ -27,6 +28,8 @@ TEST_CASE("Test store_into(bool), flag specified" *
   REQUIRE(flag == true);
 }
 
+// int cases
+
 TEST_CASE("Test store_into(int), no default value, non specified" *
           test_suite("store_into")) {
   argparse::ArgumentParser program("test");
@@ -57,6 +60,40 @@ TEST_CASE("Test store_into(int), default value, specified" *
   REQUIRE(res == 5);
 }
 
+// integral cases
+
+TEST_CASE("Test store_into(uint8_t), no default value, non specified" *
+          test_suite("store_into")) {
+  argparse::ArgumentParser program("test");
+  uint8_t res = 55;
+  program.add_argument("--int-opt").store_into(res);
+
+  program.parse_args({"./test.exe"});
+  REQUIRE(res == 55);
+}
+
+TEST_CASE("Test store_into(uint8_t), default value, non specified" *
+          test_suite("store_into")) {
+  argparse::ArgumentParser program("test");
+  uint8_t res = 55;
+  program.add_argument("--int-opt").default_value((uint8_t)3).store_into(res);
+
+  program.parse_args({"./test.exe"});
+  REQUIRE(res == 3);
+}
+
+TEST_CASE("Test store_into(uint8_t), default value, specified" *
+          test_suite("store_into")) {
+  argparse::ArgumentParser program("test");
+  uint8_t res = 55;
+  program.add_argument("--int-opt").default_value((uint8_t)3).store_into(res);
+
+  program.parse_args({"./test.exe", "--int-opt", "5"});
+  REQUIRE(res == 5);
+}
+
+// Double cases
+
 TEST_CASE("Test store_into(double), no default value, non specified" *
           test_suite("store_into")) {
   argparse::ArgumentParser program("test");