/brz/remove-bazaar

# Copyright (C) 2009 Jelmer Vernooij <jelmer@samba.org>

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

from dulwich.objects import (

    Blob,

    Tree,

    )

import os

from bzrlib import (

    knit,

    osutils,

    versionedfile,

    )

from bzrlib.bzrdir import (

    BzrDir,

    )

from bzrlib.inventory import (

    Inventory,

    )

from bzrlib.repository import (

    Repository,

    )

from bzrlib.tests import (

    TestCaseWithTransport,

    )

from bzrlib.transport import (

    get_transport,

    )

from bzrlib.plugins.git import (

    get_rich_root_format,

    )

from bzrlib.plugins.git.fetch import (

    BzrFetchGraphWalker,

    import_git_blob,

    import_git_tree,

    )

from bzrlib.plugins.git.mapping import (

    BzrGitMappingv1,

    default_mapping,

    )

from bzrlib.plugins.git.shamap import (

    DictGitShaMap,

    )

from bzrlib.plugins.git.tests import (

    GitBranchBuilder,

    run_git,

    )

class FetchGraphWalkerTests(TestCaseWithTransport):

    def setUp(self):

        TestCaseWithTransport.setUp(self)

        self.mapping = default_mapping

    def test_empty(self):

        tree = self.make_branch_and_tree("wt")

        graphwalker = BzrFetchGraphWalker(tree.branch.repository, self.mapping)

        self.assertEquals(None, graphwalker.next())

class RepositoryFetchTests:

    def make_git_repo(self, path):

        os.mkdir(path)

        os.chdir(path)

        run_git("init")

        os.chdir("..")

    def clone_git_repo(self, from_url, to_url, revision_id=None):

        oldrepos = self.open_git_repo(from_url)

        dir = BzrDir.create(to_url, get_rich_root_format())

        newrepos = dir.create_repository()

        oldrepos.copy_content_into(newrepos, revision_id=revision_id)

        return newrepos

    def test_empty(self):

        self.make_git_repo("d")

        newrepos = self.clone_git_repo("d", "f")

        self.assertEquals([], newrepos.all_revision_ids())

    def make_onerev_branch(self):

        self.make_git_repo("d")

        os.chdir("d")

        bb = GitBranchBuilder()

        bb.set_file("foobar", "foo\nbar\n", False)

        mark = bb.commit("Somebody <somebody@someorg.org>", "mymsg")

        gitsha = bb.finish()[mark]

        os.chdir("..")

        return "d", gitsha

    def test_single_rev(self):

        path, gitsha = self.make_onerev_branch()

        oldrepo = self.open_git_repo(path)

        newrepo = self.clone_git_repo(path, "f")

        self.assertEquals([oldrepo.get_mapping().revision_id_foreign_to_bzr(gitsha)], newrepo.all_revision_ids())

    def test_single_rev_specific(self):

        path, gitsha = self.make_onerev_branch()

        oldrepo = self.open_git_repo(path)

        revid = oldrepo.get_mapping().revision_id_foreign_to_bzr(gitsha)

        newrepo = self.clone_git_repo(path, "f", revision_id=revid)

        self.assertEquals([revid], newrepo.all_revision_ids())

    def test_incremental(self):

        self.make_git_repo("d")

        os.chdir("d")

        bb = GitBranchBuilder()

        bb.set_file("foobar", "foo\nbar\n", False)

        mark1 = bb.commit("Somebody <somebody@someorg.org>", "mymsg")

        bb.set_file("foobar", "fooll\nbar\n", False)

        mark2 = bb.commit("Somebody <somebody@someorg.org>", "nextmsg")

        marks = bb.finish()

        gitsha1 = marks[mark1]

        gitsha2 = marks[mark2]

        os.chdir("..")

        oldrepo = self.open_git_repo("d")

        revid1 = oldrepo.get_mapping().revision_id_foreign_to_bzr(gitsha1)

        newrepo = self.clone_git_repo("d", "f", revision_id=revid1)

        self.assertEquals([revid1], newrepo.all_revision_ids())

        revid2 = oldrepo.get_mapping().revision_id_foreign_to_bzr(gitsha2)

        newrepo.fetch(oldrepo, revision_id=revid2)

        self.assertEquals(set([revid1, revid2]), set(newrepo.all_revision_ids()))

    def test_executable(self):

        self.make_git_repo("d")

        os.chdir("d")

        bb = GitBranchBuilder()

        bb.set_file("foobar", "foo\nbar\n", True)

        bb.set_file("notexec", "foo\nbar\n", False)

        mark = bb.commit("Somebody <somebody@someorg.org>", "mymsg")

        gitsha = bb.finish()[mark]

        os.chdir("..")

        oldrepo = self.open_git_repo("d")

        newrepo = self.clone_git_repo("d", "f")

        revid = oldrepo.get_mapping().revision_id_foreign_to_bzr(gitsha)

        tree = newrepo.revision_tree(revid)

        self.assertTrue(tree.has_filename("foobar"))

        self.assertEquals(True, tree.inventory[tree.path2id("foobar")].executable)

        self.assertTrue(tree.has_filename("notexec"))

        self.assertEquals(False, tree.inventory[tree.path2id("notexec")].executable)

    def test_non_ascii_characters(self):

        self.make_git_repo("d")

        os.chdir("d")

        bb = GitBranchBuilder()

        bb.set_file(u"foőbar", "foo\nbar\n", False)

        mark = bb.commit("Somebody <somebody@someorg.org>", "mymsg")

        gitsha = bb.finish()[mark]

        os.chdir("..")

        oldrepo = self.open_git_repo("d")

        newrepo = self.clone_git_repo("d", "f")

        revid = oldrepo.get_mapping().revision_id_foreign_to_bzr(gitsha)

        tree = newrepo.revision_tree(revid)

        self.assertTrue(tree.has_filename(u"foőbar"))

