1 files changed, 22 insertions, 9 deletions
diff --git a/tests/unit_tests/main.cpp b/tests/unit_tests/main.cpp
index b2abad942..1706c43c9 100644
--- a/tests/unit_tests/main.cpp
+++ b/tests/unit_tests/main.cpp
@@ -31,31 +31,44 @@
 #include "gtest/gtest.h"
 
 #include <boost/filesystem.hpp>
+#include <boost/program_options.hpp>
 
 #include "include_base_utils.h"
+#include "common/command_line.h"
 #include "unit_tests_utils.h"
 
+namespace po = boost::program_options;
+
 boost::filesystem::path unit_test::data_dir;
 
 int main(int argc, char** argv)
 {
+  epee::string_tools::set_module_name_and_folder(argv[0]);
   mlog_configure(mlog_get_default_log_path("unit_tests.log"), true);
   epee::debug::get_set_enable_assert(true, false);
 
   ::testing::InitGoogleTest(&argc, argv);
 
-  // Process remaining arguments
-  if (argc == 2 && argv[1] != NULL) { // one arg: path to dir with test data
-    unit_test::data_dir = argv[1];
-  } else if (argc == 1) { // legacy: assume test binaries in 'build/release'
-    epee::string_tools::set_module_name_and_folder(argv[0]);
+  po::options_description desc_options("Command line options");
+  const command_line::arg_descriptor<std::string> arg_data_dir = {"data-dir", "Data files directory", "", true};
+  command_line::add_arg(desc_options, command_line::arg_data_dir, "");
+
+  po::variables_map vm;
+  bool r = command_line::handle_error_helper(desc_options, [&]()
+  {
+    po::store(po::parse_command_line(argc, argv, desc_options), vm);
+    po::notify(vm);
+    return true;
+  });
+  if (! r)
+    return 1;
+
+  if (vm["data-dir"].defaulted())
     unit_test::data_dir = boost::filesystem::path(epee::string_tools::get_current_module_folder())
                           .parent_path().parent_path().parent_path().parent_path()
                           .append("tests").append("data");
-  } else {
-    std::cerr << "Usage: " << argv[0] << " [<path-to-test-data-dir>]" << std::endl;
-    return 1;
-  }
+  else
+    unit_test::data_dir = command_line::get_arg(vm, arg_data_dir);
 
   return RUN_ALL_TESTS();
 }