/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_permissions.py

  • Committer: John Arbash Meinel
  • Date: 2006-04-25 15:05:42 UTC
  • mfrom: (1185.85.85 bzr-encoding)
  • mto: This revision was merged to the branch mainline in revision 1752.
  • Revision ID: john@arbash-meinel.com-20060425150542-c7b518dca9928691
[merge] the old bzr-encoding changes, reparenting them on bzr.dev

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_permissions.py
1
 
# Copyright (C) 2005-2011 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
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  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
import stat
 
36
from StringIO import StringIO
35
37
 
36
 
from breezy import urlutils
37
 
from breezy.branch import Branch
38
 
from breezy.controldir import ControlDir
39
 
from breezy.tests import TestCaseWithTransport, TestSkipped
40
 
from breezy.tests.test_sftp_transport import TestCaseWithSFTPServer
41
 
from breezy.workingtree import WorkingTree
 
38
from bzrlib.branch import Branch
 
39
from bzrlib.bzrdir import BzrDir
 
40
from bzrlib.lockable_files import LockableFiles, TransportLock
 
41
from bzrlib.tests import TestCaseWithTransport, TestSkipped
 
42
from bzrlib.tests.test_sftp_transport import TestCaseWithSFTPServer
 
43
from bzrlib.transport import get_transport
 
44
from bzrlib.workingtree import WorkingTree
42
45
 
43
46
 
44
47
def chmod_r(base, file_mode, dir_mode):
45
48
    """Recursively chmod from a base directory"""
 
49
    assert os.path.isdir(base)
46
50
    os.chmod(base, dir_mode)
47
51
    for root, dirs, files in os.walk(base):
48
52
        for d in dirs:
62
66
    :param dir_mode: The mode for all directories
63
67
    :param include_base: If false, only check the subdirectories
