~bzr-pqm/bzr/bzr.dev

# Copyright (C) 2005-2010 Canonical Ltd

#

# 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

#

"""Tests of the 'bzr clean-tree' command."""

import os

from bzrlib import ignores

from bzrlib.tests import TestCaseWithTransport

class TestBzrTools(TestCaseWithTransport):

    @staticmethod

    def touch(filename):

        my_file = open(filename, 'wb')

        try:

            my_file.write('')

        finally:

            my_file.close()

    def test_clean_tree(self):

        self.run_bzr('init')

        self.run_bzr('ignore *~')

        self.run_bzr('ignore *.pyc')

        self.touch('name')

        self.touch('name~')

        self.assertPathExists('name~')

        self.touch('name.pyc')

        self.run_bzr('clean-tree --force')

        self.assertPathExists('name~')

        self.assertPathDoesNotExist('name')

        self.touch('name')

        self.run_bzr('clean-tree --detritus --force')

        self.assertPathExists('name')

        self.assertPathDoesNotExist('name~')

        self.assertPathExists('name.pyc')

        self.run_bzr('clean-tree --ignored --force')

        self.assertPathExists('name')

        self.assertPathDoesNotExist('name.pyc')

        self.run_bzr('clean-tree --unknown --force')

        self.assertPathDoesNotExist('name')

        self.touch('name')

        self.touch('name~')

        self.touch('name.pyc')

        self.run_bzr('clean-tree --unknown --ignored --force')

        self.assertPathDoesNotExist('name')

        self.assertPathDoesNotExist('name~')

        self.assertPathDoesNotExist('name.pyc')

    def test_clean_tree_nested_bzrdir(self):

        # clean-tree should not blindly delete nested bzrdirs (branches)

        # bug https://bugs.launchpad.net/bzr/+bug/572098

        # so it will play well with scmproj/bzr-externals plugins.

        wt1 = self.make_branch_and_tree('.')

        wt2 = self.make_branch_and_tree('foo')

        wt3 = self.make_branch_and_tree('bar')

        ignores.tree_ignores_add_patterns(wt1, ['./foo'])

        self.run_bzr(['clean-tree', '--unknown', '--force'])

        self.assertPathExists('foo')

        self.assertPathExists('bar')

        self.run_bzr(['clean-tree', '--ignored', '--force'])

        self.assertPathExists('foo')

        self.assertPathExists('bar')

    def test_clean_tree_directory(self):

        """Test --directory option"""

        tree = self.make_branch_and_tree('a')

        self.build_tree(['a/added', 'a/unknown', 'a/ignored'])

        tree.add('added')

        self.run_bzr('clean-tree -d a --unknown --ignored --force')

        self.assertPathDoesNotExist('a/unknown')

        self.assertPathDoesNotExist('a/ignored')

        self.assertPathExists('a/added')