← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_permissions.py

  • Committer: Aaron Bentley
  • Date: 2006-06-14 19:45:57 UTC
  • mto: This revision was merged to the branch mainline in revision 1777.
  • Revision ID: abentley@panoramicfeedback.com-20060614194557-6b499aa1cf03f7e6
Use create_signature for signing policy, deprecate check_signatures for this

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_permissions.py
1
 
# Copyright (C) 2005 Canonical Ltd
 
1
# Copyright (C) 2005 by Canonical Ltd
2
2
# -*- coding: utf-8 -*-
3
 
#
 
3
 
4
4
# This program is free software; you can redistribute it and/or modify
5
5
# it under the terms of the GNU General Public License as published by
6
6
# the Free Software Foundation; either version 2 of the License, or
7
7
# (at your option) any later version.
8
 
#
 
8
 
9
9
# This program is distributed in the hope that it will be useful,
10
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
12
# GNU General Public License for more details.
13
 
#
 
13
 
14
14
# You should have received a copy of the GNU General Public License
15
15
# along with this program; if not, write to the Free Software
16
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
29
29
# TODO: jam 20051215 Currently the default behavior for 'bzr branch' is just 
30
30
#                    defined by the local umask. This isn't terrible, is it
31
31
#                    the truly desired behavior?
32
 
 
 
32
 
33
33
import os
34
34
import sys
35
35
import stat
36
 
from cStringIO import StringIO
37
 
import urllib
 
36
from StringIO import StringIO
38
37
 
39
38
from bzrlib.branch import Branch
40
39
from bzrlib.bzrdir import BzrDir
47
46
 
48
47
def chmod_r(base, file_mode, dir_mode):
49
48
    """Recursively chmod from a base directory"""
 
49
    assert os.path.isdir(base)
50
50
    os.chmod(base, dir_mode)
51
51
    for root, dirs, files in os.walk(base):
52
52
        for d in dirs:
66
66
    :param dir_mode: The mode for all directories
67
67
    :param include_base: If false, only check the subdirectories
