Wed, 31 Dec 2014 07:53:36 +0100
Correct small whitespace inconsistency, lost while renaming variables.
michael@0 | 1 | #!/usr/bin/env python |
michael@0 | 2 | # |
michael@0 | 3 | # Copyright 2008, Google Inc. |
michael@0 | 4 | # All rights reserved. |
michael@0 | 5 | # |
michael@0 | 6 | # Redistribution and use in source and binary forms, with or without |
michael@0 | 7 | # modification, are permitted provided that the following conditions are |
michael@0 | 8 | # met: |
michael@0 | 9 | # |
michael@0 | 10 | # * Redistributions of source code must retain the above copyright |
michael@0 | 11 | # notice, this list of conditions and the following disclaimer. |
michael@0 | 12 | # * Redistributions in binary form must reproduce the above |
michael@0 | 13 | # copyright notice, this list of conditions and the following disclaimer |
michael@0 | 14 | # in the documentation and/or other materials provided with the |
michael@0 | 15 | # distribution. |
michael@0 | 16 | # * Neither the name of Google Inc. nor the names of its |
michael@0 | 17 | # contributors may be used to endorse or promote products derived from |
michael@0 | 18 | # this software without specific prior written permission. |
michael@0 | 19 | # |
michael@0 | 20 | # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
michael@0 | 21 | # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
michael@0 | 22 | # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
michael@0 | 23 | # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
michael@0 | 24 | # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
michael@0 | 25 | # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
michael@0 | 26 | # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
michael@0 | 27 | # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
michael@0 | 28 | # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
michael@0 | 29 | # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
michael@0 | 30 | # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
michael@0 | 31 | |
michael@0 | 32 | """Verifies that Google Test correctly parses environment variables.""" |
michael@0 | 33 | |
michael@0 | 34 | __author__ = 'wan@google.com (Zhanyong Wan)' |
michael@0 | 35 | |
michael@0 | 36 | import os |
michael@0 | 37 | import gtest_test_utils |
michael@0 | 38 | |
michael@0 | 39 | |
michael@0 | 40 | IS_WINDOWS = os.name == 'nt' |
michael@0 | 41 | IS_LINUX = os.name == 'posix' and os.uname()[0] == 'Linux' |
michael@0 | 42 | |
michael@0 | 43 | COMMAND = gtest_test_utils.GetTestExecutablePath('gtest_env_var_test_') |
michael@0 | 44 | |
michael@0 | 45 | environ = os.environ.copy() |
michael@0 | 46 | |
michael@0 | 47 | |
michael@0 | 48 | def AssertEq(expected, actual): |
michael@0 | 49 | if expected != actual: |
michael@0 | 50 | print 'Expected: %s' % (expected,) |
michael@0 | 51 | print ' Actual: %s' % (actual,) |
michael@0 | 52 | raise AssertionError |
michael@0 | 53 | |
michael@0 | 54 | |
michael@0 | 55 | def SetEnvVar(env_var, value): |
michael@0 | 56 | """Sets the env variable to 'value'; unsets it when 'value' is None.""" |
michael@0 | 57 | |
michael@0 | 58 | if value is not None: |
michael@0 | 59 | environ[env_var] = value |
michael@0 | 60 | elif env_var in environ: |
michael@0 | 61 | del environ[env_var] |
michael@0 | 62 | |
michael@0 | 63 | |
michael@0 | 64 | def GetFlag(flag): |
michael@0 | 65 | """Runs gtest_env_var_test_ and returns its output.""" |
michael@0 | 66 | |
michael@0 | 67 | args = [COMMAND] |
michael@0 | 68 | if flag is not None: |
michael@0 | 69 | args += [flag] |
michael@0 | 70 | return gtest_test_utils.Subprocess(args, env=environ, |
michael@0 | 71 | capture_stderr=False).output |
michael@0 | 72 | |
michael@0 | 73 | |
michael@0 | 74 | def TestFlag(flag, test_val, default_val): |
michael@0 | 75 | """Verifies that the given flag is affected by the corresponding env var.""" |
michael@0 | 76 | |
michael@0 | 77 | env_var = 'GTEST_' + flag.upper() |
michael@0 | 78 | SetEnvVar(env_var, test_val) |
michael@0 | 79 | AssertEq(test_val, GetFlag(flag)) |
michael@0 | 80 | SetEnvVar(env_var, None) |
michael@0 | 81 | AssertEq(default_val, GetFlag(flag)) |
michael@0 | 82 | |
michael@0 | 83 | |
michael@0 | 84 | class GTestEnvVarTest(gtest_test_utils.TestCase): |
michael@0 | 85 | def testEnvVarAffectsFlag(self): |
michael@0 | 86 | """Tests that environment variable should affect the corresponding flag.""" |
michael@0 | 87 | |
michael@0 | 88 | TestFlag('break_on_failure', '1', '0') |
michael@0 | 89 | TestFlag('color', 'yes', 'auto') |
michael@0 | 90 | TestFlag('filter', 'FooTest.Bar', '*') |
michael@0 | 91 | TestFlag('output', 'xml:tmp/foo.xml', '') |
michael@0 | 92 | TestFlag('print_time', '0', '1') |
michael@0 | 93 | TestFlag('repeat', '999', '1') |
michael@0 | 94 | TestFlag('throw_on_failure', '1', '0') |
michael@0 | 95 | TestFlag('death_test_style', 'threadsafe', 'fast') |
michael@0 | 96 | TestFlag('catch_exceptions', '0', '1') |
michael@0 | 97 | |
michael@0 | 98 | if IS_LINUX: |
michael@0 | 99 | TestFlag('death_test_use_fork', '1', '0') |
michael@0 | 100 | TestFlag('stack_trace_depth', '0', '100') |
michael@0 | 101 | |
michael@0 | 102 | |
michael@0 | 103 | if __name__ == '__main__': |
michael@0 | 104 | gtest_test_utils.Main() |