Index: testing/gmock/test/gmock_test_utils.py |
diff --git a/testing/gmock/test/gmock_test_utils.py b/testing/gmock/test/gmock_test_utils.py |
index fa896a47a969b1ab3c073d1081f40f52d4343c78..ac3d67ae9a5647b33b062b1f435d56fc12efeb41 100755 |
--- a/testing/gmock/test/gmock_test_utils.py |
+++ b/testing/gmock/test/gmock_test_utils.py |
@@ -51,62 +51,10 @@ sys.path.append(GTEST_TESTS_UTIL_DIR) |
import gtest_test_utils # pylint: disable-msg=C6204 |
-# Initially maps a flag to its default value. After |
-# _ParseAndStripGMockFlags() is called, maps a flag to its actual |
-# value. |
-_flag_map = {'gmock_source_dir': os.path.dirname(sys.argv[0]), |
- 'gmock_build_dir': os.path.dirname(sys.argv[0])} |
-_gmock_flags_are_parsed = False |
- |
- |
-def _ParseAndStripGMockFlags(argv): |
- """Parses and strips Google Test flags from argv. This is idempotent.""" |
- |
- global _gmock_flags_are_parsed |
- if _gmock_flags_are_parsed: |
- return |
- |
- _gmock_flags_are_parsed = True |
- for flag in _flag_map: |
- # The environment variable overrides the default value. |
- if flag.upper() in os.environ: |
- _flag_map[flag] = os.environ[flag.upper()] |
- |
- # The command line flag overrides the environment variable. |
- i = 1 # Skips the program name. |
- while i < len(argv): |
- prefix = '--' + flag + '=' |
- if argv[i].startswith(prefix): |
- _flag_map[flag] = argv[i][len(prefix):] |
- del argv[i] |
- break |
- else: |
- # We don't increment i in case we just found a --gmock_* flag |
- # and removed it from argv. |
- i += 1 |
- |
- |
-def GetFlag(flag): |
- """Returns the value of the given flag.""" |
- |
- # In case GetFlag() is called before Main(), we always call |
- # _ParseAndStripGMockFlags() here to make sure the --gmock_* flags |
- # are parsed. |
- _ParseAndStripGMockFlags(sys.argv) |
- |
- return _flag_map[flag] |
- |
- |
def GetSourceDir(): |
"""Returns the absolute path of the directory where the .py files are.""" |
- return os.path.abspath(GetFlag('gmock_source_dir')) |
- |
- |
-def GetBuildDir(): |
- """Returns the absolute path of the directory where the test binaries are.""" |
- |
- return os.path.abspath(GetFlag('gmock_build_dir')) |
+ return gtest_test_utils.GetSourceDir() |
def GetTestExecutablePath(executable_name): |
@@ -122,7 +70,7 @@ def GetTestExecutablePath(executable_name): |
The absolute path of the test binary. |
""" |
- return gtest_test_utils.GetTestExecutablePath(executable_name, GetBuildDir()) |
+ return gtest_test_utils.GetTestExecutablePath(executable_name) |
def GetExitStatus(exit_code): |
@@ -160,8 +108,4 @@ TestCase = gtest_test_utils.TestCase |
def Main(): |
"""Runs the unit test.""" |
- # We must call _ParseAndStripGMockFlags() before calling |
- # gtest_test_utils.Main(). Otherwise unittest.main it calls will be |
- # confused by the --gmock_* flags. |
- _ParseAndStripGMockFlags(sys.argv) |
gtest_test_utils.Main() |