~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_permissions.py

• browse files at revision 3696.3.9
• compare with another revision
• download diff
• download tarball
• view history from revision 3696.3.9

• reverse the comparison (3696.3.9 to 1728.1.1)
• stop comparing with revision 1728.1.1

Show diffs side-by-side

added

removed

bzrlib/tests/test_permissions.py

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

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 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

46

47

 

47

48

def chmod_r(base, file_mode, dir_mode):

48

49

    """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)

70

69

    t = get_transport(".")

71

70

    if include_base:

72

71

        test.assertTransportMode(t, base, dir_mode)

73

72

    for root, dirs, files in os.walk(base):

74

73

        for d in dirs:

75

 

            p = os.path.join(root, d)

 

74

            p = '/'.join([urllib.quote(x) for x in root.split('/\\') + [d]])

76

75

            test.assertTransportMode(t, p, dir_mode)

77

76

        for f in files:

78

77

            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

 

        t.add('a')

 

91

        # ensure check_mode_r works with capital-letter file-ids like TREE_ROOT

 

92

        t.add('a', 'CAPS-ID')

92

93

        t.commit('foo')

93

94

 

94

95

        chmod_r('.bzr', 0644, 0755)

103

104

        b = t.branch

104

105

        self.assertEqualMode(0755, b.control_files._dir_mode)

105

106

        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())

106

109

 

107

110

        # Modifying a file shouldn't break the permissions

108

111

        open('a', 'wb').write('foo2\n')

123

126

        b = t.branch

124

127

        self.assertEqualMode(0775, b.control_files._dir_mode)

125

128

        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())

126

131

 

127

132

        open('a', 'wb').write('foo3\n')

128

133

        t.commit('foo3')

133

138

        t.commit('new c')

134

139

        check_mode_r(self, '.bzr', 0664, 0775)

135

140

 

 

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

 

136

153

        # Test the group sticky bit

137

154

        # Recursively update the modes of all files

138

155

        chmod_r('.bzr', 0664, 02775)

141

158

        b = t.branch

142

159

        self.assertEqualMode(02775, b.control_files._dir_mode)

143

160

        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())

144

163

 

145

164

        open('a', 'wb').write('foo4\n')

146

165

        t.commit('foo4')

151

170

        t.commit('new d')

152

171

        check_mode_r(self, '.bzr', 0664, 02775)

153

172

 

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

173

 

192

174

class TestSftpPermissions(TestCaseWithSFTPServer):

193

175

 

216

198

        b_local = t.branch

217

199

        self.assertEqualMode(0755, b_local.control_files._dir_mode)

218

200

        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())

219

203

 

220

204

        os.mkdir('sftp')

221

205

        sftp_url = self.get_url('sftp')

229

213

        b_sftp = Branch.open(sftp_url)

230

214

        self.assertEqualMode(0755, b_sftp.control_files._dir_mode)

231

215

        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())

232

218

 

233

219

        open('local/a', 'wb').write('foo2\n')

234

220

        t_local.commit('foo2')

250

236

        b_sftp = Branch.open(sftp_url)

251

237

        self.assertEqualMode(0775, b_sftp.control_files._dir_mode)

252

238

        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())

253

241

 

254

242

        open('local/a', 'wb').write('foo3\n')

255

243

        t_local.commit('foo3')

276

264

            self.assertTransportMode(t, 'a', 0666 &~umask)

277

265

 

278

266

            # but Transport overrides umask

279

 

            t.put('b', 'txt', mode=0666)

 

267

            t.put_bytes('b', 'txt', mode=0666)

280

268

            self.assertTransportMode(t, 'b', 0666)

281

269

 

282

 

            t._sftp.mkdir('c', mode=0777)

 

270

            t._get_sftp().mkdir('c', mode=0777)

283

271

            self.assertTransportMode(t, 'c', 0777 &~umask)

284

272

 

285

273

            t.mkdir('d', mode=0777)

• Older »

Loggerhead is a web-based interface for Breezy
Version: 2.0.1