/brz/remove-bazaar

# Copyright (C) 2008 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

import time

from bzrlib import (

    branch,

    tests,

    )

from bzrlib.plugins.fastimport import (

    commands,

    errors,

    )

from bzrlib.plugins.fastimport.processors import (

    generic_processor,

    )

class TestCaseForGenericProcessor(tests.TestCaseWithTransport):

    def get_handler(self):

        branch = self.make_branch('.')

        handler = generic_processor.GenericProcessor(branch.bzrdir)

        return handler, branch

    # FIXME: [] as a default is bad, as it is mutable, but I want

    # to use None to mean "don't check this".

    def assertChanges(self, branch, revno, expected_added=[],

            expected_removed=[], expected_modified=[],

            expected_renamed=[]):

        """Check the changes introduced in a revision of a branch.

        This method checks that a revision introduces expected changes.

        The required changes are passed in as a list, where

        each entry contains the needed information about the change.

        If you do not wish to assert anything about a particular

        category then pass None instead.

        branch: The branch.

        revno: revision number of revision to check.

        expected_added: a list of (filename,) tuples that must have

            been added in the delta.

        expected_removed: a list of (filename,) tuples that must have

            been removed in the delta.

        expected_modified: a list of (filename,) tuples that must have

            been modified in the delta.

        expected_renamed: a list of (old_path, new_path) tuples that

            must have been renamed in the delta.

        :return: revtree1, revtree2

        """

        repo = branch.repository

        revtree1 = repo.revision_tree(branch.get_rev_id(revno - 1))

        revtree2 = repo.revision_tree(branch.get_rev_id(revno))

        changes = revtree2.changes_from(revtree1)

        self.check_changes(changes, expected_added, expected_removed,

            expected_modified, expected_renamed)

        return revtree1, revtree2

    def check_changes(self, changes, expected_added=[],

            expected_removed=[], expected_modified=[],

            expected_renamed=[]):

        """Check the changes in a TreeDelta

        This method checks that the TreeDelta contains the expected

        modifications between the two trees that were used to generate

        it. The required changes are passed in as a list, where

        each entry contains the needed information about the change.

        If you do not wish to assert anything about a particular

        category then pass None instead.

        changes: The TreeDelta to check.

        expected_added: a list of (filename,) tuples that must have

            been added in the delta.

        expected_removed: a list of (filename,) tuples that must have

            been removed in the delta.

        expected_modified: a list of (filename,) tuples that must have

            been modified in the delta.

        expected_renamed: a list of (old_path, new_path) tuples that

            must have been renamed in the delta.

        """

        renamed = changes.renamed

        added = changes.added

        removed = changes.removed

        modified = changes.modified

        if expected_renamed is not None:

            self.assertEquals(len(renamed), len(expected_renamed),

                "%s is renamed, expected %s" % (renamed, expected_renamed))

            renamed_files = [(item[0], item[1]) for item in renamed]

            for expected_renamed_entry in expected_renamed:

                self.assertTrue(expected_renamed_entry in renamed_files,

                    "%s is not renamed, %s are" % (str(expected_renamed_entry),

                        renamed_files))

        if expected_added is not None:

            self.assertEquals(len(added), len(expected_added),

                "%s is added" % str(added))

            added_files = [(item[0],) for item in added]

            for expected_added_entry in expected_added:

                self.assertTrue(expected_added_entry in added_files,

                    "%s is not added, %s are" % (str(expected_added_entry),

                        added_files))

        if expected_removed is not None:

            self.assertEquals(len(removed), len(expected_removed),

                "%s is removed" % str(removed))

            removed_files = [(item[0],) for item in removed]

            for expected_removed_entry in expected_removed:

                self.assertTrue(expected_removed_entry in removed_files,

                    "%s is not removed, %s are" % (str(expected_removed_entry),

                        removed_files))

        if expected_modified is not None:

            self.assertEquals(len(modified), len(expected_modified),

                "%s is modified" % str(modified))

            modified_files = [(item[0],) for item in modified]

            for expected_modified_entry in expected_modified:

                self.assertTrue(expected_modified_entry in modified_files,

                    "%s is not modified, %s are" % (str(expected_modified_entry),

                        modified_files))

    def assertContent(self, branch, tree, path, content):

        file_id = tree.inventory.path2id(path)

        branch.lock_read()

        self.addCleanup(branch.unlock)

        self.assertEqual(tree.get_file_text(file_id), content)

    def assertSymlinkTarget(self, branch, tree, path, target):

        file_id = tree.inventory.path2id(path)

        branch.lock_read()

        self.addCleanup(branch.unlock)

        self.assertEqual(tree.get_symlink_target(file_id), target)

    def assertRevisionRoot(self, revtree, path):

        self.assertEqual(revtree.get_revision_id(),

                         revtree.inventory.root.children[path].revision)