class LocalRepositoryFetchTests(RepositoryFetchTests, TestCaseWithTransport):

    def open_git_repo(self, path):

        return Repository.open(path)

class ImportObjects(TestCaseWithTransport):

    def setUp(self):

        super(ImportObjects, self).setUp()

        self._map = DictGitShaMap()

        self._mapping = BzrGitMappingv1()

        factory = knit.make_file_factory(True, versionedfile.PrefixMapper())

        self._texts = factory(self.get_transport('texts'))

    def test_import_blob_simple(self):

        blob = Blob.from_string("bar")

        base_inv = Inventory()

        objs = { "blobname": blob}

        ret = import_git_blob(self._texts, self._mapping, "bla", "blobname", 

            base_inv, None, "somerevid", [], self._map, objs.__getitem__, False,

            False)

        self.assertEquals(set([('bla', 'somerevid')]), self._texts.keys())

        self.assertEquals(self._texts.get_record_stream([('bla', 'somerevid')],

            "unordered", True).next().get_bytes_as("fulltext"), "bar")

        self.assertEquals(1, len(ret)) 

        self.assertEquals(None, ret[0][0])

        self.assertEquals("bla", ret[0][1])

        ie = ret[0][3]

        self.assertEquals(False, ie.executable)

        self.assertEquals("file", ie.kind)

        self.assertEquals("somerevid", ie.revision)

        self.assertEquals(osutils.sha_strings(["bar"]), ie.text_sha1)

    def test_import_tree_empty_root(self):

        base_inv = Inventory(root_id=None)

        tree = Tree()

        tree.serialize()

        ret = import_git_tree(self._texts, self._mapping, "", 

               tree.id, base_inv, 

               None, "somerevid", [], self._map, {tree.id: tree}.__getitem__)

        self.assertEquals(set([("TREE_ROOT", 'somerevid')]), self._texts.keys())

        self.assertEquals(1, len(ret))

        self.assertEquals(None, ret[0][0])

        self.assertEquals("", ret[0][1])

        ie = ret[0][3]

        self.assertEquals(False, ie.executable)

        self.assertEquals("directory", ie.kind)

        self.assertEquals({}, ie.children)

        self.assertEquals("somerevid", ie.revision)

        self.assertEquals(None, ie.text_sha1)

    def test_import_tree_empty(self):

        base_inv = Inventory()

        tree = Tree()

        tree.serialize()

        ret = import_git_tree(self._texts, self._mapping, "bla", 

           tree.id, base_inv, None, "somerevid", [], 

           self._map, { tree.id: tree }.__getitem__)

        self.assertEquals(set([("bla", 'somerevid')]), self._texts.keys())

        self.assertEquals(1, len(ret))

        self.assertEquals(None, ret[0][0])

        self.assertEquals("bla", ret[0][1])

        ie = ret[0][3]

        self.assertEquals("directory", ie.kind)

        self.assertEquals(False, ie.executable)

        self.assertEquals({}, ie.children)

        self.assertEquals("somerevid", ie.revision)

        self.assertEquals(None, ie.text_sha1)

    def test_import_tree_with_file(self):

        base_inv = Inventory()

        blob = Blob.from_string("bar1")

        tree = Tree()

        tree.add(0100600, "foo", blob.id)

        tree.serialize()

        objects = { blob.id: blob, tree.id: tree }

        ret = import_git_tree(self._texts, self._mapping, "bla", tree.id, 

            base_inv, None, "somerevid", [], self._map, objects.__getitem__)

        self.assertEquals(2, len(ret))

        self.assertEquals(None, ret[0][0])

        self.assertEquals("bla", ret[0][1])

        self.assertEquals(None, ret[1][0])

        self.assertEquals("bla/foo", ret[1][1])

        ie = ret[0][3]

        self.assertEquals("directory", ie.kind)

        ie = ret[1][3]

        self.assertEquals("file", ie.kind)

        self.assertEquals("bla/foo", ie.file_id)

        self.assertEquals("somerevid", ie.revision)

        self.assertEquals(osutils.sha_strings(["bar1"]), ie.text_sha1)

        self.assertEquals(False, ie.executable)

    def test_import_tree_with_file_exe(self):

        base_inv = Inventory(root_id=None)

        blob = Blob.from_string("bar")

        tree = Tree()

        tree.add(0100755, "foo", blob.id)

        tree.serialize()

        objects = { blob.id: blob, tree.id: tree }

        ret = import_git_tree(self._texts, self._mapping, "", tree.id, 

            base_inv, None, "somerevid", [], self._map, objects.__getitem__)

        self.assertEquals(2, len(ret))

        self.assertEquals(None, ret[0][0])

        self.assertEquals("", ret[0][1])

        self.assertEquals(None, ret[1][0])

        self.assertEquals("foo", ret[1][1])

        ie = ret[0][3]

        self.assertEquals("directory", ie.kind)

        ie = ret[1][3]

        self.assertEquals("file", ie.kind)

        self.assertEquals(True, ie.executable)