← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/features.py

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/features.py
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
import os
 
18
import stat
17
19
 
18
20
from bzrlib import tests
19
21
from bzrlib.symbol_versioning import deprecated_in
23
25
paramiko = tests.ModuleAvailableFeature('paramiko')
24
26
pycurl = tests.ModuleAvailableFeature('pycurl')
25
27
subunit = tests.ModuleAvailableFeature('subunit')
 
28
 
 
29
 
 
30
class _PosixPermissionsFeature(tests.Feature):
 
31
 
 
32
    def _probe(self):
 
33
        def has_perms():
 
34
            # create temporary file and check if specified perms are maintained.
 
35
            import tempfile
 
36
 
 
37
            write_perms = stat.S_IRUSR | stat.S_IWUSR | stat.S_IXUSR
 
38
            f = tempfile.mkstemp(prefix='bzr_perms_chk_')
 
39
            fd, name = f
 
40
            os.close(fd)
 
41
            os.chmod(name, write_perms)
 
42
 
 
43
            read_perms = os.stat(name).st_mode & 0777
 
44
            os.unlink(name)
 
45
            return (write_perms == read_perms)
 
46
 
 
47
        return (os.name == 'posix') and has_perms()
 
48
 
 
49
    def feature_name(self):
 
50
        return 'POSIX permissions support'
 
51
 
 
52
 
 
53
posix_permissions_feature = _PosixPermissionsFeature()
 
54
 
 
55
 
 
56
class _ChownFeature(tests.Feature):
 
57
    """os.chown is supported"""
 
58
 
 
59
    def _probe(self):
 
60
        return os.name == 'posix' and hasattr(os, 'chown')
 
61
 
 
62
chown_feature = _ChownFeature()
 
63