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