← 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: 2006-07-12 12:36:57 UTC
  • mfrom: (1732.3.4 bzr.revnoX)
  • Revision ID: pqm@pqm.ubuntu.com-20060712123657-365eeb32b69308bf
(matthieu) revno:x:url revision spec

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/features.py
1
 
# Copyright (C) 2009, 2010 Canonical Ltd
2
 
#
3
 
# This program is free software; you can redistribute it and/or modify
4
 
# it under the terms of the GNU General Public License as published by
5
 
# the Free Software Foundation; either version 2 of the License, or
6
 
# (at your option) any later version.
7
 
#
8
 
# This program is distributed in the hope that it will be useful,
9
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
 
# GNU General Public License for more details.
12
 
#
13
 
# You should have received a copy of the GNU General Public License
14
 
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
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
 
 
42
 
apport = tests.ModuleAvailableFeature('apport')
43
 
meliae = tests.ModuleAvailableFeature('meliae')
44
 
paramiko = tests.ModuleAvailableFeature('paramiko')
45
 
pycurl = tests.ModuleAvailableFeature('pycurl')
46
 
pywintypes = tests.ModuleAvailableFeature('pywintypes')
47
 
sphinx = tests.ModuleAvailableFeature('sphinx')
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')