← 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-02-11 04:02:41 UTC
  • mfrom: (5017.2.2 tariff)
  • Revision ID: pqm@pqm.ubuntu.com-20100211040241-w6n021dz0uus341n
(mbp) add import-tariff tests

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/features.py
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
21
17
 
22
18
from bzrlib import tests
23
19
from bzrlib.symbol_versioning import deprecated_in
26
22
apport = tests.ModuleAvailableFeature('apport')
27
23
paramiko = tests.ModuleAvailableFeature('paramiko')
28
24
pycurl = tests.ModuleAvailableFeature('pycurl')
29
 
pywintypes = tests.ModuleAvailableFeature('pywintypes')
30
25
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')