← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_permissions.py

merge from aaron - fixes bare excepts, adds ancestor namespace

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_permissions.py
1
 
# Copyright (C) 2005 by Canonical Ltd
2
 
# -*- coding: utf-8 -*-
3
 
 
4
 
# This program is free software; you can redistribute it and/or modify
5
 
# it under the terms of the GNU General Public License as published by
6
 
# the Free Software Foundation; either version 2 of the License, or
7
 
# (at your option) any later version.
8
 
 
9
 
# This program is distributed in the hope that it will be useful,
10
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
 
# GNU General Public License for more details.
13
 
 
14
 
# You should have received a copy of the GNU General Public License
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
17
 
 
18
 
 
19
 
"""Tests for bzr setting permissions.
20
 
 
21
 
Files which are created underneath .bzr/ should inherit its permissions.
22
 
So if the directory is group writable, the files and subdirs should be as well.
23
 
 
24
 
In the future, when we have Repository/Branch/Checkout information, the
25
 
permissions should be inherited individually, rather than all be the same.
26
 
 
27
 
TODO: jam 20051215 There are no tests for ftp yet, because we have no ftp server
28
 
TODO: jam 20051215 Currently the default behavior for 'bzr branch' is just 
29
 
                   defined by the local umask. This isn't terrible, is it
30
 
                   the truly desired behavior?
31
 
"""
32
 
 
33
 
import os
34
 
import sys
35
 
import stat
36
 
 
37
 
from bzrlib.branch import Branch
38
 
from bzrlib.tests import TestCaseInTempDir, TestSkipped
39
 
from bzrlib.tests.test_sftp_transport import TestCaseWithSFTPServer
40
 
from bzrlib.tests.test_transport import check_mode
41
 
 
42
 
 
43
 
def chmod_r(base, file_mode, dir_mode):
44
 
    """Recursively chmod from a base directory"""
45
 
    assert os.path.isdir(base)
46
 
    os.chmod(base, dir_mode)
47
 
    for root, dirs, files in os.walk(base):
48
 
        for d in dirs:
49
 
            p = os.path.join(root, d)
50
 
            os.chmod(p, dir_mode)
51
 
        for f in files:
52
 
            p = os.path.join(root, f)
53
 
            os.chmod(p, file_mode)
54
 
 
55
 
 
56
 
def check_mode_r(test, base, file_mode, dir_mode, include_base=True):
57
 
    """Check that all permissions match
58
 
 
59
 
    :param test: The TestCase being run
60
 
    :param base: The path to the root directory to check
61
 
    :param file_mode: The mode for all files
62
 
    :param dir_mode: The mode for all directories
63
 
    :param include_base: If false, only check the subdirectories
64
 
    """
65
 
    assert os.path.isdir(base)
66
 
    if include_base:
67
 
        check_mode(test, base, dir_mode)
68
 
    for root, dirs, files in os.walk(base):
69
 
        for d in dirs:
70
 
            p = os.path.join(root, d)
71
 
            check_mode(test, p, dir_mode)
72
 
        for f in files:
73
 
            p = os.path.join(root, f)
74
 
            check_mode(test, p, file_mode)
75
 
 
76
 
 
77
 
def assertEqualMode(test, mode, mode_test):
78
 
    test.assertEqual(mode, mode_test,
79
 
                     'mode mismatch %o != %o' % (mode, mode_test))
80
 
 
81
 
 
82
 
class TestPermissions(TestCaseInTempDir):
83
 
 
84
 
    def test_new_files(self):
85
 
        if sys.platform == 'win32':
86
 
            raise TestSkipped('chmod has no effect on win32')
87
 
 
88
 
        b = Branch.initialize(u'.')
89
 
        t = b.working_tree()
90
 
        open('a', 'wb').write('foo\n')
91
 
        t.add('a')
92
 
        t.commit('foo')
93
 
 
94
 
        # Delete them because we are modifying the filesystem underneath them
95
 
        del b, t 
96
 
        chmod_r('.bzr', 0644, 0755)
97
 
        check_mode_r(self, '.bzr', 0644, 0755)
98
 
 
99
 
        b = Branch.open('.')
100
 
        t = b.working_tree()
101
 
        assertEqualMode(self, 0755, b.control_files._dir_mode)
102
 
        assertEqualMode(self, 0644, b.control_files._file_mode)
103
 
 
104
 
        # Modifying a file shouldn't break the permissions
105
 
        open('a', 'wb').write('foo2\n')
106
 
        t.commit('foo2')
107
 
        # The mode should be maintained after commit
108
 
        check_mode_r(self, '.bzr', 0644, 0755)
109
 
 
110
 
        # Adding a new file should maintain the permissions
111
 
        open('b', 'wb').write('new b\n')
112
 
        t.add('b')
113
 
        t.commit('new b')
114
 
        check_mode_r(self, '.bzr', 0644, 0755)
115
 
 
116
 
        del b, t
117
 
        # Recursively update the modes of all files
118
 
        chmod_r('.bzr', 0664, 0775)
119
 
        check_mode_r(self, '.bzr', 0664, 0775)
