/brz/remove-bazaar

# Copyright (C) 2005 by Canonical Ltd

# -*- coding: utf-8 -*-

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

"""Tests for bzr setting permissions.

Files in the branch control directory (.bzr or .bzr/branch) should inherit

the .bzr directory permissions.

So if the directory is group writable, the files and subdirs should be as well.

"""

# TODO: jam 20051215 Currently the default behavior for 'bzr branch' is just 

#                    defined by the local umask. This isn't terrible, is it

#                    the truly desired behavior?

import os

import sys

import stat

from StringIO import StringIO

from bzrlib.branch import Branch

from bzrlib.bzrdir import BzrDir

from bzrlib.lockable_files import LockableFiles

from bzrlib.tests import TestCaseWithTransport, TestSkipped

from bzrlib.tests.test_sftp_transport import TestCaseWithSFTPServer

from bzrlib.transport import get_transport

from bzrlib.workingtree import WorkingTree

# TODO RBC consolidate the helper methods here and in tests/test_permissions.py

def chmod_r(base, file_mode, dir_mode):

    """Recursively chmod from a base directory"""

    assert os.path.isdir(base)

    os.chmod(base, dir_mode)

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

        for d in dirs:

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

            os.chmod(p, dir_mode)

        for f in files:

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

            os.chmod(p, file_mode)

def check_mode_r(test, base, file_mode, dir_mode, include_base=True):

    """Check that all permissions match

    :param test: The TestCase being run

    :param base: The path to the root directory to check

    :param file_mode: The mode for all files

    :param dir_mode: The mode for all directories

    :param include_base: If false, only check the subdirectories

    """

    assert os.path.isdir(base)

    t = get_transport(".")

    if include_base:

        test.assertTransportMode(t, base, dir_mode)

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

        for d in dirs:

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

            test.assertTransportMode(t, p, dir_mode)

        for f in files:

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

            test.assertTransportMode(t, p, file_mode)

class TestPermissions(TestCaseWithTransport):

    def test_new_branch(self):

        if sys.platform == 'win32':

            raise TestSkipped('chmod has no effect on win32')

        # also, these are BzrBranch format specific things..

        os.mkdir('a')

        mode = stat.S_IMODE(os.stat('a').st_mode)

        t = self.make_branch_and_tree('.')

        b = t.branch

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

        self.assertEqualMode(mode & ~07111, b.control_files._file_mode)

        os.mkdir('b')

        os.chmod('b', 02777)

        b = BzrDir.create('b').create_branch()

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

        self.assertEqualMode(00666, b.control_files._file_mode)

        check_mode_r(self, b.control_files._transport.base, 00666, 02777)

        os.mkdir('c')

        os.chmod('c', 02750)

        b = BzrDir.create('c').create_branch()

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

        self.assertEqualMode(00640, b.control_files._file_mode)

        check_mode_r(self, b.control_files._transport.base, 00640, 02750)

        os.mkdir('d')

        os.chmod('d', 0700)

        b = BzrDir.create('d').create_branch()

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

        self.assertEqualMode(0600, b.control_files._file_mode)

        check_mode_r(self, b.control_files._transport.base, 00600, 00700)