← 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-10-01 23:09:27 UTC
  • mfrom: (5439.3.4 doc-https-support-596239)
  • Revision ID: pqm@pqm.ubuntu.com-20101001230927-8ucht96vmm4ydiab
(jameinel) Update documentation to make it clearer that we support HTTPS.
 (#596239) (John C Barstow)

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
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:
 
55
            return False
 
56
        else:
 
57
            return True
 
58
 
 
59
    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')