120
 
        b = Branch.open('.')
121
 
        t = b.working_tree()
122
 
        assertEqualMode(self, 0775, b.control_files._dir_mode)
123
 
        assertEqualMode(self, 0664, b.control_files._file_mode)
124
 
 
125
 
        open('a', 'wb').write('foo3\n')
126
 
        t.commit('foo3')
127
 
        check_mode_r(self, '.bzr', 0664, 0775)
128
 
 
129
 
        open('c', 'wb').write('new c\n')
130
 
        t.add('c')
131
 
        t.commit('new c')
132
 
        check_mode_r(self, '.bzr', 0664, 0775)
133
 
 
134
 
        # Test the group sticky bit
135
 
        del b, t
136
 
        # Recursively update the modes of all files
137
 
        chmod_r('.bzr', 0664, 02775)
138
 
        check_mode_r(self, '.bzr', 0664, 02775)
139
 
        b = Branch.open('.')
140
 
        t = b.working_tree()
141
 
        assertEqualMode(self, 02775, b.control_files._dir_mode)
142
 
        assertEqualMode(self, 0664, b.control_files._file_mode)
143
 
 
144
 
        open('a', 'wb').write('foo4\n')
145
 
        t.commit('foo4')
146
 
        check_mode_r(self, '.bzr', 0664, 02775)
147
 
 
148
 
        open('d', 'wb').write('new d\n')
149
 
        t.add('d')
150
 
        t.commit('new d')
151
 
        check_mode_r(self, '.bzr', 0664, 02775)
152
 
 
153
 
    def test_disable_set_mode(self):
154
 
        # TODO: jam 20051215 Ultimately, this test should probably test that
155
 
        #                    extra chmod calls aren't being made
156
 
        import bzrlib.lockable_files
157
 
        try:
158
 
            b = Branch.initialize(u'.')
159
 
            self.assertNotEqual(None, b.control_files._dir_mode)
160
 
            self.assertNotEqual(None, b.control_files._file_mode)
161
 
 
162
 
            bzrlib.lockable_files.LockableFiles._set_dir_mode = False
163
 
            b = Branch.open(u'.')
164
 
            self.assertEqual(None, b.control_files._dir_mode)
165
 
            self.assertNotEqual(None, b.control_files._file_mode)
166
 
 
167
 
            bzrlib.lockable_files.LockableFiles._set_file_mode = False
168
 
            b = Branch.open(u'.')
169
 
            self.assertEqual(None, b.control_files._dir_mode)
170
 
            self.assertEqual(None, b.control_files._file_mode)
171
 
 
172
 
            bzrlib.lockable_files.LockableFiles._set_dir_mode = True
173
 
            b = Branch.open(u'.')
174
 
            self.assertNotEqual(None, b.control_files._dir_mode)
175
 
            self.assertEqual(None, b.control_files._file_mode)
176
 
 
177
 
            bzrlib.lockable_files.LockableFiles._set_file_mode = True
178
 
            b = Branch.open(u'.')
179
 
            self.assertNotEqual(None, b.control_files._dir_mode)
180
 
            self.assertNotEqual(None, b.control_files._file_mode)
181
 
        finally:
182
 
            bzrlib.lockable_files.LockableFiles._set_dir_mode = True
183
 
            bzrlib.lockable_files.LockableFiles._set_file_mode = True
184
 
 
185
 
    def test_new_branch(self):
186
 
        if sys.platform == 'win32':
187
 
            raise TestSkipped('chmod has no effect on win32')
188
 
 
189
 
        os.mkdir('a')
190
 
        mode = stat.S_IMODE(os.stat('a').st_mode)
191
 
        b = Branch.initialize('a')
192
 
        assertEqualMode(self, mode, b.control_files._dir_mode)
193
 
        assertEqualMode(self, mode & ~07111, b.control_files._file_mode)
194
 
 
195
 
        os.mkdir('b')
196
 
        os.chmod('b', 02777)
197
 
        b = Branch.initialize('b')
198
 
        assertEqualMode(self, 02777, b.control_files._dir_mode)
199
 
        assertEqualMode(self, 00666, b.control_files._file_mode)
200
 
        check_mode_r(self, 'b/.bzr', 00666, 02777)
201
 
 
202
 
        os.mkdir('c')
203
 
        os.chmod('c', 02750)
204
 
        b = Branch.initialize('c')
205
 
        assertEqualMode(self, 02750, b.control_files._dir_mode)
206
 
        assertEqualMode(self, 00640, b.control_files._file_mode)
207
 
        check_mode_r(self, 'c/.bzr', 00640, 02750)
208
 
 
209
 
        os.mkdir('d')
210
 
        os.chmod('d', 0700)
211
 
        b = Branch.initialize('d')
212
 
        assertEqualMode(self, 0700, b.control_files._dir_mode)
213
 
        assertEqualMode(self, 0600, b.control_files._file_mode)
214
 
        check_mode_r(self, 'd/.bzr', 00600, 0700)
215
 
 
216
 
 
217
 
