1
# Copyright (C) 2009, 2010, 2011 Canonical Ltd
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.
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.
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
17
"""A collection of commonly used 'Features' to optionally run tests.
32
class Feature(object):
33
"""An operating system Feature."""
36
self._available = None
39
"""Is the feature available?
41
:return: True if the feature is available.
43
if self._available is None:
44
self._available = self._probe()
45
return self._available
48
"""Implement this method in concrete features.
50
:return: True if the feature is available.
52
raise NotImplementedError
55
if getattr(self, 'feature_name', None):
56
return self.feature_name()
57
return self.__class__.__name__
60
class _SymlinkFeature(Feature):
63
return osutils.has_symlinks()
65
def feature_name(self):
68
SymlinkFeature = _SymlinkFeature()
71
class _HardlinkFeature(Feature):
74
return osutils.has_hardlinks()
76
def feature_name(self):
79
HardlinkFeature = _HardlinkFeature()
82
class _OsFifoFeature(Feature):
85
return getattr(os, 'mkfifo', None)
87
def feature_name(self):
88
return 'filesystem fifos'
90
OsFifoFeature = _OsFifoFeature()
93
class _UnicodeFilenameFeature(Feature):
94
"""Does the filesystem support Unicode filenames?"""
98
# Check for character combinations unlikely to be covered by any
99
# single non-unicode encoding. We use the characters
100
# - greek small letter alpha (U+03B1) and
101
# - braille pattern dots-123456 (U+283F).
102
os.stat(u'\u03b1\u283f')
103
except UnicodeEncodeError:
105
except (IOError, OSError):
106
# The filesystem allows the Unicode filename but the file doesn't
110
# The filesystem allows the Unicode filename and the file exists,
114
UnicodeFilenameFeature = _UnicodeFilenameFeature()
117
class _CompatabilityThunkFeature(Feature):
118
"""This feature is just a thunk to another feature.
120
It issues a deprecation warning if it is accessed, to let you know that you
121
should really use a different feature.
124
def __init__(self, dep_version, module, name,
125
replacement_name, replacement_module=None):
126
super(_CompatabilityThunkFeature, self).__init__()
127
self._module = module
128
if replacement_module is None:
129
replacement_module = module
130
self._replacement_module = replacement_module
132
self._replacement_name = replacement_name
133
self._dep_version = dep_version
137
if self._feature is None:
138
from bzrlib import pyutils
139
depr_msg = self._dep_version % ('%s.%s'
140
% (self._module, self._name))
141
use_msg = ' Use %s.%s instead.' % (self._replacement_module,
142
self._replacement_name)
143
symbol_versioning.warn(depr_msg + use_msg, DeprecationWarning,
145
# Import the new feature and use it as a replacement for the
147
self._feature = pyutils.get_named_object(
148
self._replacement_module, self._replacement_name)
152
return self._feature._probe()
155
class ModuleAvailableFeature(Feature):
156
"""This is a feature than describes a module we want to be available.
158
Declare the name of the module in __init__(), and then after probing, the
159
module will be available as 'self.module'.
161
:ivar module: The module if it is available, else None.
164
def __init__(self, module_name):
165
super(ModuleAvailableFeature, self).__init__()
166
self.module_name = module_name
170
module = sys.modules.get(self.module_name, sentinel)
171
if module is sentinel:
173
self._module = __import__(self.module_name, {}, {}, [''])
178
self._module = module
187
def feature_name(self):
188
return self.module_name
191
class _HTTPSServerFeature(Feature):
192
"""Some tests want an https Server, check if one is available.
194
Right now, the only way this is available is under python2.6 which provides
205
def feature_name(self):
209
HTTPSServerFeature = _HTTPSServerFeature()
212
class _ByteStringNamedFilesystem(Feature):
213
"""Is the filesystem based on bytes?"""
216
if os.name == "posix":
220
ByteStringNamedFilesystem = _ByteStringNamedFilesystem()
223
class _UTF8Filesystem(Feature):
224
"""Is the filesystem UTF-8?"""
227
if osutils._fs_enc.upper() in ('UTF-8', 'UTF8'):
231
UTF8Filesystem = _UTF8Filesystem()
234
class _BreakinFeature(Feature):
235
"""Does this platform support the breakin feature?"""
238
from bzrlib import breakin
239
if breakin.determine_signal() is None:
241
if sys.platform == 'win32':
242
# Windows doesn't have os.kill, and we catch the SIGBREAK signal.
243
# We trigger SIGBREAK via a Console api so we need ctypes to
244
# access the function
251
def feature_name(self):
252
return "SIGQUIT or SIGBREAK w/ctypes on win32"
255
BreakinFeature = _BreakinFeature()
258
class _CaseInsCasePresFilenameFeature(Feature):
259
"""Is the file-system case insensitive, but case-preserving?"""
262
fileno, name = tempfile.mkstemp(prefix='MixedCase')
264
# first check truly case-preserving for created files, then check
265
# case insensitive when opening existing files.
266
name = osutils.normpath(name)
267
base, rel = osutils.split(name)
268
found_rel = osutils.canonical_relpath(base, name)
269
return (found_rel == rel
270
and os.path.isfile(name.upper())
271
and os.path.isfile(name.lower()))
276
def feature_name(self):
277
return "case-insensitive case-preserving filesystem"
279
CaseInsCasePresFilenameFeature = _CaseInsCasePresFilenameFeature()
282
class _CaseInsensitiveFilesystemFeature(Feature):
283
"""Check if underlying filesystem is case-insensitive but *not* case
286
# Note that on Windows, Cygwin, MacOS etc, the file-systems are far
287
# more likely to be case preserving, so this case is rare.
290
if CaseInsCasePresFilenameFeature.available():
293
from bzrlib import tests
295
if tests.TestCaseWithMemoryTransport.TEST_ROOT is None:
296
root = osutils.mkdtemp(prefix='testbzr-', suffix='.tmp')
297
tests.TestCaseWithMemoryTransport.TEST_ROOT = root
299
root = tests.TestCaseWithMemoryTransport.TEST_ROOT
300
tdir = osutils.mkdtemp(prefix='case-sensitive-probe-', suffix='',
302
name_a = osutils.pathjoin(tdir, 'a')
303
name_A = osutils.pathjoin(tdir, 'A')
305
result = osutils.isdir(name_A)
306
tests._rmtree_temp_dir(tdir)
309
def feature_name(self):
310
return 'case-insensitive filesystem'
312
CaseInsensitiveFilesystemFeature = _CaseInsensitiveFilesystemFeature()
315
class _CaseSensitiveFilesystemFeature(Feature):
318
if CaseInsCasePresFilenameFeature.available():
320
elif CaseInsensitiveFilesystemFeature.available():
325
def feature_name(self):
326
return 'case-sensitive filesystem'
328
# new coding style is for feature instances to be lowercase
329
case_sensitive_filesystem_feature = _CaseSensitiveFilesystemFeature()
332
class _NotRunningAsRoot(Feature):
337
except AttributeError:
338
# If there is no uid, chances are there is no root either
342
def feature_name(self):
343
return 'Not running as root'
346
not_running_as_root = _NotRunningAsRoot()
348
apport = ModuleAvailableFeature('apport')
349
gpgme = ModuleAvailableFeature('gpgme')
350
lzma = ModuleAvailableFeature('lzma')
351
meliae = ModuleAvailableFeature('meliae.scanner')
352
paramiko = ModuleAvailableFeature('paramiko')
353
pycurl = ModuleAvailableFeature('pycurl')
354
pywintypes = ModuleAvailableFeature('pywintypes')
355
subunit = ModuleAvailableFeature('subunit')
356
testtools = ModuleAvailableFeature('testtools')
358
compiled_patiencediff_feature = ModuleAvailableFeature(
359
'bzrlib._patiencediff_c')
360
lsprof_feature = ModuleAvailableFeature('bzrlib.lsprof')
363
class _BackslashDirSeparatorFeature(Feature):
367
os.lstat(os.getcwd() + '\\')
373
def feature_name(self):
374
return "Filesystem treats '\\' as a directory separator."
376
backslashdir_feature = _BackslashDirSeparatorFeature()
379
class _ChownFeature(Feature):
380
"""os.chown is supported"""
383
return os.name == 'posix' and hasattr(os, 'chown')
385
chown_feature = _ChownFeature()
388
class ExecutableFeature(Feature):
389
"""Feature testing whether an executable of a given name is on the PATH."""
391
def __init__(self, name):
392
super(ExecutableFeature, self).__init__()
398
# This is a property, so accessing path ensures _probe was called
403
self._path = osutils.find_executable_on_path(self.name)
404
return self._path is not None
406
def feature_name(self):
407
return '%s executable' % self.name
410
bash_feature = ExecutableFeature('bash')
411
diff_feature = ExecutableFeature('diff')
412
sed_feature = ExecutableFeature('sed')
413
msgmerge_feature = ExecutableFeature('msgmerge')
416
class _PosixPermissionsFeature(Feature):
420
# Create temporary file and check if specified perms are
422
write_perms = stat.S_IRUSR | stat.S_IWUSR | stat.S_IXUSR
423
f = tempfile.mkstemp(prefix='bzr_perms_chk_')
426
osutils.chmod_if_possible(name, write_perms)
428
read_perms = os.stat(name).st_mode & 0777
430
return (write_perms == read_perms)
432
return (os.name == 'posix') and has_perms()
434
def feature_name(self):
435
return 'POSIX permissions support'
438
posix_permissions_feature = _PosixPermissionsFeature()
441
class _StraceFeature(Feature):
445
proc = subprocess.Popen(['strace'],
446
stderr=subprocess.PIPE,
447
stdout=subprocess.PIPE)
451
if e.errno == errno.ENOENT:
452
# strace is not installed
457
def feature_name(self):
461
strace_feature = _StraceFeature()
464
class _AttribFeature(Feature):
467
if (sys.platform not in ('cygwin', 'win32')):
470
proc = subprocess.Popen(['attrib', '.'], stdout=subprocess.PIPE)
473
return (0 == proc.wait())
475
def feature_name(self):
476
return 'attrib Windows command-line tool'
479
AttribFeature = _AttribFeature()
482
class Win32Feature(Feature):
483
"""Feature testing whether we're running selftest on Windows
484
or Windows-like platform.
488
return sys.platform == 'win32'
490
def feature_name(self):
491
return "win32 platform"
494
win32_feature = Win32Feature()
497
class _ColorFeature(Feature):
500
from bzrlib._termcolor import allow_color
503
def feature_name(self):
504
return "Terminal supports color."
506
ColorFeature = _ColorFeature()