← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/features.py

  • Committer: Martin Pool
  • Date: 2009-09-14 02:30:23 UTC
  • mto: This revision was merged to the branch mainline in revision 4693.
  • Revision ID: mbp@sourcefrog.net-20090914023023-ros0f3ndo04j3bww
https://launchpad.net/bugs/429151
Clearer docs about bzr help.  (Thanks to Naoki)

Show diffs side-by-side

added added

removed removed

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