OLD | NEW |
| (Empty) |
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).output | |
71 | |
72 | |
73 def TestFlag(flag, test_val, default_val): | |
74 """Verifies that the given flag is affected by the corresponding env var.""" | |
75 | |
76 env_var = 'GTEST_' + flag.upper() | |
77 SetEnvVar(env_var, test_val) | |
78 AssertEq(test_val, GetFlag(flag)) | |
79 SetEnvVar(env_var, None) | |
80 AssertEq(default_val, GetFlag(flag)) | |
81 | |
82 | |
83 class GTestEnvVarTest(gtest_test_utils.TestCase): | |
84 def testEnvVarAffectsFlag(self): | |
85 """Tests that environment variable should affect the corresponding flag.""" | |
86 | |
87 TestFlag('break_on_failure', '1', '0') | |
88 TestFlag('color', 'yes', 'auto') | |
89 TestFlag('filter', 'FooTest.Bar', '*') | |
90 TestFlag('output', 'xml:tmp/foo.xml', '') | |
91 TestFlag('print_time', '0', '1') | |
92 TestFlag('repeat', '999', '1') | |
93 TestFlag('throw_on_failure', '1', '0') | |
94 TestFlag('death_test_style', 'threadsafe', 'fast') | |
95 TestFlag('catch_exceptions', '0', '1') | |
96 | |
97 if IS_LINUX: | |
98 TestFlag('death_test_use_fork', '1', '0') | |
99 TestFlag('stack_trace_depth', '0', '100') | |
100 | |
101 | |
102 if __name__ == '__main__': | |
103 gtest_test_utils.Main() | |
OLD | NEW |