github.com/jlmucb/cloudproxy@v0.0.0-20170830161738-b5aa0b619bc4/src/third_party/googlemock/gtest/test/gtest_env_var_test.py (about)

     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()