64
68
    """
65
 
    t = test.get_transport()
 
69
    assert os.path.isdir(base)
 
70
    t = get_transport(".")
66
71
    if include_base:
67
72
        test.assertTransportMode(t, base, dir_mode)
68
73
    for root, dirs, files in os.walk(base):
69
74
        for d in dirs:
70
 
            p = '/'.join([urlutils.quote(x) for x in root.split('/\\') + [d]])
 
75
            p = os.path.join(root, d)
71
76
            test.assertTransportMode(t, p, dir_mode)
72
77
        for f in files:
73
78
            p = os.path.join(root, f)
74
 
            p = '/'.join([urlutils.quote(x) for x in root.split('/\\') + [f]])
75
79
            test.assertTransportMode(t, p, file_mode)
76
80
 
77
81
 
83
87
 
84
88
        t = self.make_branch_and_tree('.')
85
89
        b = t.branch
86
 
        with open('a', 'wb') as f:
87
 
            f.write(b'foo\n')
88
 
        # ensure check_mode_r works with capital-letter file-ids like TREE_ROOT
89
 
        t.add('a', b'CAPS-ID')
 
90
        open('a', 'wb').write('foo\n')
 
91
        t.add('a')
90
92
        t.commit('foo')
91
93
 
92
 
        chmod_r('.bzr', 0o644, 0o755)
93
 
        check_mode_r(self, '.bzr', 0o644, 0o755)
 
94
        chmod_r('.bzr', 0644, 0755)
 
95
        check_mode_r(self, '.bzr', 0644, 0755)
94
96
 
95
97
        # although we are modifying the filesystem
96
98
        # underneath the objects, they are not locked, and thus it must
97
 
        # 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 
98
100
        # mode change in the control bits, which current do not refresh
99
101
        # when a new lock is taken out.
100
102
        t = WorkingTree.open('.')
101
103
        b = t.branch
102
 
        self.assertEqualMode(0o755, b.control_files._dir_mode)
103
 
        self.assertEqualMode(0o644, b.control_files._file_mode)
104
 
        self.assertEqualMode(0o755, b.controldir._get_dir_mode())
105
 
        self.assertEqualMode(0o644, b.controldir._get_file_mode())
 
104
        self.assertEqualMode(0755, b.control_files._dir_mode)
 
105
        self.assertEqualMode(0644, b.control_files._file_mode)
106
106
 
107
107
        # Modifying a file shouldn't break the permissions
108
 
        with open('a', 'wb') as f:
109
 
            f.write(b'foo2\n')
 
108
        open('a', 'wb').write('foo2\n')
110
109
        t.commit('foo2')
111
110
        # The mode should be maintained after commit
112
 
        check_mode_r(self, '.bzr', 0o644, 0o755)
 
111
        check_mode_r(self, '.bzr', 0644, 0755)
113
112
 
114
113
        # Adding a new file should maintain the permissions
115
 
        with open('b', 'wb') as f:
116
 
            f.write(b'new b\n')
 
114
        open('b', 'wb').write('new b\n')
117
115
        t.add('b')
118
116
        t.commit('new b')
119
 
        check_mode_r(self, '.bzr', 0o644, 0o755)
 
117
        check_mode_r(self, '.bzr', 0644, 0755)
120
118
 
121
119
        # Recursively update the modes of all files
122
 
        chmod_r('.bzr', 0o664, 0o775)
123
 
        check_mode_r(self, '.bzr', 0o664, 0o775)
 
120
        chmod_r('.bzr', 0664, 0775)
 
121
        check_mode_r(self, '.bzr', 0664, 0775)
124
122
        t = WorkingTree.open('.')
125
123
        b = t.branch
126
 
        self.assertEqualMode(0o775, b.control_files._dir_mode)
127
 
        self.assertEqualMode(0o664, b.control_files._file_mode)
128
 
        self.assertEqualMode(0o775, b.controldir._get_dir_mode())
129
 
        self.assertEqualMode(0o664, b.controldir._get_file_mode())
 
124
        self.assertEqualMode(0775, b.control_files._dir_mode)
 
125
        self.assertEqualMode(0664, b.control_files._file_mode)
130
126
 
131
 
        with open('a', 'wb') as f:
132
 
            f.write(b'foo3\n')
 
127
        open('a', 'wb').write('foo3\n')
133
128
        t.commit('foo3')
134
 
        check_mode_r(self, '.bzr', 0o664, 0o775)
 
129
        check_mode_r(self, '.bzr', 0664, 0775)
135
130
 
136
 
        with open('c', 'wb') as f:
137
 
            f.write(b'new c\n')
 
131
        open('c', 'wb').write('new c\n')
138
132
        t.add('c')
139
133
        t.commit('new c')
140
 
        check_mode_r(self, '.bzr', 0o664, 0o775)
141
 
 
142
 
    def test_new_files_group_sticky_bit(self):
143
 
        if sys.platform == 'win32':
144
 
            raise TestSkipped('chmod has no effect on win32')
145
 
        elif sys.platform == 'darwin' or 'freebsd' in sys.platform:
146
 
            # FreeBSD-based platforms create temp dirs with the 'wheel' group,
147
 
            # which users are not likely to be in, and this prevents us from
148
 
            # setting the sgid bit
149
 
            os.chown(self.test_dir, os.getuid(), os.getgid())
150
 
 
151
 
        t = self.make_branch_and_tree('.')
152
 
        b = t.branch
 
134
        check_mode_r(self, '.bzr', 0664, 0775)
153
135
 
154
136
        # Test the group sticky bit
155
137
        # Recursively update the modes of all files
156
 
        chmod_r('.bzr', 0o664, 0o2775)
157
 
        check_mode_r(self, '.bzr', 0o664, 0o2775)
 
138
        chmod_r('.bzr', 0664, 02775)
 
139
        check_mode_r(self, '.bzr', 0664, 02775)
158
140
        t = WorkingTree.open('.')
159
141
        b = t.branch
160
 
        self.assertEqualMode(0o2775, b.control_files._dir_mode)
161
 
        self.assertEqualMode(0o664, b.control_files._file_mode)
162
 
        self.assertEqualMode(0o2775, b.controldir._get_dir_mode())
163
 
        self.assertEqualMode(0o664, b.controldir._get_file_mode())
 
142
        self.assertEqualMode(02775, b.control_files._dir_mode)
 
143
        self.assertEqualMode(0664, b.control_files._file_mode)
164
144
 
165
 
        with open('a', 'wb') as f:
166
 
            f.write(b'foo4\n')
 
145
        open('a', 'wb').write('foo4\n')
167
146
        t.commit('foo4')
168
 
        check_mode_r(self, '.bzr', 0o664, 0o2775)
 
147
        check_mode_r(self, '.bzr', 0664, 02775)
169
148
 
170
 
        with open('d', 'wb') as f:
171
 
            f.write(b'new d\n')
 
149
        open('d', 'wb').write('new d\n')
172
150
        t.add('d')
173
151
        t.commit('new d')
174
 
        check_mode_r(self, '.bzr', 0o664, 0o2775)
 
152
        check_mode_r(self, '.bzr', 0664, 02775)
 
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
175
190
 
176
191
 
177
192
class TestSftpPermissions(TestCaseWithSFTPServer):
184
199
 
185
200
        # bodge around for stubsftpserver not letting use connect
186
201
        # more than once
187
 
        _t = self.get_transport()
 
202
        _t = get_transport(self.get_url())
188
203
 
189
204
        os.mkdir('local')
190
205
        t_local = self.make_branch_and_tree('local')
191
206
        b_local = t_local.branch
192
 
        with open('local/a', 'wb') as f:
193
 
            f.write(b'foo\n')
 
207
        open('local/a', 'wb').write('foo\n')
194
208
        t_local.add('a')
195
209
        t_local.commit('foo')
196
210
 
197
211
        # Delete them because we are modifying the filesystem underneath them
198
 
        chmod_r('local/.bzr', 0o644, 0o755)
199
 
        check_mode_r(self, 'local/.bzr', 0o644, 0o755)
 
212
        chmod_r('local/.bzr', 0644, 0755)
 
213
        check_mode_r(self, 'local/.bzr', 0644, 0755)
200
214
 
201
215
        t = WorkingTree.open('local')
202
216
        b_local = t.branch
203
 
        self.assertEqualMode(0o755, b_local.control_files._dir_mode)
204
 
        self.assertEqualMode(0o644, b_local.control_files._file_mode)
205
 
        self.assertEqualMode(0o755, b_local.controldir._get_dir_mode())
206
 
        self.assertEqualMode(0o644, b_local.controldir._get_file_mode())
 
217
        self.assertEqualMode(0755, b_local.control_files._dir_mode)
 
218
        self.assertEqualMode(0644, b_local.control_files._file_mode)
207
219
 
208
220
        os.mkdir('sftp')
209
221
        sftp_url = self.get_url('sftp')
210
 
        b_sftp = ControlDir.create_branch_and_repo(sftp_url)
 
222
        b_sftp = BzrDir.create_branch_and_repo(sftp_url)
211
223
 
212
224
        b_sftp.pull(b_local)
213
225
        del b_sftp
214
 
        chmod_r('sftp/.bzr', 0o644, 0o755)
215
 
        check_mode_r(self, 'sftp/.bzr', 0o644, 0o755)
 
226
        chmod_r('sftp/.bzr', 0644, 0755)
 
227
        check_mode_r(self, 'sftp/.bzr', 0644, 0755)
216
228
 
217
229
        b_sftp = Branch.open(sftp_url)
218
 
        self.assertEqualMode(0o755, b_sftp.control_files._dir_mode)
219
 
        self.assertEqualMode(0o644, b_sftp.control_files._file_mode)
220
 
        self.assertEqualMode(0o755, b_sftp.controldir._get_dir_mode())
221
 
        self.assertEqualMode(0o644, b_sftp.controldir._get_file_mode())
 
230
        self.assertEqualMode(0755, b_sftp.control_files._dir_mode)
 
231
        self.assertEqualMode(0644, b_sftp.control_files._file_mode)
222
232
 
223
 
        with open('local/a', 'wb') as f:
224
 
            f.write(b'foo2\n')
 
233
        open('local/a', 'wb').write('foo2\n')
225
234
        t_local.commit('foo2')
226
235
        b_sftp.pull(b_local)
227
236
        # The mode should be maintained after commit
228
 
        check_mode_r(self, 'sftp/.bzr', 0o644, 0o755)
 
237
        check_mode_r(self, 'sftp/.bzr', 0644, 0755)
229
238
 
230
 
        with open('local/b', 'wb') as f:
231
 
            f.write(b'new b\n')
 
239
        open('local/b', 'wb').write('new b\n')
232
240
        t_local.add('b')
233
241
        t_local.commit('new b')
234
242
        b_sftp.pull(b_local)
235
 
        check_mode_r(self, 'sftp/.bzr', 0o644, 0o755)
 
243
        check_mode_r(self, 'sftp/.bzr', 0644, 0755)
236
244
 
237
245
        del b_sftp
238
246
        # Recursively update the modes of all files
239
 
        chmod_r('sftp/.bzr', 0o664, 0o775)
240
 
        check_mode_r(self, 'sftp/.bzr', 0o664, 0o775)
 
247
        chmod_r('sftp/.bzr', 0664, 0775)
 
248
        check_mode_r(self, 'sftp/.bzr', 0664, 0775)
241
249
 
242
250
        b_sftp = Branch.open(sftp_url)
243
 
        self.assertEqualMode(0o775, b_sftp.control_files._dir_mode)
244
 
        self.assertEqualMode(0o664, b_sftp.control_files._file_mode)
245
 
        self.assertEqualMode(0o775, b_sftp.controldir._get_dir_mode())
246
 
        self.assertEqualMode(0o664, b_sftp.controldir._get_file_mode())
 
251
        self.assertEqualMode(0775, b_sftp.control_files._dir_mode)
 
252
        self.assertEqualMode(0664, b_sftp.control_files._file_mode)
247
253
 
248
 
        with open('local/a', 'wb') as f:
249
 
            f.write(b'foo3\n')
 
254
        open('local/a', 'wb').write('foo3\n')
250
255
        t_local.commit('foo3')
251
256
        b_sftp.pull(b_local)
252
 
        check_mode_r(self, 'sftp/.bzr', 0o664, 0o775)
 
257
        check_mode_r(self, 'sftp/.bzr', 0664, 0775)
253
258
 
254
 
        with open('local/c', 'wb') as f:
255
 
            f.write(b'new c\n')
 
259
        open('local/c', 'wb').write('new c\n')
256
260
        t_local.add('c')
257
261
        t_local.commit('new c')
258
262
        b_sftp.pull(b_local)
259
 
        check_mode_r(self, 'sftp/.bzr', 0o664, 0o775)
 
263
        check_mode_r(self, 'sftp/.bzr', 0664, 0775)
260
264
 
261
265
    def test_sftp_server_modes(self):
262
266
        if sys.platform == 'win32':
263
267
            raise TestSkipped('chmod has no effect on win32')
264
268
 
265
 
        umask = 0o022
 
269
        umask = 0022
266
270
        original_umask = os.umask(umask)
267
271
 
268
272
        try:
269
 
            t = self.get_transport()
 
273
            t = get_transport(self.get_url())
270
274
            # Direct access should be masked by umask
271
 
            with t._sftp_open_exclusive('a', mode=0o666) as f:
272
 
                f.write(b'foo\n')
273
 
            self.assertTransportMode(t, 'a', 0o666 & ~umask)
 
275
            t._sftp_open_exclusive('a', mode=0666).write('foo\n')
 
276
            self.assertTransportMode(t, 'a', 0666 &~umask)
274
277
 
275
278
            # but Transport overrides umask
276
 
            t.put_bytes('b', b'txt', mode=0o666)
277
 
            self.assertTransportMode(t, 'b', 0o666)
278
 
 
279
 
            t._get_sftp().mkdir('c', mode=0o777)
280
 
            self.assertTransportMode(t, 'c', 0o777 & ~umask)
281
 
 
282
 
            t.mkdir('d', mode=0o777)
283
 
            self.assertTransportMode(t, 'd', 0o777)
 
279
            t.put('b', 'txt', mode=0666)
 
280
            self.assertTransportMode(t, 'b', 0666)
 
281
 
 
282
            t._sftp.mkdir('c', mode=0777)
 
283
            self.assertTransportMode(t, 'c', 0777 &~umask)
 
284
 
 
285
            t.mkdir('d', mode=0777)
 
286
            self.assertTransportMode(t, 'd', 0777)
284
287
        finally:
285
288
            os.umask(original_umask)