~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_permissions.py

  • Committer: Aaron Bentley
  • Date: 2009-03-24 12:15:01 UTC
  • mfrom: (4195 +trunk)
  • mto: This revision was merged to the branch mainline in revision 4196.
  • Revision ID: aaron@aaronbentley.com-20090324121501-rd8zjsdk9fok8mvk
Merge and resolve NEWS conflicts

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005 by Canonical Ltd
 
1
# Copyright (C) 2005 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
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
17
 
18
18
 
19
19
"""Tests for bzr setting permissions.
26
26
"""
27
27
 
28
28
# TODO: jam 20051215 There are no tests for ftp yet, because we have no ftp server
29
 
# TODO: jam 20051215 Currently the default behavior for 'bzr branch' is just 
 
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 StringIO import StringIO
 
36
from cStringIO import StringIO
 
37
import urllib
37
38
 
38
39
from bzrlib.branch import Branch
39
40
from bzrlib.bzrdir import BzrDir
40
 
from bzrlib.lockable_files import LockableFiles, TransportLock
41
41
from bzrlib.tests import TestCaseWithTransport, TestSkipped
42
42
from bzrlib.tests.test_sftp_transport import TestCaseWithSFTPServer
43
43
from bzrlib.transport import get_transport
46
46
 
47
47
def chmod_r(base, file_mode, dir_mode):
48
48
    """Recursively chmod from a base directory"""
49
 
    assert os.path.isdir(base)
50
49
    os.chmod(base, dir_mode)
51
50
    for root, dirs, files in os.walk(base):
52
51
        for d in dirs:
66
65
    :param dir_mode: The mode for all directories
67
66
    :param include_base: If false, only check the subdirectories