class TestSftpPermissions(TestCaseWithSFTPServer):
218
 
 
219
 
    def test_new_files(self):
220
 
        if sys.platform == 'win32':
221
 
            raise TestSkipped('chmod has no effect on win32')
222
 
        # Though it would be nice to test that SFTP to a server
223
 
        # which does support chmod has the right effect
224
 
 
225
 
        from bzrlib.transport.sftp import SFTPTransport
226
 
 
227
 
        # We don't actually use it directly, we just want to
228
 
        # keep the connection open, since StubSFTPServer only
229
 
        # allows 1 connection
230
 
        self.delayed_setup()
231
 
        _transport = SFTPTransport(self._sftp_url)
232
 
 
233
 
        os.mkdir('local')
234
 
        b_local = Branch.initialize(u'local')
235
 
        t_local = b_local.working_tree()
236
 
        open('local/a', 'wb').write('foo\n')
237
 
        t_local.add('a')
238
 
        t_local.commit('foo')
239
 
 
240
 
        # Delete them because we are modifying the filesystem underneath them
241
 
        del b_local, t_local 
242
 
        chmod_r('local/.bzr', 0644, 0755)
243
 
        check_mode_r(self, 'local/.bzr', 0644, 0755)
244
 
 
245
 
        b_local = Branch.open(u'local')
246
 
        t_local = b_local.working_tree()
247
 
        assertEqualMode(self, 0755, b_local.control_files._dir_mode)
248
 
        assertEqualMode(self, 0644, b_local.control_files._file_mode)
249
 
 
250
 
        os.mkdir('sftp')
251
 
        # Why does self._sftp_url end with a slash????
252
 
        sftp_url = self._sftp_url + 'sftp'
253
 
        b_sftp = Branch.initialize(sftp_url)
254
 
 
255
 
        b_sftp.pull(b_local)
256
 
        del b_sftp
257
 
        chmod_r('sftp/.bzr', 0644, 0755)
258
 
        check_mode_r(self, 'sftp/.bzr', 0644, 0755)
259
 
 
260
 
        b_sftp = Branch.open(sftp_url)
261
 
        assertEqualMode(self, 0755, b_sftp.control_files._dir_mode)
262
 
        assertEqualMode(self, 0644, b_sftp.control_files._file_mode)
263
 
 
264
 
        open('local/a', 'wb').write('foo2\n')
265
 
        t_local.commit('foo2')
266
 
        b_sftp.pull(b_local)
267
 
        # The mode should be maintained after commit
268
 
        check_mode_r(self, 'sftp/.bzr', 0644, 0755)
269
 
 
270
 
        open('local/b', 'wb').write('new b\n')
271
 
        t_local.add('b')
272
 
        t_local.commit('new b')
273
 
        b_sftp.pull(b_local)
274
 
        check_mode_r(self, 'sftp/.bzr', 0644, 0755)
275
 
 
276
 
        del b_sftp
277
 
        # Recursively update the modes of all files
278
 
        chmod_r('sftp/.bzr', 0664, 0775)
279
 
        check_mode_r(self, 'sftp/.bzr', 0664, 0775)
280
 
 
281
 
        b_sftp = Branch.open(sftp_url)
282
 
        assertEqualMode(self, 0775, b_sftp.control_files._dir_mode)
283
 
        assertEqualMode(self, 0664, b_sftp.control_files._file_mode)
284
 
 
285
 
        open('local/a', 'wb').write('foo3\n')
286
 
        t_local.commit('foo3')
287
 
        b_sftp.pull(b_local)
288
 
        check_mode_r(self, 'sftp/.bzr', 0664, 0775)
289
 
 
290
 
        open('local/c', 'wb').write('new c\n')
291
 
        t_local.add('c')
292
 
        t_local.commit('new c')
293
 
        b_sftp.pull(b_local)
294
 
        check_mode_r(self, 'sftp/.bzr', 0664, 0775)
295
 
 
296
 
    def test_sftp_server_modes(self):
297
 
        if sys.platform == 'win32':
298
 
            raise TestSkipped('chmod has no effect on win32')
299
 
 
300
 
        umask = 0022
301
 
        original_umask = os.umask(umask)
302
 
 
303
 
        try:
304
 
            from bzrlib.transport.sftp import SFTPTransport
305
 
            self.delayed_setup()
306
 
            t = SFTPTransport(self._sftp_url)
307
 
            # Direct access should be masked by umask
308
 
            t._sftp_open_exclusive('a', mode=0666).write('foo\n')
309
 
            check_mode(self, 'a', 0666 &~umask)
310
 
 
311
 
            # but Transport overrides umask
312
 
            t.put('b', 'txt', mode=0666)
313
 
            check_mode(self, 'b', 0666)
314
 
 
315
 
            t._sftp.mkdir('c', mode=0777)
316
 
            check_mode(self, 'c', 0777 &~umask)
317
 
 
318
 
            t.mkdir('d', mode=0777)
319
 
            check_mode(self, 'd', 0777)
320
 
        finally:
321
 
            os.umask(original_umask)
322
 
 
323