class TestModify(TestCaseForGenericProcessor):

    def file_command_iter(self, path, kind='file'):

        def command_list():

            author = ['', 'bugs@a.com', time.time(), time.timezone]

            committer = ['', 'elmer@a.com', time.time(), time.timezone]

            def files_one():

                yield commands.FileModifyCommand(path, kind, False,

                        None, "aaa")

            yield commands.CommitCommand('head', '1', author,

                committer, "commit 1", None, [], files_one)

            def files_two():

                yield commands.FileModifyCommand(path, kind, False,

                        None, "bbb")

            yield commands.CommitCommand('head', '2', author,

                committer, "commit 2", ":1", [], files_two)

        return command_list

    def test_modify_file_in_root(self):

        handler, branch = self.get_handler()

        path = 'a'

        handler.process(self.file_command_iter(path))

        revtree0, revtree1 = self.assertChanges(branch, 1,

            expected_added=[(path,)])

        revtree1, revtree2 = self.assertChanges(branch, 2,

            expected_modified=[(path,)])

        self.assertContent(branch, revtree1, path, "aaa")

        self.assertContent(branch, revtree2, path, "bbb")

        self.assertRevisionRoot(revtree1, path)

        self.assertRevisionRoot(revtree2, path)

    def test_modify_file_in_subdir(self):

        handler, branch = self.get_handler()

        path = 'a/a'

        handler.process(self.file_command_iter(path))

        revtree0, revtree1 = self.assertChanges(branch, 1,

            expected_added=[('a',), (path,)])

        revtree1, revtree2 = self.assertChanges(branch, 2,

            expected_modified=[(path,)])

        self.assertContent(branch, revtree1, path, "aaa")

        self.assertContent(branch, revtree2, path, "bbb")

    def test_modify_symlink_in_root(self):

        handler, branch = self.get_handler()

        path = 'a'

        handler.process(self.file_command_iter(path, kind='symlink'))

        revtree1, revtree2 = self.assertChanges(branch, 2,

            expected_modified=[(path,)])

        self.assertSymlinkTarget(branch, revtree1, path, "aaa")

        self.assertSymlinkTarget(branch, revtree2, path, "bbb")

        self.assertRevisionRoot(revtree1, path)

        self.assertRevisionRoot(revtree2, path)

    def test_modify_symlink_in_subdir(self):

        handler, branch = self.get_handler()

        path = 'a/a'

        handler.process(self.file_command_iter(path, kind='symlink'))

        revtree0, revtree1 = self.assertChanges(branch, 1,

            expected_added=[('a',), (path,)])

        revtree1, revtree2 = self.assertChanges(branch, 2,

            expected_modified=[(path,)])

        self.assertSymlinkTarget(branch, revtree1, path, "aaa")

        self.assertSymlinkTarget(branch, revtree2, path, "bbb")

class TestRename(TestCaseForGenericProcessor):

    def get_command_iter(self, old_path, new_path):

        def command_list():

            author = ['', 'bugs@a.com', time.time(), time.timezone]

            committer = ['', 'elmer@a.com', time.time(), time.timezone]

            def files_one():

                yield commands.FileModifyCommand(old_path, 'file', False,

                        None, "aaa")

            yield commands.CommitCommand('head', '1', author,

                committer, "commit 1", None, [], files_one)

            def files_two():

                yield commands.FileRenameCommand(old_path, new_path)

            yield commands.CommitCommand('head', '2', author,

                committer, "commit 2", ":1", [], files_two)

        return command_list

    def test_rename_in_root(self):

        handler, branch = self.get_handler()

        old_path = 'a'

        new_path = 'b'

        handler.process(self.get_command_iter(old_path, new_path))

        revtree1, revtree2 = self.assertChanges(branch, 2,

            expected_renamed=[(old_path, new_path)])

        self.assertRevisionRoot(revtree1, old_path)

        self.assertRevisionRoot(revtree2, new_path)

    def test_rename_in_subdir(self):

        handler, branch = self.get_handler()

        old_path = 'a/a'

        new_path = 'a/b'

        handler.process(self.get_command_iter(old_path, new_path))

        self.assertChanges(branch, 2, expected_renamed=[(old_path, new_path)])

    def test_move_to_new_dir(self):

        handler, branch = self.get_handler()

        old_path = 'a/a'

        new_path = 'b/a'

        handler.process(self.get_command_iter(old_path, new_path))

        self.assertChanges(branch, 2, expected_renamed=[(old_path, new_path)],

            expected_added=[('b',)])