68
68
    """
 
69
    assert os.path.isdir(base)
69
70
    t = get_transport(".")
70
71
    if include_base:
71
72
        test.assertTransportMode(t, base, dir_mode)
72
73
    for root, dirs, files in os.walk(base):
73
74
        for d in dirs:
74
 
            p = '/'.join([urllib.quote(x) for x in root.split('/\\') + [d]])
 
75
            p = os.path.join(root, d)
75
76
            test.assertTransportMode(t, p, dir_mode)
76
77
        for f in files:
77
78
            p = os.path.join(root, f)
78
 
            p = '/'.join([urllib.quote(x) for x in root.split('/\\') + [f]])
79
79
            test.assertTransportMode(t, p, file_mode)
80
80
 
81
81
 
88
88
        t = self.make_branch_and_tree('.')
89
89
        b = t.branch
90
90
        open('a', 'wb').write('foo\n')
91
 
        # ensure check_mode_r works with capital-letter file-ids like TREE_ROOT
92
 
        t.add('a', 'CAPS-ID')
 
91
        t.add('a')
93
92
        t.commit('foo')
94
93
 
95
94
        chmod_r('.bzr', 0644, 0755)
104
103
        b = t.branch
105
104
        self.assertEqualMode(0755, b.control_files._dir_mode)
106
105
        self.assertEqualMode(0644, b.control_files._file_mode)
107
 
        self.assertEqualMode(0755, b.bzrdir._get_dir_mode())
108
 
        self.assertEqualMode(0644, b.bzrdir._get_file_mode())
109
106
 
110
107
        # Modifying a file shouldn't break the permissions
111
108
        open('a', 'wb').write('foo2\n')
126
123
        b = t.branch
127
124
        self.assertEqualMode(0775, b.control_files._dir_mode)
128
125
        self.assertEqualMode(0664, b.control_files._file_mode)
129
 
        self.assertEqualMode(0775, b.bzrdir._get_dir_mode())
130
 
        self.assertEqualMode(0664, b.bzrdir._get_file_mode())
131
126
 
132
127
        open('a', 'wb').write('foo3\n')
133
128
        t.commit('foo3')
138
133
        t.commit('new c')
139
134
        check_mode_r(self, '.bzr', 0664, 0775)
140
135
 
141
 
    def test_new_files_group_sticky_bit(self):
142
 
        if sys.platform == 'win32':
143
 
            raise TestSkipped('chmod has no effect on win32')
144
 
        elif sys.platform == 'darwin':
145
 
            # OS X creates temp dirs with the 'wheel' group, which users are
146
 
            # not likely to be in, and this prevents us from setting the sgid
147
 
            # bit
148
 
            os.chown(self.test_dir, os.getuid(), os.getgid())
149
 
 
150
 
        t = self.make_branch_and_tree('.')
151
 
        b = t.branch
152
 
 
153
136
        # Test the group sticky bit
154
137
        # Recursively update the modes of all files
155
138
        chmod_r('.bzr', 0664, 02775)
158
141
        b = t.branch
159
142
        self.assertEqualMode(02775, b.control_files._dir_mode)
160
143
        self.assertEqualMode(0664, b.control_files._file_mode)
161
 
        self.assertEqualMode(02775, b.bzrdir._get_dir_mode())
162
 
        self.assertEqualMode(0664, b.bzrdir._get_file_mode())
163
144
 
164
145
        open('a', 'wb').write('foo4\n')
165
146
        t.commit('foo4')
170
151
        t.commit('new d')
171
152
        check_mode_r(self, '.bzr', 0664, 02775)
172
153
 
 
154
    def test_disable_set_mode(self):
 
155
        # TODO: jam 20051215 Ultimately, this test should probably test that
 
156
        #                    extra chmod calls aren't being made
 
157
        try:
 
158
            transport = get_transport(self.get_url())
 
159
            transport.put('my-lock', StringIO(''))
 
160
            lockable = LockableFiles(transport, 'my-lock', TransportLock)
 
161
            self.assertNotEqual(None, lockable._dir_mode)
 
162
            self.assertNotEqual(None, lockable._file_mode)
 
163
 
 
164
            LockableFiles._set_dir_mode = False
 
165
            transport = get_transport('.')
 
166
            lockable = LockableFiles(transport, 'my-lock', TransportLock)
 
167
            self.assertEqual(None, lockable._dir_mode)
 
168
            self.assertNotEqual(None, lockable._file_mode)
 
169
 
 
170
            LockableFiles._set_file_mode = False
 
171
            transport = get_transport('.')
 
172
            lockable = LockableFiles(transport, 'my-lock', TransportLock)
 
173
            self.assertEqual(None, lockable._dir_mode)
 
174
            self.assertEqual(None, lockable._file_mode)
 
175
 
 
176
            LockableFiles._set_dir_mode = True
 
177
            transport = get_transport('.')
 
178
            lockable = LockableFiles(transport, 'my-lock', TransportLock)
 
179
            self.assertNotEqual(None, lockable._dir_mode)
 
180
            self.assertEqual(None, lockable._file_mode)
 
181
 
 
182
            LockableFiles._set_file_mode = True
 
183
            transport = get_transport('.')
 
184
            lockable = LockableFiles(transport, 'my-lock', TransportLock)
 
185
            self.assertNotEqual(None, lockable._dir_mode)
 
186
            self.assertNotEqual(None, lockable._file_mode)
 
187
        finally:
 
188
            LockableFiles._set_dir_mode = True
 
189
            LockableFiles._set_file_mode = True
 
190
 
173
191
 
174
192
class TestSftpPermissions(TestCaseWithSFTPServer):
175
193
 
198
216
        b_local = t.branch
199
217
        self.assertEqualMode(0755, b_local.control_files._dir_mode)
200
218
        self.assertEqualMode(0644, b_local.control_files._file_mode)
201
 
        self.assertEqualMode(0755, b_local.bzrdir._get_dir_mode())
202
 
        self.assertEqualMode(0644, b_local.bzrdir._get_file_mode())
203
219
 
204
220
        os.mkdir('sftp')
205
221
        sftp_url = self.get_url('sftp')
213
229
        b_sftp = Branch.open(sftp_url)
214
230
        self.assertEqualMode(0755, b_sftp.control_files._dir_mode)
215
231
        self.assertEqualMode(0644, b_sftp.control_files._file_mode)
216
 
        self.assertEqualMode(0755, b_sftp.bzrdir._get_dir_mode())
217
 
        self.assertEqualMode(0644, b_sftp.bzrdir._get_file_mode())
218
232
 
219
233
        open('local/a', 'wb').write('foo2\n')
220
234
        t_local.commit('foo2')
236
250
        b_sftp = Branch.open(sftp_url)
237
251
        self.assertEqualMode(0775, b_sftp.control_files._dir_mode)
238
252
        self.assertEqualMode(0664, b_sftp.control_files._file_mode)
239
 
        self.assertEqualMode(0775, b_sftp.bzrdir._get_dir_mode())
240
 
        self.assertEqualMode(0664, b_sftp.bzrdir._get_file_mode())
241
253
 
242
254
        open('local/a', 'wb').write('foo3\n')
243
255
        t_local.commit('foo3')
264
276
            self.assertTransportMode(t, 'a', 0666 &~umask)
265
277
 
266
278
            # but Transport overrides umask
267
 
            t.put_bytes('b', 'txt', mode=0666)
 
279
            t.put('b', 'txt', mode=0666)
268
280
            self.assertTransportMode(t, 'b', 0666)
269
281
 
270
 
            t._get_sftp().mkdir('c', mode=0777)
 
282
            t._sftp.mkdir('c', mode=0777)
271
283
            self.assertTransportMode(t, 'c', 0777 &~umask)
272
284
 
273
285
            t.mkdir('d', mode=0777)