68
67
    """
69
 
    assert os.path.isdir(base)
70
68
    t = get_transport(".")
71
69
    if include_base:
72
70
        test.assertTransportMode(t, base, dir_mode)
73
71
    for root, dirs, files in os.walk(base):
74
72
        for d in dirs:
75
 
            p = os.path.join(root, d)
 
73
            p = '/'.join([urllib.quote(x) for x in root.split('/\\') + [d]])
76
74
            test.assertTransportMode(t, p, dir_mode)
77
75
        for f in files:
78
76
            p = os.path.join(root, f)
 
77
            p = '/'.join([urllib.quote(x) for x in root.split('/\\') + [f]])
79
78
            test.assertTransportMode(t, p, file_mode)
80
79
 
81
80
 
88
87
        t = self.make_branch_and_tree('.')
89
88
        b = t.branch
90
89
        open('a', 'wb').write('foo\n')
91
 
        t.add('a')
 
90
        # ensure check_mode_r works with capital-letter file-ids like TREE_ROOT
 
91
        t.add('a', 'CAPS-ID')
92
92
        t.commit('foo')
93
93
 
94
94
        chmod_r('.bzr', 0644, 0755)
96
96
 
97
97
        # although we are modifying the filesystem
98
98
        # underneath the objects, they are not locked, and thus it must
99
 
        # be safe for most operations. But here we want to observe a 
 
99
        # be safe for most operations. But here we want to observe a
100
100
        # mode change in the control bits, which current do not refresh
101
101
        # when a new lock is taken out.
102
102
        t = WorkingTree.open('.')
103
103
        b = t.branch
104
104
        self.assertEqualMode(0755, b.control_files._dir_mode)
105
105
        self.assertEqualMode(0644, b.control_files._file_mode)
 
106
        self.assertEqualMode(0755, b.bzrdir._get_dir_mode())
 
107
        self.assertEqualMode(0644, b.bzrdir._get_file_mode())
106
108
 
107
109
        # Modifying a file shouldn't break the permissions
108
110
        open('a', 'wb').write('foo2\n')
123
125
        b = t.branch
124
126
        self.assertEqualMode(0775, b.control_files._dir_mode)
125
127
        self.assertEqualMode(0664, b.control_files._file_mode)
 
128
        self.assertEqualMode(0775, b.bzrdir._get_dir_mode())
 
129
        self.assertEqualMode(0664, b.bzrdir._get_file_mode())
126
130
 
127
131
        open('a', 'wb').write('foo3\n')
128
132
        t.commit('foo3')
133
137
        t.commit('new c')
134
138
        check_mode_r(self, '.bzr', 0664, 0775)
135
139
 
 
140
    def test_new_files_group_sticky_bit(self):
 
141
        if sys.platform == 'win32':
 
142
            raise TestSkipped('chmod has no effect on win32')
 
143
        elif sys.platform == 'darwin':
 
144
            # OS X creates temp dirs with the 'wheel' group, which users are
 
145
            # not likely to be in, and this prevents us from setting the sgid
 
146
            # bit
 
147
            os.chown(self.test_dir, os.getuid(), os.getgid())
 
148
 
 
149
        t = self.make_branch_and_tree('.')
 
150
        b = t.branch
 
151
 
136
152
        # Test the group sticky bit
137
153
        # Recursively update the modes of all files
138
154
        chmod_r('.bzr', 0664, 02775)
141
157
        b = t.branch
142
158
        self.assertEqualMode(02775, b.control_files._dir_mode)
143
159
        self.assertEqualMode(0664, b.control_files._file_mode)
 
160
        self.assertEqualMode(02775, b.bzrdir._get_dir_mode())
 
161
        self.assertEqualMode(0664, b.bzrdir._get_file_mode())
144
162
 
145
163
        open('a', 'wb').write('foo4\n')
146
164
        t.commit('foo4')
151
169
        t.commit('new d')
152
170
        check_mode_r(self, '.bzr', 0664, 02775)
153
171
 
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
 
 
191
172
 
192
173
class TestSftpPermissions(TestCaseWithSFTPServer):
193
174
 
216
197
        b_local = t.branch
217
198
        self.assertEqualMode(0755, b_local.control_files._dir_mode)
218
199
        self.assertEqualMode(0644, b_local.control_files._file_mode)
 
200
        self.assertEqualMode(0755, b_local.bzrdir._get_dir_mode())
 
201
        self.assertEqualMode(0644, b_local.bzrdir._get_file_mode())
219
202
 
220
203
        os.mkdir('sftp')
221
204
        sftp_url = self.get_url('sftp')
229
212
        b_sftp = Branch.open(sftp_url)
230
213
        self.assertEqualMode(0755, b_sftp.control_files._dir_mode)
231
214
        self.assertEqualMode(0644, b_sftp.control_files._file_mode)
 
215
        self.assertEqualMode(0755, b_sftp.bzrdir._get_dir_mode())
 
216
        self.assertEqualMode(0644, b_sftp.bzrdir._get_file_mode())
232
217
 
233
218
        open('local/a', 'wb').write('foo2\n')
234
219
        t_local.commit('foo2')
250
235
        b_sftp = Branch.open(sftp_url)
251
236
        self.assertEqualMode(0775, b_sftp.control_files._dir_mode)
252
237
        self.assertEqualMode(0664, b_sftp.control_files._file_mode)
 
238
        self.assertEqualMode(0775, b_sftp.bzrdir._get_dir_mode())
 
239
        self.assertEqualMode(0664, b_sftp.bzrdir._get_file_mode())
253
240
 
254
241
        open('local/a', 'wb').write('foo3\n')
255
242
        t_local.commit('foo3')
276
263
            self.assertTransportMode(t, 'a', 0666 &~umask)
277
264
 
278
265
            # but Transport overrides umask
279
 
            t.put('b', 'txt', mode=0666)
 
266
            t.put_bytes('b', 'txt', mode=0666)
280
267
            self.assertTransportMode(t, 'b', 0666)
281
268
 
282
 
            t._sftp.mkdir('c', mode=0777)
 
269
            t._get_sftp().mkdir('c', mode=0777)
283
270
            self.assertTransportMode(t, 'c', 0777 &~umask)
284
271
 
285
272
            t.mkdir('d', mode=0777)