[Core] Add support for TESTBRIDGE_TEST_ONLY in UnityParseOptions

Allow Bazel --test_filter to select tests when
UNITY_USE_COMMAND_LINE_ARGS is enabled.

Remarks: Avoid to include <stdlib.h>. Instead used extern ... . Maybe stdlib.h is more compatible.

Author: tragisch

src/unity.c

@@ -2402,10 +2402,22 @@ int UnityStrictMatch          = 0;
 int UnityParseOptions(int argc, char** argv)
 {
     int i;
+    const char* testbridge_filter = NULL;
     UnityOptionIncludeNamed = NULL;
     UnityOptionExcludeNamed = NULL;
     UnityStrictMatch = 0;
 
+#ifndef UNITY_GETENV
+/*Allow Bazel test filtering via TESTBRIDGE_TEST_ONLY*/
+extern char* getenv(const char* name);
+#define UNITY_GETENV(name) getenv(name)
+#endif
+    testbridge_filter = UNITY_GETENV("TESTBRIDGE_TEST_ONLY");
+    if (testbridge_filter && testbridge_filter[0] != 0)
+    {
+        UnityOptionIncludeNamed = (char*)testbridge_filter;
+    }
+
     for (i = 1; i < argc; i++)
     {
         if (argv[i][0] == '-')

test/tests/test_unity_core.c

@@ -6,6 +6,10 @@
 ========================================================================= */
 
 #include "unity.h"
+#ifdef UNITY_USE_COMMAND_LINE_ARGS
+#include "unity_internals.h"
+#include <stdlib.h>
+#endif
 #define TEST_INSTANCES
 #include "self_assessment_utils.h"
 
@@ -194,6 +198,67 @@ void testFail(void)
     VERIFY_FAILS_END
 }
 
+#ifdef UNITY_USE_COMMAND_LINE_ARGS
+static void UnitySetTestbridgeFilter(const char* value)
+{
+#if defined(_WIN32) || defined(_MSC_VER)
+    if (value)
+    {
+        _putenv_s("TESTBRIDGE_TEST_ONLY", value);
+    }
+    else
+    {
+        _putenv_s("TESTBRIDGE_TEST_ONLY", "");
+    }
+#else
+    if (value)
+    {
+        setenv("TESTBRIDGE_TEST_ONLY", value, 1);
+    }
+    else
+    {
+        unsetenv("TESTBRIDGE_TEST_ONLY");
+    }
+#endif
+}
+
+static void UnitySetTestContext(const char* testfile, const char* testname)
+{
+    Unity.TestFile = testfile;
+    Unity.CurrentTestName = testname;
+}
+
+void testUnityParseOptionsUsesTestbridgeFilter(void)
+{
+    char* argv[] = { (char*)"prog", NULL };
+
+    UnitySetTestbridgeFilter("test_my_function");
+    UnityParseOptions(1, argv);
+
+    UnitySetTestContext("file.c", "test_my_function");
+    TEST_ASSERT_TRUE(UnityTestMatches());
+    UnitySetTestContext("file.c", "other");
+    TEST_ASSERT_FALSE(UnityTestMatches());
+
+    UnitySetTestbridgeFilter(NULL);
+}
+
+void testUnityParseOptionsArgsOverrideTestbridgeFilter(void)
+{
+    char* argv[] = { (char*)"prog", (char*)"-n", (char*)"other", NULL };
+
+    UnitySetTestbridgeFilter("test_my_function");
+    UnityParseOptions(3, argv);
+
+    UnitySetTestContext("file.c", "other");
+    TEST_ASSERT_TRUE(UnityTestMatches());
+    UnitySetTestContext("file.c", "test_my_function");
+    TEST_ASSERT_FALSE(UnityTestMatches());
+
+    UnitySetTestbridgeFilter(NULL);
+}
+#endif
+
 void testIsNull(void)
 {
     char* ptr1 = NULL;