← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/features.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-09-01 08:02:42 UTC
  • mfrom: (5390.3.3 faster-revert-593560)
  • Revision ID: pqm@pqm.ubuntu.com-20100901080242-esg62ody4frwmy66
(spiv) Avoid repeatedly calling self.target.all_file_ids() in
 InterTree.iter_changes. (Andrew Bennetts)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/features.py
1
 
# Copyright (C) 2009 Canonical Ltd
 
1
# Copyright (C) 2009, 2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
 
17
"""A collection of commonly used 'Features' which bzrlib uses to skip tests."""
 
18
 
 
19
import os
 
20
import stat
17
21
 
18
22
from bzrlib import tests
19
23
from bzrlib.symbol_versioning import deprecated_in
20
24
 
21
25
 
22
26
apport = tests.ModuleAvailableFeature('apport')
23
 
ApportFeature = tests._CompatabilityThunkFeature('bzrlib.tests.features',
24
 
    'ApportFeature', 'bzrlib.tests.features.apport', deprecated_in((2,1,0)))
25
27
paramiko = tests.ModuleAvailableFeature('paramiko')
26
28
pycurl = tests.ModuleAvailableFeature('pycurl')
 
29
pywintypes = tests.ModuleAvailableFeature('pywintypes')
27
30
subunit = tests.ModuleAvailableFeature('subunit')
 
31
sphinx = tests.ModuleAvailableFeature('sphinx')
 
32
 
 
33
 
 
34
class _BackslashDirSeparatorFeature(tests.Feature):
 
35
 
 
36
    def _probe(self):
 
37
        try:
 
38
            os.lstat(os.getcwd() + '\\')
 
39
        except OSError:
 
40
            return False
 
41
        else:
 
42
            return True
 
43
 
 
44
    def feature_name(self):
 
45
        return "Filesystem treats '\\' as a directory separator."
 
46
 
 
47
backslashdir_feature = _BackslashDirSeparatorFeature()
 
48
 
 
49
 
 
50
class _PosixPermissionsFeature(tests.Feature):
 
51
 
 
52
    def _probe(self):
 
53
        def has_perms():
 
54
            # create temporary file and check if specified perms are maintained.
 
55
            import tempfile
 
56
 
 
57
            write_perms = stat.S_IRUSR | stat.S_IWUSR | stat.S_IXUSR
 
58
            f = tempfile.mkstemp(prefix='bzr_perms_chk_')
 
59
            fd, name = f
 
60
            os.close(fd)
 
61
            os.chmod(name, write_perms)
 
62
 
 
63
            read_perms = os.stat(name).st_mode & 0777
 
64
            os.unlink(name)
 
65
            return (write_perms == read_perms)
 
66
 
 
67
        return (os.name == 'posix') and has_perms()
 
68
 
 
69
    def feature_name(self):
 
70
        return 'POSIX permissions support'
 
71
 
 
72
 
 
73
posix_permissions_feature = _PosixPermissionsFeature()
 
74
 
 
75
 
 
76
class _ChownFeature(tests.Feature):
 
77
    """os.chown is supported"""
 
78
 
 
79
    def _probe(self):
 
80
        return os.name == 'posix' and hasattr(os, 'chown')
 
81
 
 
82
chown_feature = _ChownFeature()
 
83
 
 
84
 
 
85
class ExecutableFeature(tests.Feature):
 
86
    """Feature testing whether an executable of a given name is on the PATH."""
 
87
 
 
88
    def __init__(self, name):
 
89
        super(ExecutableFeature, self).__init__()
 
90
        self.name = name
 
91
        self._path = None
 
92
 
 
93
    @property
 
94
    def path(self):
 
95
        # This is a property, so accessing path ensures _probe was called
 
96
        self.available()
 
97
        return self._path
 
98
 
 
99
    def _probe(self):
 
100
        path = os.environ.get('PATH')
 
101
        if path is None:
 
102
            return False
 
103
        for d in path.split(os.pathsep):
 
104
            if d:
 
105
                f = os.path.join(d, self.name)
 
106
                if os.access(f, os.X_OK):
 
107
                    self._path = f
 
108
                    return True
 
109
        return False
 
110
 
 
111
    def feature_name(self):
 
112
        return '%s executable' % self.name
 
113
 
 
114
 
 
115
bash_feature = ExecutableFeature('bash')
 
116
sed_feature = ExecutableFeature('sed')
 
117
diff_feature = ExecutableFeature('diff')