~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/branch_implementations/test_permissions.py

  • Committer: Martin Pool
  • Date: 2009-03-24 05:21:02 UTC
  • mfrom: (4192 +trunk)
  • mto: This revision was merged to the branch mainline in revision 4202.
  • Revision ID: mbp@sourcefrog.net-20090324052102-8kk087b32tep3d9h
merge trunk

Show diffs side-by-side

added added

removed removed

Lines of Context:
22
22
So if the directory is group writable, the files and subdirs should be as well.
23
23
"""
24
24
 
25
 
# TODO: jam 20051215 Currently the default behavior for 'bzr branch' is just 
 
25
# TODO: jam 20051215 Currently the default behavior for 'bzr branch' is just
26
26
#                    defined by the local umask. This isn't terrible, is it
27
27
#                    the truly desired behavior?
28
 
 
 
28
 
29
29
import os
30
30
import sys
31
31
import stat
32
32
from StringIO import StringIO
33
33
 
 
34
from bzrlib import tests
34
35
from bzrlib.branch import Branch
35
36
from bzrlib.bzrdir import BzrDir
36
37
from bzrlib.lockable_files import LockableFiles
37
38
from bzrlib.remote import RemoteBranchFormat
38
 
from bzrlib.tests import TestCaseWithTransport, TestSkipped
39
39
from bzrlib.tests.test_permissions import chmod_r, check_mode_r
40
40
from bzrlib.tests.test_sftp_transport import TestCaseWithSFTPServer
41
41
from bzrlib.transport import get_transport
42
42
from bzrlib.workingtree import WorkingTree
43
43
 
44
44
 
45
 
class TestPermissions(TestCaseWithTransport):
 
45
class _NullPermsStat(object):
 
46
    """A class that proxy's a stat result and strips permissions."""
 
47
 
 
48
    def __init__(self, orig_stat):
 
49
        self._orig_stat = orig_stat
 
50
        # We strip all permission bits from st_mode
 
51
        self.st_mode = orig_stat.st_mode & ~07777
 
52
 
 
53
    def __getattr__(self, name):
 
54
        return getattr(self._orig_stat, name)
 
55
 
 
56
 
 
57
class TestPermissions(tests.TestCaseWithTransport):
46
58
 
47
59
    def test_new_branch(self):
48
60
        if isinstance(self.branch_format, RemoteBranchFormat):
49
61
            # Remote branch format have no permission logic in them; there's
50
62
            # nothing to test here.
51
 
            return
 
63
            raise tests.TestNotApplicable('Remote branches have no'
 
64
                                          ' permission logic')
52
65
        if sys.platform == 'win32':
53
 
            raise TestSkipped('chmod has no effect on win32')
 
66
            raise tests.TestNotApplicable('chmod has no effect on win32')
54
67
        # also, these are BzrBranch format specific things..
55
68
        os.mkdir('a')
56
69
        mode = stat.S_IMODE(os.stat('a').st_mode)
61
74
        self.assertEqualMode(mode, b.control_files._dir_mode)
62
75
        self.assertEqualMode(mode & ~07111, b.control_files._file_mode)
63
76
 
 
77
        os.mkdir('d')
 
78
        os.chmod('d', 0700)
 
79
        b = self.make_branch('d')
 
80
        self.assertEqualMode(0700, b.bzrdir._get_dir_mode())
 
81
        self.assertEqualMode(0600, b.bzrdir._get_file_mode())
 
82
        self.assertEqualMode(0700, b.control_files._dir_mode)
 
83
        self.assertEqualMode(0600, b.control_files._file_mode)
 
84
        check_mode_r(self, 'd/.bzr', 00600, 00700)
 
85
 
 
86
    def test_new_branch_group_sticky_bit(self):
 
87
        if isinstance(self.branch_format, RemoteBranchFormat):
 
88
            # Remote branch format have no permission logic in them; there's
 
89
            # nothing to test here.
 
90
            raise tests.TestNotApplicable('Remote branches have no'
 
91
                                          ' permission logic')
 
92
        if sys.platform == 'win32':
 
93
            raise tests.TestNotApplicable('chmod has no effect on win32')
 
94
        elif sys.platform == 'darwin':
 
95
            # OS X creates temp dirs with the 'wheel' group, which users are
 
96
            # not likely to be in, and this prevents us from setting the sgid
 
97
            # bit
 
98
            os.chown(self.test_dir, os.getuid(), os.getgid())
 
99
        # also, these are BzrBranch format specific things..
 
100
        t = self.make_branch_and_tree('.')
 
101
        b = t.branch
64
102
        os.mkdir('b')
65
103
        os.chmod('b', 02777)
66
104
        b = self.make_branch('b')
79
117
        self.assertEqualMode(00640, b.control_files._file_mode)
80
118
        check_mode_r(self, 'c/.bzr', 00640, 02750)
81
119
 
82
 
        os.mkdir('d')
83
 
        os.chmod('d', 0700)
84
 
        b = self.make_branch('d')
85
 
        self.assertEqualMode(0700, b.bzrdir._get_dir_mode())
86
 
        self.assertEqualMode(0600, b.bzrdir._get_file_mode())
87
 
        self.assertEqualMode(0700, b.control_files._dir_mode)
88
 
        self.assertEqualMode(0600, b.control_files._file_mode)
89
 
        check_mode_r(self, 'd/.bzr', 00600, 00700)
 
120
    def test_mode_0(self):
 
121
        """Test when a transport returns null permissions for .bzr"""
 
122
        if isinstance(self.branch_format, RemoteBranchFormat):
 
123
            # Remote branch format have no permission logic in them; there's
 
124
            # nothing to test here.
 
125
            raise tests.TestNotApplicable('Remote branches have no'
 
126
                                          ' permission logic')
 
127
        self.make_branch_and_tree('.')
 
128
        bzrdir = BzrDir.open('.')
 
129
        # Monkey patch the transport
 
130
        _orig_stat = bzrdir.transport.stat
 
131
        def null_perms_stat(*args, **kwargs):
 
132
            result = _orig_stat(*args, **kwargs)
 
133
            return _NullPermsStat(result)
 
134
        bzrdir.transport.stat = null_perms_stat
 
135
        self.assertIs(None, bzrdir._get_dir_mode())
 
136
        self.assertIs(None, bzrdir._get_file_mode())