← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/features.py

  • Committer: Jelmer Vernooij
  • Date: 2011-02-26 15:39:49 UTC
  • mto: (5691.1.1 bzr.dev)
  • mto: This revision was merged to the branch mainline in revision 5692.
  • Revision ID: jelmer@samba.org-20110226153949-o0fk909b30g7z570
https://launchpad.net/bugs/397556
Fix the use of "bzr tags" in branches with ghosts in their mainline /and/ tags on revisions not in the branch ancestry.

Show diffs side-by-side

added added

removed removed

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