Bug 792135 - Part 1: Add which Python package; r=glandium
Version 1.1.0 obtained from https://code.google.com/p/which/ and added to tree without modifications aside from the removal of which.exe, which has no reason to be in the tree.
This commit is contained in:
Gregory Szorc
168
python/which/test/test_which.py
Normal file
168
python/which/test/test_which.py
Normal file
@@ -0,0 +1,168 @@
|
||||
#!/usr/bin/env python
|
||||
# Copyright (c) 2002-2003 ActiveState Corp.
|
||||
# Author: Trent Mick (TrentM@ActiveState.com)
|
||||
|
||||
"""Test suite for which.py."""
|
||||
|
||||
import sys
|
||||
import os
|
||||
import re
|
||||
import tempfile
|
||||
import unittest
|
||||
|
||||
import testsupport
|
||||
|
||||
#XXX:TODO
|
||||
# - def test_registry_success(self): ...App Paths setting
|
||||
# - def test_registry_noexist(self):
|
||||
# - test all the other options
|
||||
# - test on linux
|
||||
# - test the module API
|
||||
|
||||
class WhichTestCase(unittest.TestCase):
|
||||
def setUp(self):
|
||||
"""Create a temp directory with a couple test "commands".
|
||||
The temp dir can be added to the PATH, etc, for testing purposes.
|
||||
"""
|
||||
# Find the which.py to call.
|
||||
whichPy = os.path.join(os.path.dirname(__file__),
|
||||
os.pardir, "which.py")
|
||||
self.which = sys.executable + " " + whichPy
|
||||
|
||||
# Setup the test environment.
|
||||
self.tmpdir = tempfile.mktemp()
|
||||
os.makedirs(self.tmpdir)
|
||||
if sys.platform.startswith("win"):
|
||||
self.testapps = ['whichtestapp1.exe',
|
||||
'whichtestapp2.exe',
|
||||
'whichtestapp3.wta']
|
||||
else:
|
||||
self.testapps = ['whichtestapp1', 'whichtestapp2']
|
||||
for app in self.testapps:
|
||||
path = os.path.join(self.tmpdir, app)
|
||||
open(path, 'wb').write('\n')
|
||||
os.chmod(path, 0755)
|
||||
|
||||
def tearDown(self):
|
||||
testsupport.rmtree(self.tmpdir)
|
||||
|
||||
def test_opt_h(self):
|
||||
output, error, retval = testsupport.run(self.which+' --h')
|
||||
token = 'Usage:'
|
||||
self.failUnless(output.find(token) != -1,
|
||||
"'%s' was not found in 'which -h' output: '%s' "\
|
||||
% (token, output))
|
||||
self.failUnless(retval == 0,
|
||||
"'which -h' did not return 0: retval=%d" % retval)
|
||||
|
||||
def test_opt_help(self):
|
||||
output, error, retval = testsupport.run(self.which+' --help')
|
||||
token = 'Usage:'
|
||||
self.failUnless(output.find(token) != -1,
|
||||
"'%s' was not found in 'which --help' output: '%s' "\
|
||||
% (token, output))
|
||||
self.failUnless(retval == 0,
|
||||
"'which --help' did not return 0: retval=%d" % retval)
|
||||
|
||||
def test_opt_version(self):
|
||||
output, error, retval = testsupport.run(self.which+' --version')
|
||||
versionRe = re.compile("^which \d+\.\d+\.\d+$")
|
||||
versionMatch = versionRe.search(output.strip())
|
||||
self.failUnless(versionMatch,
|
||||
"Version, '%s', from 'which --version' does not "\
|
||||
"match pattern, '%s'."\
|
||||
% (output.strip(), versionRe.pattern))
|
||||
self.failUnless(retval == 0,
|
||||
"'which --version' did not return 0: retval=%d"\
|
||||
% retval)
|
||||
|
||||
def test_no_args(self):
|
||||
output, error, retval = testsupport.run(self.which)
|
||||
self.failUnless(retval == -1,
|
||||
"'which' with no args should return -1: retval=%d"\
|
||||
% retval)
|
||||
|
||||
def test_one_failure(self):
|
||||
output, error, retval = testsupport.run(
|
||||
self.which+' whichtestapp1')
|
||||
self.failUnless(retval == 1,
|
||||
"One failure did not return 1: retval=%d" % retval)
|
||||
|
||||
def test_two_failures(self):
|
||||
output, error, retval = testsupport.run(
|
||||
self.which+' whichtestapp1 whichtestapp2')
|
||||
self.failUnless(retval == 2,
|
||||
"Two failures did not return 2: retval=%d" % retval)
|
||||
|
||||
def _match(self, path1, path2):
|
||||
#print "_match: %r =?= %r" % (path1, path2)
|
||||
if sys.platform.startswith('win'):
|
||||
path1 = os.path.normpath(os.path.normcase(path1))
|
||||
path2 = os.path.normpath(os.path.normcase(path2))
|
||||
path1 = os.path.splitext(path1)[0]
|
||||
path2 = os.path.splitext(path2)[0]
|
||||
return path1 == path2
|
||||
else:
|
||||
return os.path.samefile(path1, path2)
|
||||
|
||||
def test_one_success(self):
|
||||
os.environ["PATH"] += os.pathsep + self.tmpdir
|
||||
output, error, retval = testsupport.run(self.which+' -q whichtestapp1')
|
||||
expectedOutput = os.path.join(self.tmpdir, "whichtestapp1")
|
||||
self.failUnless(self._match(output.strip(), expectedOutput),
|
||||
"Output, %r, and expected output, %r, do not match."\
|
||||
% (output.strip(), expectedOutput))
|
||||
self.failUnless(retval == 0,
|
||||
"'which ...' should have returned 0: retval=%d" % retval)
|
||||
|
||||
def test_two_successes(self):
|
||||
os.environ["PATH"] += os.pathsep + self.tmpdir
|
||||
apps = ['whichtestapp1', 'whichtestapp2']
|
||||
output, error, retval = testsupport.run(
|
||||
self.which + ' -q ' + ' '.join(apps))
|
||||
lines = output.strip().split("\n")
|
||||
for app, line in zip(apps, lines):
|
||||
expected = os.path.join(self.tmpdir, app)
|
||||
self.failUnless(self._match(line, expected),
|
||||
"Output, %r, and expected output, %r, do not match."\
|
||||
% (line, expected))
|
||||
self.failUnless(retval == 0,
|
||||
"'which ...' should have returned 0: retval=%d" % retval)
|
||||
|
||||
if sys.platform.startswith("win"):
|
||||
def test_PATHEXT_failure(self):
|
||||
os.environ["PATH"] += os.pathsep + self.tmpdir
|
||||
output, error, retval = testsupport.run(self.which+' whichtestapp3')
|
||||
self.failUnless(retval == 1,
|
||||
"'which ...' should have returned 1: retval=%d" % retval)
|
||||
|
||||
def test_PATHEXT_success(self):
|
||||
os.environ["PATH"] += os.pathsep + self.tmpdir
|
||||
os.environ["PATHEXT"] += os.pathsep + '.wta'
|
||||
output, error, retval = testsupport.run(self.which+' whichtestapp3')
|
||||
expectedOutput = os.path.join(self.tmpdir, "whichtestapp3")
|
||||
self.failUnless(self._match(output.strip(), expectedOutput),
|
||||
"Output, %r, and expected output, %r, do not match."\
|
||||
% (output.strip(), expectedOutput))
|
||||
self.failUnless(retval == 0,
|
||||
"'which ...' should have returned 0: retval=%d" % retval)
|
||||
|
||||
def test_exts(self):
|
||||
os.environ["PATH"] += os.pathsep + self.tmpdir
|
||||
output, error, retval = testsupport.run(self.which+' -e .wta whichtestapp3')
|
||||
expectedOutput = os.path.join(self.tmpdir, "whichtestapp3")
|
||||
self.failUnless(self._match(output.strip(), expectedOutput),
|
||||
"Output, %r, and expected output, %r, do not match."\
|
||||
% (output.strip(), expectedOutput))
|
||||
self.failUnless(retval == 0,
|
||||
"'which ...' should have returned 0: retval=%d" % retval)
|
||||
|
||||
|
||||
|
||||
def suite():
|
||||
"""Return a unittest.TestSuite to be used by test.py."""
|
||||
return unittest.makeSuite(WhichTestCase)
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
|
||||
83
python/which/test/testsupport.py
Normal file
83
python/which/test/testsupport.py
Normal file
@@ -0,0 +1,83 @@
|
||||
#!/usr/bin/env python
|
||||
# Copyright (c) 2002-2003 ActiveState Corp.
|
||||
# Author: Trent Mick (TrentM@ActiveState.com)
|
||||
|
||||
import os
|
||||
import sys
|
||||
import types
|
||||
|
||||
|
||||
#---- Support routines
|
||||
|
||||
def _escapeArg(arg):
|
||||
"""Escape the given command line argument for the shell."""
|
||||
#XXX There is a *lot* more that we should escape here.
|
||||
return arg.replace('"', r'\"')
|
||||
|
||||
|
||||
def _joinArgv(argv):
|
||||
r"""Join an arglist to a string appropriate for running.
|
||||
>>> import os
|
||||
>>> _joinArgv(['foo', 'bar "baz'])
|
||||
'foo "bar \\"baz"'
|
||||
"""
|
||||
cmdstr = ""
|
||||
for arg in argv:
|
||||
if ' ' in arg:
|
||||
cmdstr += '"%s"' % _escapeArg(arg)
|
||||
else:
|
||||
cmdstr += _escapeArg(arg)
|
||||
cmdstr += ' '
|
||||
if cmdstr.endswith(' '): cmdstr = cmdstr[:-1] # strip trailing space
|
||||
return cmdstr
|
||||
|
||||
|
||||
def run(argv):
|
||||
"""Prepare and run the given arg vector, 'argv', and return the
|
||||
results. Returns (<stdout lines>, <stderr lines>, <return value>).
|
||||
Note: 'argv' may also just be the command string.
|
||||
"""
|
||||
if type(argv) in (types.ListType, types.TupleType):
|
||||
cmd = _joinArgv(argv)
|
||||
else:
|
||||
cmd = argv
|
||||
if sys.platform.startswith('win'):
|
||||
i, o, e = os.popen3(cmd)
|
||||
output = o.read()
|
||||
error = e.read()
|
||||
i.close()
|
||||
e.close()
|
||||
try:
|
||||
retval = o.close()
|
||||
except IOError:
|
||||
# IOError is raised iff the spawned app returns -1. Go
|
||||
# figure.
|
||||
retval = -1
|
||||
if retval is None:
|
||||
retval = 0
|
||||
else:
|
||||
import popen2
|
||||
p = popen2.Popen3(cmd, 1)
|
||||
i, o, e = p.tochild, p.fromchild, p.childerr
|
||||
output = o.read()
|
||||
error = e.read()
|
||||
i.close()
|
||||
o.close()
|
||||
e.close()
|
||||
retval = (p.wait() & 0xFF00) >> 8
|
||||
if retval > 2**7: # 8-bit signed 1's-complement conversion
|
||||
retval -= 2**8
|
||||
return output, error, retval
|
||||
|
||||
|
||||
def _rmtreeOnError(rmFunction, filePath, excInfo):
|
||||
if excInfo[0] == OSError:
|
||||
# presuming because file is read-only
|
||||
os.chmod(filePath, 0777)
|
||||
rmFunction(filePath)
|
||||
|
||||
def rmtree(dirname):
|
||||
import shutil
|
||||
shutil.rmtree(dirname, 0, _rmtreeOnError)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user