/brz/remove-bazaar

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

#

# 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 GitShaMap."""

import os

from bzrlib.tests import (

    TestCase,

    TestCaseInTempDir,

    UnavailableFeature,

    )

from bzrlib.plugins.git.shamap import (

    DictGitShaMap,

    SqliteGitShaMap,

    TdbGitShaMap,

    )

class TestGitShaMap:

    def test_commit(self):

        self.map.start_write_group()

        self.map._add_entry("5686645d49063c73d35436192dfc9a160c672301",

            "commit", ("myrevid", "cc9462f7f8263ef5adfbeff2fb936bb36b504cba"))

        self.map.commit_write_group()

        self.assertEquals(

            ("commit", ("myrevid", "cc9462f7f8263ef5adfbeff2fb936bb36b504cba")),

            self.map.lookup_git_sha("5686645d49063c73d35436192dfc9a160c672301"))

    def test_lookup_notfound(self):

        self.assertRaises(KeyError,

            self.map.lookup_git_sha, "5686645d49063c73d35436192dfc9a160c672301")

    def test_blob(self):

        thesha = "5686645d49063c73d35436192dfc9a160c672301"

        self.map.start_write_group()

        self.map._add_entry(thesha, "blob", ("myfileid", "myrevid"))

        self.map.commit_write_group()

        self.assertEquals(

            ("blob", ("myfileid", "myrevid")),

            self.map.lookup_git_sha(thesha))

        invshamap = self.map.get_inventory_sha_map("myrevid")

        self.assertEquals(thesha, invshamap.lookup_blob("myfileid"))

    def test_tree(self):

        thesha = "5686645d49063c73d35436192dfc9a160c672301"

        self.map.start_write_group()

        self.map._add_entry(thesha,

            "tree", ("somepath", "myrevid"))

        self.map.commit_write_group()

        self.assertEquals(

            ("tree", ("somepath", "myrevid")),

            self.map.lookup_git_sha(thesha))

    def test_revids(self):

        self.map.start_write_group()

        self.map._add_entry("5686645d49063c73d35436192dfc9a160c672301",

            "commit", ("myrevid", "cc9462f7f8263ef5adfbeff2fb936bb36b504cba"))

        self.map.commit_write_group()

        self.assertEquals(["myrevid"], list(self.map.revids()))

    def test_missing_revisions(self):

        self.map.start_write_group()

        self.map._add_entry("5686645d49063c73d35436192dfc9a160c672301",

            "commit", ("myrevid", "cc9462f7f8263ef5adfbeff2fb936bb36b504cba"))

        self.map.commit_write_group()

        self.assertEquals(set(["lala", "bla"]),

            set(self.map.missing_revisions(["myrevid", "lala", "bla"])))

class DictGitShaMapTests(TestCase,TestGitShaMap):

    def setUp(self):

        TestCase.setUp(self)

        self.map = DictGitShaMap()

class SqliteGitShaMapTests(TestCase,TestGitShaMap):

    def setUp(self):

        TestCase.setUp(self)

        self.map = SqliteGitShaMap()

class TdbGitShaMapTests(TestCaseInTempDir,TestGitShaMap):

    def setUp(self):

        TestCaseInTempDir.setUp(self)

        try:

            self.map = TdbGitShaMap(os.path.join(self.test_dir, 'foo.tdb'))

        except ImportError:

            raise UnavailableFeature("Missing tdb")