~bzr-pqm/bzr/bzr.dev

# Copyright (C) 2006, 2007 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

"""Tests for interface conformance of 'WorkingTree.remove'"""

import re

from bzrlib.tests.workingtree_implementations import TestCaseWithWorkingTree

from bzrlib import errors, osutils

class TestRemove(TestCaseWithWorkingTree):

    """Tests WorkingTree.remove"""

    files=['a', 'b/', 'b/c', 'd/']

    a = ['a']

    b = ['b']

    b_c = ['b', 'b/c']

    def getTree(self):

        self.makeAndChdirToTestDir()

        tree = self.make_branch_and_tree('.')

        self.build_tree(TestRemove.files)

        return tree

    def test_remove_keep(self):

        """Check that files are unversioned but not deleted."""

        tree = self.getTree()

        tree.add(TestRemove.files)

        self.assertInWorkingTree(TestRemove.files)

        tree.remove(TestRemove.files)

        self.assertNotInWorkingTree(TestRemove.files)

        self.failUnlessExists(TestRemove.files)

    def test_remove_unchanged_files(self):

        """Check that unchanged files are removed and deleted."""

        tree = self.getTree()

        tree.add(TestRemove.files)

        tree.commit("files must not have changes")

        self.assertInWorkingTree(TestRemove.files)

        tree.remove(TestRemove.files, keep_files=False)

        self.assertNotInWorkingTree(TestRemove.files)

        self.failIfExists(TestRemove.files)

    def test_remove_added_files(self):

        """Removal of newly added files must fail."""

        tree = self.getTree()

        tree.add(TestRemove.files)

        self.assertInWorkingTree(TestRemove.files)

        try:

            tree.remove(TestRemove.files, keep_files=False)

            self.fail('Should throw BzrRemoveChangedFilesError')

        except errors.BzrRemoveChangedFilesError, e:

            self.assertTrue(re.match('Can\'t remove changed or unknown files:'

                '.*added:.*a.*b.*b/c.*d.*',

                str(e), re.DOTALL))

        self.assertInWorkingTree(TestRemove.files)

        self.failUnlessExists(TestRemove.files)

    def test_remove_changed_file(self):

        """Removal of a changed files must fail."""

        tree = self.getTree()

        tree.add(TestRemove.a)

        tree.commit("make sure a is versioned")

        f = file('a', 'wb')

        f.write("some other new content!")

        f.close()

        self.assertInWorkingTree(TestRemove.a)

        try:

            tree.remove(TestRemove.a, keep_files=False)

            self.fail('Should throw BzrRemoveChangedFilesError')

        except errors.BzrRemoveChangedFilesError, e:

            self.assertTrue(re.match('Can\'t remove changed or unknown files:'

                '.*modified:.*a.*',

                str(e), re.DOTALL))

        self.assertInWorkingTree(TestRemove.a)

        self.failUnlessExists(TestRemove.a)

    def test_remove_deleted_files(self):

        """Check that files are removed if they don't exist any more."""

        tree = self.getTree()

        tree.add(TestRemove.files)

        tree.commit("make sure files are versioned")

        for f in ['b/c', 'b', 'a', 'd']:

            osutils.delete_any(f)

        self.assertInWorkingTree(TestRemove.files)

        self.failIfExists(TestRemove.files)

        tree.remove(TestRemove.files, keep_files=False)

        self.assertNotInWorkingTree(TestRemove.files)

        self.failIfExists(TestRemove.files)

    def test_force_remove_changed_files(self):

        """Check that changed files are removed and deleted when forced."""

        tree = self.getTree()

        tree.add(TestRemove.files)

        self.assertInWorkingTree(TestRemove.files)

        tree.remove(TestRemove.files, keep_files=False, force=True)

        self.assertNotInWorkingTree(TestRemove.files)

        self.failIfExists(TestRemove.files)

    def test_remove_unknown_files(self):

        """Try to delete unknown files."""

        tree = self.getTree()

        try:

            tree.remove(TestRemove.files, keep_files=False)

            self.fail('Should throw BzrRemoveChangedFilesError')

        except errors.BzrRemoveChangedFilesError, e:

            self.assertTrue(re.match('Can\'t remove changed or unknown files:'

                '.*unknown:.*b/c.*b.*a.*d.*',

                str(e), re.DOTALL))

    def test_remove_nonexisting_files(self):

        """Try to delete non-existing files."""

        tree = self.getTree()

        tree.remove([''], keep_files=False)

        try:

            tree.remove(['xyz', 'abc/def'], keep_files=False)

            self.fail('Should throw BzrRemoveChangedFilesError')

        except errors.BzrRemoveChangedFilesError, e:

            self.assertTrue(re.match('Can\'t remove changed or unknown files:'

                '.*unknown:.*xyz.*abc/def.*',

                str(e), re.DOTALL))

    def test_remove_nonempty_directory(self):

        """Unchanged non-empty directories should be deleted."""

        tree = self.getTree()

        tree.add(TestRemove.files)

        tree.commit("make sure b is versioned")

        self.assertInWorkingTree(TestRemove.files)

        self.failUnlessExists(TestRemove.files)

        tree.remove(TestRemove.b, keep_files=False)

        self.assertNotInWorkingTree(TestRemove.b)

        self.failIfExists(TestRemove.b)

    def test_remove_nonempty_directory_with_unknowns(self):

        """Unchanged non-empty directories should be deleted."""

        tree = self.getTree()

        tree.add(TestRemove.files)

        tree.commit("make sure b is versioned")

        self.assertInWorkingTree(TestRemove.files)

        self.failUnlessExists(TestRemove.files)

        f = file('b/my_unknown_file', 'wb')

        f.write("some content!")

        f.close()

        try:

            tree.remove(TestRemove.b, keep_files=False)

            self.fail('Should throw BzrRemoveChangedFilesError')

        except errors.BzrRemoveChangedFilesError, e:

            self.assertTrue(re.match('Can\'t remove changed or unknown files:'

                '.*unknown:.*b/my_unknown_file.*',

                str(e), re.DOTALL))

        self.assertInWorkingTree(TestRemove.b)

        self.failUnlessExists(TestRemove.b)

    def test_force_remove_nonempty_directory(self):

        tree = self.getTree()

        tree.add(TestRemove.files)

        tree.commit("make sure b is versioned")

        self.assertInWorkingTree(TestRemove.files)

        self.failUnlessExists(TestRemove.files)

        tree.remove(TestRemove.b, keep_files=False, force=True)

        self.assertNotInWorkingTree(TestRemove.b_c)

        self.failIfExists(TestRemove.b_c)

    def test_remove_directory_with_changed_file(self):

        """Refuse to delete directories with changed files."""

        tree = self.getTree()

        tree.add(TestRemove.b_c)

        tree.commit("make sure b and c are versioned")

        f = file('b/c', 'wb')

        f.write("some other new content!")

        f.close()

        self.assertInWorkingTree(TestRemove.b_c)

        try:

            tree.remove(TestRemove.b, keep_files=False)

            self.fail('Should throw BzrRemoveChangedFilesError')

        except errors.BzrRemoveChangedFilesError, e:

            self.assertTrue(re.match('Can\'t remove changed or unknown files:'

                '.*modified:.*b/c.*',

                str(e), re.DOTALL))

        self.assertInWorkingTree(TestRemove.b_c)

        self.failUnlessExists(TestRemove.b_c)

        #see if we can force it now..

        tree.remove(TestRemove.b, keep_files=False, force=True)

        self.assertNotInWorkingTree(TestRemove.b_c)

        self.failIfExists(TestRemove.b_c)