1dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter#!/usr/bin/env python 2dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# 3dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# Copyright 2008, Google Inc. 4dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# All rights reserved. 5dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# 6dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# Redistribution and use in source and binary forms, with or without 7dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# modification, are permitted provided that the following conditions are 8dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# met: 9dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# 10dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# * Redistributions of source code must retain the above copyright 11dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# notice, this list of conditions and the following disclaimer. 12dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# * Redistributions in binary form must reproduce the above 13dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# copyright notice, this list of conditions and the following disclaimer 14dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# in the documentation and/or other materials provided with the 15dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# distribution. 16dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# * Neither the name of Google Inc. nor the names of its 17dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# contributors may be used to endorse or promote products derived from 18dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# this software without specific prior written permission. 19dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# 20dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 21dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 22dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 23dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 24dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 25dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 26dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 27dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 28dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 29dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 30dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 31dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 32dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter"""Verifies that Google Test correctly determines whether to use colors.""" 33dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 34dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter__author__ = 'wan@google.com (Zhanyong Wan)' 35dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 36dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixterimport os 37dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixterimport gtest_test_utils 38dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 39dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 40dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken MixterIS_WINDOWS = os.name = 'nt' 41dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 42dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken MixterCOLOR_ENV_VAR = 'GTEST_COLOR' 43dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken MixterCOLOR_FLAG = 'gtest_color' 44dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken MixterCOMMAND = gtest_test_utils.GetTestExecutablePath('gtest_color_test_') 45dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 46dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 47dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixterdef SetEnvVar(env_var, value): 48dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter """Sets the env variable to 'value'; unsets it when 'value' is None.""" 49dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 50dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter if value is not None: 51dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter os.environ[env_var] = value 52dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter elif env_var in os.environ: 53dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter del os.environ[env_var] 54dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 55dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 56dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixterdef UsesColor(term, color_env_var, color_flag): 57dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter """Runs gtest_color_test_ and returns its exit code.""" 58dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 59dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter SetEnvVar('TERM', term) 60dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter SetEnvVar(COLOR_ENV_VAR, color_env_var) 6146108a219a4b812dd8f36fee479a0340ea5963f5Ben Chan 6246108a219a4b812dd8f36fee479a0340ea5963f5Ben Chan if color_flag is None: 6346108a219a4b812dd8f36fee479a0340ea5963f5Ben Chan args = [] 6446108a219a4b812dd8f36fee479a0340ea5963f5Ben Chan else: 6546108a219a4b812dd8f36fee479a0340ea5963f5Ben Chan args = ['--%s=%s' % (COLOR_FLAG, color_flag)] 6646108a219a4b812dd8f36fee479a0340ea5963f5Ben Chan p = gtest_test_utils.Subprocess([COMMAND] + args) 6746108a219a4b812dd8f36fee479a0340ea5963f5Ben Chan return not p.exited or p.exit_code 68dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 69dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 70dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixterclass GTestColorTest(gtest_test_utils.TestCase): 71dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter def testNoEnvVarNoFlag(self): 72dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter """Tests the case when there's neither GTEST_COLOR nor --gtest_color.""" 73dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 74dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter if not IS_WINDOWS: 75dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(not UsesColor('dumb', None, None)) 76dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(not UsesColor('emacs', None, None)) 77dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(not UsesColor('xterm-mono', None, None)) 78dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(not UsesColor('unknown', None, None)) 79dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(not UsesColor(None, None, None)) 8046108a219a4b812dd8f36fee479a0340ea5963f5Ben Chan self.assert_(UsesColor('linux', None, None)) 81dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(UsesColor('cygwin', None, None)) 82dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(UsesColor('xterm', None, None)) 83dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(UsesColor('xterm-color', None, None)) 84dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(UsesColor('xterm-256color', None, None)) 85dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 86dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter def testFlagOnly(self): 87dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter """Tests the case when there's --gtest_color but not GTEST_COLOR.""" 88dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 89dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(not UsesColor('dumb', None, 'no')) 90dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(not UsesColor('xterm-color', None, 'no')) 91dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter if not IS_WINDOWS: 92dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(not UsesColor('emacs', None, 'auto')) 93dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(UsesColor('xterm', None, 'auto')) 94dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(UsesColor('dumb', None, 'yes')) 95dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(UsesColor('xterm', None, 'yes')) 96dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 97dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter def testEnvVarOnly(self): 98dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter """Tests the case when there's GTEST_COLOR but not --gtest_color.""" 99dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 100dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(not UsesColor('dumb', 'no', None)) 101dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(not UsesColor('xterm-color', 'no', None)) 102dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter if not IS_WINDOWS: 103dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(not UsesColor('dumb', 'auto', None)) 104dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(UsesColor('xterm-color', 'auto', None)) 105dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(UsesColor('dumb', 'yes', None)) 106dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(UsesColor('xterm-color', 'yes', None)) 107dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 108dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter def testEnvVarAndFlag(self): 109dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter """Tests the case when there are both GTEST_COLOR and --gtest_color.""" 110dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 111dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(not UsesColor('xterm-color', 'no', 'no')) 112dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(UsesColor('dumb', 'no', 'yes')) 113dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(UsesColor('xterm-color', 'no', 'auto')) 114dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 115dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter def testAliasesOfYesAndNo(self): 116dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter """Tests using aliases in specifying --gtest_color.""" 117dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 118dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(UsesColor('dumb', None, 'true')) 119dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(UsesColor('dumb', None, 'YES')) 120dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(UsesColor('dumb', None, 'T')) 121dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(UsesColor('dumb', None, '1')) 122dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 123dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(not UsesColor('xterm', None, 'f')) 124dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(not UsesColor('xterm', None, 'false')) 125dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(not UsesColor('xterm', None, '0')) 126dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter self.assert_(not UsesColor('xterm', None, 'unknown')) 127dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 128dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter 129dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixterif __name__ == '__main__': 130dd1c93d5709e32713961cfd95fe30489a4ad2d26Ken Mixter gtest_test_utils.Main() 131