class TestCopy(TestCaseForGenericProcessor):

    def file_command_iter(self, src_path, dest_path, kind='file'):

        def command_list():

            author = ['', 'bugs@a.com', time.time(), time.timezone]

            committer = ['', 'elmer@a.com', time.time(), time.timezone]

            def files_one():

                yield commands.FileModifyCommand(src_path, kind, False,

                        None, "aaa")

            yield commands.CommitCommand('head', '1', author,

                committer, "commit 1", None, [], files_one)

            def files_two():

                yield commands.FileCopyCommand(src_path, dest_path)

            yield commands.CommitCommand('head', '2', author,

                committer, "commit 2", ":1", [], files_two)

        return command_list

    def test_copy_file_in_root(self):

        handler, branch = self.get_handler()

        src_path = 'a'

        dest_path = 'b'

        handler.process(self.file_command_iter(src_path, dest_path))

        revtree1, revtree2 = self.assertChanges(branch, 2,

            expected_added=[(dest_path,)])

        self.assertContent(branch, revtree1, src_path, "aaa")

        self.assertContent(branch, revtree2, src_path, "aaa")

        self.assertContent(branch, revtree2, dest_path, "aaa")

        self.assertRevisionRoot(revtree1, src_path)

        self.assertRevisionRoot(revtree2, dest_path)

    def test_copy_file_in_subdir(self):

        handler, branch = self.get_handler()

        src_path = 'a/a'

        dest_path = 'a/b'

        handler.process(self.file_command_iter(src_path, dest_path))

        revtree1, revtree2 = self.assertChanges(branch, 2,

            expected_added=[(dest_path,)])

        self.assertContent(branch, revtree1, src_path, "aaa")

        self.assertContent(branch, revtree2, src_path, "aaa")

        self.assertContent(branch, revtree2, dest_path, "aaa")

    def test_copy_file_to_new_dir(self):

        handler, branch = self.get_handler()

        src_path = 'a/a'

        dest_path = 'b/a'

        handler.process(self.file_command_iter(src_path, dest_path))

        revtree1, revtree2 = self.assertChanges(branch, 2,

            expected_added=[('b',), (dest_path,)])

        self.assertContent(branch, revtree1, src_path, "aaa")

        self.assertContent(branch, revtree2, src_path, "aaa")

        self.assertContent(branch, revtree2, dest_path, "aaa")

    def test_copy_symlink_in_root(self):

        handler, branch = self.get_handler()

        src_path = 'a'

        dest_path = 'b'

        handler.process(self.file_command_iter(src_path, dest_path, 'symlink'))

        revtree1, revtree2 = self.assertChanges(branch, 2,

            expected_added=[(dest_path,)])

        self.assertSymlinkTarget(branch, revtree1, src_path, "aaa")

        self.assertSymlinkTarget(branch, revtree2, src_path, "aaa")

        self.assertSymlinkTarget(branch, revtree2, dest_path, "aaa")

        self.assertRevisionRoot(revtree1, src_path)

        self.assertRevisionRoot(revtree2, dest_path)

    def test_copy_symlink_in_subdir(self):

        handler, branch = self.get_handler()

        src_path = 'a/a'

        dest_path = 'a/b'

        handler.process(self.file_command_iter(src_path, dest_path, 'symlink'))

        revtree1, revtree2 = self.assertChanges(branch, 2,

            expected_added=[(dest_path,)])

        self.assertSymlinkTarget(branch, revtree1, src_path, "aaa")

        self.assertSymlinkTarget(branch, revtree2, src_path, "aaa")

        self.assertSymlinkTarget(branch, revtree2, dest_path, "aaa")

    def test_copy_symlink_to_new_dir(self):

        handler, branch = self.get_handler()

        src_path = 'a/a'

        dest_path = 'b/a'

        handler.process(self.file_command_iter(src_path, dest_path, 'symlink'))

        revtree1, revtree2 = self.assertChanges(branch, 2,

            expected_added=[('b',), (dest_path,)])

        self.assertSymlinkTarget(branch, revtree1, src_path, "aaa")

        self.assertSymlinkTarget(branch, revtree2, src_path, "aaa")

        self.assertSymlinkTarget(branch, revtree2, dest_path, "aaa")

class TestFileKinds(TestCaseForGenericProcessor):

    def get_command_iter(self, path, kind, content):

        def command_list():

            committer = ['', 'elmer@a.com', time.time(), time.timezone]

            def files_one():

                yield commands.FileModifyCommand(path, kind, False,

                        None, content)

            yield commands.CommitCommand('head', '1', None,

                committer, "commit 1", None, [], files_one)

        return command_list

    def test_import_plainfile(self):

        handler, branch = self.get_handler()

        handler.process(self.get_command_iter('foo', 'file', 'aaa'))

    def test_import_symlink(self):

        handler, branch = self.get_handler()

        handler.process(self.get_command_iter('foo', 'symlink', 'bar'))