add argument passing via TEST_ARGS environment variable to all tests with an explicit main function

unittest/commands/test_simple_commands.cpp

@@ -26,6 +26,8 @@
 // whether to print verbose output (i.e. not capturing LAMMPS screen output).
 bool verbose = false;
 
+using LAMMPS_NS::utils::split_words;
+
 namespace LAMMPS_NS {
 using ::testing::Eq;
 
@@ -127,6 +129,17 @@ int main(int argc, char **argv)
 {
     MPI_Init(&argc, &argv);
     ::testing::InitGoogleMock(&argc, argv);
+
+    // handle arguments passed via environment variable
+    if (const char *var = getenv("TEST_ARGS")) {
+        std::vector<std::string> env = split_words(var);
+        for (auto arg : env) {
+            if (arg == "-v") {
+                verbose = true;
+            }
+        }
+    }
+
     if ((argc > 1) && (strcmp(argv[1], "-v") == 0)) verbose = true;
 
     int rv = RUN_ALL_TESTS();

unittest/force-styles/test_main.cpp

@@ -82,7 +82,8 @@ int main(int argc, char **argv)
     }
 
     // handle arguments passed via environment variable
-    std::vector<std::string> env = split_words(getenv("TEST_ARGS"));
+    if (const char *var = getenv("TEST_ARGS")) {
+        std::vector<std::string> env = split_words(var);
         for (auto arg : env) {
             if (arg == "-u") {
                 generate_yaml_file(argv[1], test_config);
@@ -93,6 +94,7 @@ int main(int argc, char **argv)
                 verbose = true;
             }
         }
+    }
 
     int iarg = 2;
     while (iarg < argc) {

unittest/formats/test_atom_styles.cpp

@@ -38,6 +38,8 @@
 
 #define GETIDX(i) lmp->atom->map(i)
 
+using LAMMPS_NS::utils::split_words;
+
 // whether to print verbose output (i.e. not capturing LAMMPS screen output).
 bool verbose = false;
 
@@ -1167,6 +1169,16 @@ int main(int argc, char **argv)
 {
     MPI_Init(&argc, &argv);
     ::testing::InitGoogleMock(&argc, argv);
+
+    // handle arguments passed via environment variable
+    if (const char *var = getenv("TEST_ARGS")) {
+        std::vector<std::string> env = split_words(var);
+        for (auto arg : env) {
+            if (arg == "-v") {
+                verbose = true;
+            }
+        }
+    }
     if ((argc > 1) && (strcmp(argv[1], "-v") == 0)) verbose = true;
 
     int rv = RUN_ALL_TESTS();

unittest/formats/test_eim_potential_file_reader.cpp

@@ -23,6 +23,7 @@
 #include <mpi.h>
 
 using namespace LAMMPS_NS;
+using utils::split_words;
 
 // whether to print verbose output (i.e. not capturing LAMMPS screen output).
 bool verbose = false;
@@ -208,6 +209,17 @@ int main(int argc, char **argv)
 {
     MPI_Init(&argc, &argv);
     ::testing::InitGoogleMock(&argc, argv);
+
+    // handle arguments passed via environment variable
+    if (const char *var = getenv("TEST_ARGS")) {
+        std::vector<std::string> env = split_words(var);
+        for (auto arg : env) {
+            if (arg == "-v") {
+                verbose = true;
+            }
+        }
+    }
+
     if ((argc > 1) && (strcmp(argv[1], "-v") == 0)) verbose = true;
 
     int rv = RUN_ALL_TESTS();

unittest/formats/test_potential_file_reader.cpp

@@ -34,6 +34,7 @@
 #include <mpi.h>
 
 using namespace LAMMPS_NS;
+using utils::split_words;
 
 // whether to print verbose output (i.e. not capturing LAMMPS screen output).
 bool verbose = false;
@@ -198,6 +199,16 @@ int main(int argc, char **argv)
 {
     MPI_Init(&argc, &argv);
     ::testing::InitGoogleMock(&argc, argv);
+
+    // handle arguments passed via environment variable
+    if (const char *var = getenv("TEST_ARGS")) {
+        std::vector<std::string> env = split_words(var);
+        for (auto arg : env) {
+            if (arg == "-v") {
+                verbose = true;
+            }
+        }
+    }
     if ((argc > 1) && (strcmp(argv[1], "-v") == 0)) verbose = true;
 
     int rv = RUN_ALL_TESTS();