gecko-dev/python/which/test/test_which.py

#!/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()