← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/features.py

(gz) Backslash escape selftest output when printing to non-unicode consoles
 (Martin [gz])

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
 
 
26
class _NotRunningAsRoot(tests.Feature):
 
27
 
 
28
    def _probe(self):
 
29
        try:
 
30
            uid = os.getuid()
 
31
        except AttributeError:
 
32
            # If there is no uid, chances are there is no root either
 
33
            return True
 
34
        return uid != 0
 
35
 
 
36
    def feature_name(self):
 
37
        return 'Not running as root'
 
38
 
 
39
 
 
40
not_running_as_root = _NotRunningAsRoot()
 
41
 
22
42
apport = tests.ModuleAvailableFeature('apport')
23
 
ApportFeature = tests._CompatabilityThunkFeature('bzrlib.tests.features',
24
 
    'ApportFeature', 'bzrlib.tests.features.apport', deprecated_in((2,1,0)))
 
43
meliae = tests.ModuleAvailableFeature('meliae')
25
44
paramiko = tests.ModuleAvailableFeature('paramiko')
26
45
pycurl = tests.ModuleAvailableFeature('pycurl')
 
46
pywintypes = tests.ModuleAvailableFeature('pywintypes')
 
47
sphinx = tests.ModuleAvailableFeature('sphinx')
27
48
subunit = tests.ModuleAvailableFeature('subunit')
 
49
 
 
50
 
 
51
class _BackslashDirSeparatorFeature(tests.Feature):
 
52
 
 
53
    def _probe(self):
 
54
        try:
 
55
            os.lstat(os.getcwd() + '\\')
 
56
        except OSError:
 
57
            return False
 
58
        else:
 
59
            return True
 
60
 
 
61
    def feature_name(self):
 
62
        return "Filesystem treats '\\' as a directory separator."
 
63
 
 
64
backslashdir_feature = _BackslashDirSeparatorFeature()
 
65
 
 
66
 
 
67
class _PosixPermissionsFeature(tests.Feature):
 
68
 
 
69
    def _probe(self):
 
70
        def has_perms():
 
71
            # create temporary file and check if specified perms are maintained.
 
72
            import tempfile
 
73
 
 
74
            write_perms = stat.S_IRUSR | stat.S_IWUSR | stat.S_IXUSR
 
75
            f = tempfile.mkstemp(prefix='bzr_perms_chk_')
 
76
            fd, name = f
 
77
            os.close(fd)
 
78
            os.chmod(name, write_perms)
 
79
 
 
80
            read_perms = os.stat(name).st_mode & 0777
 
81
            os.unlink(name)
 
82
            return (write_perms == read_perms)
 
83
 
 
84
        return (os.name == 'posix') and has_perms()
 
85
 
 
86
    def feature_name(self):
 
87
        return 'POSIX permissions support'
 
88
 
 
89
 
 
90
posix_permissions_feature = _PosixPermissionsFeature()
 
91
 
 
92
 
 
93
class _ChownFeature(tests.Feature):
 
94
    """os.chown is supported"""
 
95
 
 
96
    def _probe(self):
 
97
        return os.name == 'posix' and hasattr(os, 'chown')
 
98
 
 
99
chown_feature = _ChownFeature()
 
100
 
 
101
 
 
102
class ExecutableFeature(tests.Feature):
 
103
    """Feature testing whether an executable of a given name is on the PATH."""
 
104
 
 
105
    def __init__(self, name):
 
106
        super(ExecutableFeature, self).__init__()
 
107
        self.name = name
 
108
        self._path = None
 
109
 
 
110
    @property
 
111
    def path(self):
 
112
        # This is a property, so accessing path ensures _probe was called
 
113
        self.available()
 
114
        return self._path
 
115
 
 
116
    def _probe(self):
 
117
        path = os.environ.get('PATH')
 
118
        if path is None:
 
119
            return False
 
120
        for d in path.split(os.pathsep):
 
121
            if d:
 
122
                f = os.path.join(d, self.name)
 
123
                if os.access(f, os.X_OK):
 
124
                    self._path = f
 
125
                    return True
 
126
        return False
 
127
 
 
128
    def feature_name(self):
 
129
        return '%s executable' % self.name
 
130
 
 
131
 
 
132
bash_feature = ExecutableFeature('bash')
 
133
sed_feature = ExecutableFeature('sed')
 
134
diff_feature = ExecutableFeature('diff')