/brz/remove-bazaar

# Copyright (C) 2009-2018 Jelmer Vernooij <jelmer@jelmer.uk>

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

"""Tests from fetching from git into bzr."""

from __future__ import absolute_import

from dulwich.objects import (

    Blob,

    Tag,

    Tree,

    S_IFGITLINK,

    )

from dulwich.repo import (

    Repo as GitRepo,

    )

import os

import stat

import time

from .... import (

    osutils,

    )

from ....bzr import (

    knit,

    versionedfile,

    )

from ....branch import (

    Branch,

    )

from ....controldir import (

    ControlDir,

    )

from ....bzr.inventory import (

    Inventory,

    )

from ....repository import (

    Repository,

    )

from ....tests import (

    TestCaseWithTransport,

    )

from ..fetch import (

    import_git_blob,

    import_git_tree,

    import_git_submodule,

    )

from ..mapping import (

    BzrGitMappingv1,

    DEFAULT_FILE_MODE,

    )

from . import (

    GitBranchBuilder,

    )

class RepositoryFetchTests(object):

    def make_git_repo(self, path):

        os.mkdir(path)

        return GitRepo.init(os.path.abspath(path))

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

        oldrepos = self.open_git_repo(from_url)

        dir = ControlDir.create(to_url)

        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.assertEqual([], newrepos.all_revision_ids())

    def make_onerev_branch(self):

        self.make_git_repo("d")

        os.chdir("d")

        bb = GitBranchBuilder()

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

        mark = bb.commit(b"Somebody <somebody@someorg.org>", b"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")

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

        self.assertEqual([revid], 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.assertEqual([revid], newrepo.all_revision_ids())

    def test_incremental(self):

        self.make_git_repo("d")

        os.chdir("d")

        bb = GitBranchBuilder()

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

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

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

        mark2 = bb.commit(b"Somebody <somebody@someorg.org>", b"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.assertEqual([revid1], newrepo.all_revision_ids())

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

        newrepo.fetch(oldrepo, revision_id=revid2)

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

    def test_dir_becomes_symlink(self):

        self.make_git_repo("d")

        os.chdir("d")

        bb = GitBranchBuilder()

        bb.set_file("mylink/somefile", b"foo\nbar\n", False)

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

        bb.set_symlink("mylink", "target/")

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

        marks = bb.finish()

        gitsha1 = marks[mark1]

        gitsha2 = marks[mark2]

        os.chdir("..")

        oldrepo = self.open_git_repo("d")

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

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

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

        tree1 = newrepo.revision_tree(revid1)

        tree2 = newrepo.revision_tree(revid2)

        self.assertEqual(revid1, tree1.get_file_revision("mylink"))

        self.assertEqual("directory", tree1.kind("mylink"))

        self.assertEqual(None, tree1.get_symlink_target("mylink"))

        self.assertEqual(revid2, tree2.get_file_revision("mylink"))

        self.assertEqual("symlink", tree2.kind("mylink"))

        self.assertEqual("target/", tree2.get_symlink_target("mylink"))

    def test_symlink_becomes_dir(self):

        self.make_git_repo("d")

        os.chdir("d")

        bb = GitBranchBuilder()

        bb.set_symlink("mylink", "target/")

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

        bb.set_file("mylink/somefile", b"foo\nbar\n", False)

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

        marks = bb.finish()

        gitsha1 = marks[mark1]

        gitsha2 = marks[mark2]

        os.chdir("..")

        oldrepo = self.open_git_repo("d")

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

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

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

        tree1 = newrepo.revision_tree(revid1)

        tree2 = newrepo.revision_tree(revid2)

        self.assertEqual(revid1, tree1.get_file_revision("mylink"))

        self.assertEqual("symlink", tree1.kind("mylink"))

        self.assertEqual("target/", tree1.get_symlink_target("mylink"))

        self.assertEqual(revid2, tree2.get_file_revision("mylink"))

        self.assertEqual("directory", tree2.kind("mylink"))

        self.assertEqual(None, tree2.get_symlink_target("mylink"))

    def test_changing_symlink(self):

        self.make_git_repo("d")

        os.chdir("d")

        bb = GitBranchBuilder()

        bb.set_symlink("mylink", "target")

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

        bb.set_symlink("mylink", "target/")

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

        marks = bb.finish()

        gitsha1 = marks[mark1]

        gitsha2 = marks[mark2]

        os.chdir("..")

        oldrepo = self.open_git_repo("d")

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

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

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

        tree1 = newrepo.revision_tree(revid1)

        tree2 = newrepo.revision_tree(revid2)

        self.assertEqual(revid1, tree1.get_file_revision("mylink"))

        self.assertEqual("target", tree1.get_symlink_target("mylink"))

        self.assertEqual(revid2, tree2.get_file_revision("mylink"))

        self.assertEqual("target/", tree2.get_symlink_target("mylink"))

    def test_executable(self):

        self.make_git_repo("d")

        os.chdir("d")

        bb = GitBranchBuilder()

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

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

        mark = bb.commit(b"Somebody <somebody@someorg.org>", b"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.assertEqual(True, tree.is_executable("foobar"))

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

        self.assertEqual(False, tree.is_executable("notexec"))

    def test_becomes_executable(self):

        self.make_git_repo("d")

        os.chdir("d")

        bb = GitBranchBuilder()

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

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

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

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

        gitsha2 = bb.finish()[mark2]

        os.chdir("..")

        oldrepo = self.open_git_repo("d")

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

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

        tree = newrepo.revision_tree(revid)

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

        self.assertEqual(True, tree.is_executable("foobar"))

        self.assertEqual(revid, tree.get_file_revision("foobar"))

    def test_into_stacked_on(self):

        r = self.make_git_repo("d")

        os.chdir("d")

        bb = GitBranchBuilder()

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

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

        gitsha1 = bb.finish()[mark1]

        os.chdir("..")

        stacked_on = self.clone_git_repo("d", "stacked-on")

        oldrepo = Repository.open("d")

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

        self.assertEqual([revid1], stacked_on.all_revision_ids())

        b = stacked_on.controldir.create_branch()

        b.generate_revision_history(revid1)

        self.assertEqual(b.last_revision(), revid1)

        tree = self.make_branch_and_tree("stacked")

        tree.branch.set_stacked_on_url(b.user_url)

        os.chdir("d")

        bb = GitBranchBuilder()

        bb.set_file(u"barbar", b"bar\n", False)

        bb.set_file(u"foo/blie/bla", b"bla\n", False)

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

        gitsha2 = bb.finish()[mark2]

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

        os.chdir("..")

        tree.branch.fetch(Branch.open("d"))

        tree.branch.repository.check()

        self.addCleanup(tree.lock_read().unlock)

        self.assertEqual(

            set([(revid2,)]),

            tree.branch.repository.revisions.without_fallbacks().keys())

        self.assertEqual(

            set([revid1, revid2]),

            set(tree.branch.repository.all_revision_ids()))

    def test_non_ascii_characters(self):

        self.make_git_repo("d")

        os.chdir("d")

        bb = GitBranchBuilder()

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

        mark = bb.commit(b"Somebody <somebody@someorg.org>", b"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"))

    def test_tagged_tree(self):

        r = self.make_git_repo("d")

        os.chdir("d")

        bb = GitBranchBuilder()

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

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

        marks = bb.finish()

        gitsha = marks[mark]

        tag = Tag()

        tag.name = b"sometag"

        tag.tag_time = int(time.time())

        tag.tag_timezone = 0

        tag.tagger = b"Somebody <somebody@example.com>"

        tag.message = b"Created tag pointed at tree"

        tag.object = (Tree, r[gitsha].tree)

        r.object_store.add_object(tag)

        r[b"refs/tags/sometag"] = tag

        os.chdir("..")

        oldrepo = self.open_git_repo("d")

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

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

        self.assertEqual(set([revid]), set(newrepo.all_revision_ids()))

class LocalRepositoryFetchTests(RepositoryFetchTests, TestCaseWithTransport):

    def open_git_repo(self, path):

        return Repository.open(path)

class DummyStoreUpdater(object):

    def add_object(self, obj, ie, path):

        pass

    def finish(self):

        pass

class ImportObjects(TestCaseWithTransport):

    def setUp(self):

        super(ImportObjects, self).setUp()

        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(b"bar")

        base_inv = Inventory()

        objs = { "blobname": blob}

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

            (None, "blobname"), 

            base_inv, None, b"somerevid", [], objs.__getitem__, 

            (None, DEFAULT_FILE_MODE), DummyStoreUpdater(),

            self._mapping.generate_file_id)

        self.assertEqual(set([(b'git:bla', b'somerevid')]), self._texts.keys())

        self.assertEqual(next(self._texts.get_record_stream([(b'git:bla', b'somerevid')],

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

        self.assertEqual(1, len(ret)) 

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

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

        ie = ret[0][3]

        self.assertEqual(False, ie.executable)

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

        self.assertEqual(b"somerevid", ie.revision)

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

    def test_import_tree_empty_root(self):

        base_inv = Inventory(root_id=None)

        tree = Tree()

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

               (None, tree.id), base_inv, 

               None, b"somerevid", [], {tree.id: tree}.__getitem__,

               (None, stat.S_IFDIR), DummyStoreUpdater(),

               self._mapping.generate_file_id)

        self.assertEqual(child_modes, {})

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

        self.assertEqual(1, len(ret))

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

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

        ie = ret[0][3]

        self.assertEqual(False, ie.executable)

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

        self.assertEqual({}, ie.children)

        self.assertEqual(b"somerevid", ie.revision)

        self.assertEqual(None, ie.text_sha1)

    def test_import_tree_empty(self):

        base_inv = Inventory()

        tree = Tree()

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

           (None, tree.id), base_inv, None, b"somerevid", [], 

           { tree.id: tree }.__getitem__,

           (None, stat.S_IFDIR), DummyStoreUpdater(),

           self._mapping.generate_file_id)

        self.assertEqual(child_modes, {})

        self.assertEqual(set([(b"git:bla", b'somerevid')]), self._texts.keys())

        self.assertEqual(1, len(ret))

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

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

        ie = ret[0][3]

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

        self.assertEqual(False, ie.executable)

        self.assertEqual({}, ie.children)

        self.assertEqual(b"somerevid", ie.revision)

        self.assertEqual(None, ie.text_sha1)

    def test_import_tree_with_file(self):

        base_inv = Inventory()

        blob = Blob.from_string(b"bar1")

        tree = Tree()

        tree.add(b"foo", stat.S_IFREG | 0o644, blob.id)

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

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

                (None, tree.id), base_inv, None, b"somerevid", [],

            objects.__getitem__, (None, stat.S_IFDIR), DummyStoreUpdater(),

            self._mapping.generate_file_id)

        self.assertEqual(child_modes, {})

        self.assertEqual(2, len(ret))

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

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

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

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

        ie = ret[0][3]

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

        ie = ret[1][3]

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

        self.assertEqual(b"git:bla/foo", ie.file_id)

        self.assertEqual(b"somerevid", ie.revision)

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

        self.assertEqual(False, ie.executable)

    def test_import_tree_with_unusual_mode_file(self):

        base_inv = Inventory()

        blob = Blob.from_string(b"bar1")

        tree = Tree()

        tree.add(b"foo", stat.S_IFREG | 0o664, blob.id)

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

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

            b"bla", b"bla", (None, tree.id), base_inv, None, b"somerevid", [],

            objects.__getitem__, (None, stat.S_IFDIR), DummyStoreUpdater(),

            self._mapping.generate_file_id)

        self.assertEqual(child_modes, { "bla/foo": stat.S_IFREG | 0o664 })

    def test_import_tree_with_file_exe(self):

        base_inv = Inventory(root_id=None)

        blob = Blob.from_string(b"bar")

        tree = Tree()

        tree.add(b"foo", 0o100755, blob.id)

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

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

                (None, tree.id), base_inv, None, b"somerevid", [],

            objects.__getitem__, (None, stat.S_IFDIR), DummyStoreUpdater(),

            self._mapping.generate_file_id)

        self.assertEqual(child_modes, {})

        self.assertEqual(2, len(ret))

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

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

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

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

        ie = ret[0][3]

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

        ie = ret[1][3]

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

        self.assertEqual(True, ie.executable)

    def test_directory_converted_to_submodule(self):

        base_inv = Inventory()

        base_inv.add_path("foo", "directory")

        base_inv.add_path("foo/bar", "file")

        othertree = Blob.from_string(b"someotherthing")

        blob = Blob.from_string(b"bar")

        tree = Tree()

        tree.add(b"bar", 0o160000, blob.id)

        objects = { tree.id: tree }

        ret, child_modes = import_git_submodule(self._texts, self._mapping, b"foo", b"foo",

                (tree.id, othertree.id), base_inv, base_inv.root.file_id, b"somerevid", [],

                objects.__getitem__, (stat.S_IFDIR | 0o755, S_IFGITLINK), DummyStoreUpdater(),

                self._mapping.generate_file_id)

        self.assertEqual(child_modes, {})

        self.assertEqual(2, len(ret))

        self.assertEqual(ret[0], ("foo/bar", None, base_inv.path2id("foo/bar"), None))

        self.assertEqual(ret[1][:3], ("foo", b"foo", self._mapping.generate_file_id("foo")))

        ie = ret[1][3]

        self.assertEqual(ie.kind, "tree-reference")