~bzr-pqm/bzr/bzr.dev

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

"""Helper classes for repository implementation tests."""

from bzrlib import (

    inventory,

    revision as _mod_revision,

    )

from bzrlib.repofmt.knitrepo import RepositoryFormatKnit

from bzrlib.tests.repository_implementations import TestCaseWithRepository

from bzrlib.tests import TestNotApplicable

class TestCaseWithBrokenRevisionIndex(TestCaseWithRepository):

    def make_repo_with_extra_ghost_index(self):

        """Make a corrupt repository.

        It will contain one revision, 'revision-id'.  The knit index will claim

        that it has one parent, 'incorrect-parent', but the revision text will

        claim it has no parents.

        Note: only the *cache* of the knit index is corrupted.  Thus the

        corruption will only last while the repository is locked.  For this

        reason, the returned repo is locked.

        """

        if not isinstance(self.repository_format, RepositoryFormatKnit):

            # XXX: Broken revision graphs can happen to weaves too, but they're

            # pretty deprecated.  Ideally these tests should apply to any repo

            # where repo.revision_graph_can_have_wrong_parents() is True, but

            # at the moment we only know how to corrupt knit repos.

            raise TestNotApplicable(

                "%s isn't a knit format" % self.repository_format)

        repo = self.make_repository('broken')

        repo.lock_write()

        repo.start_write_group()

        try:

            inv = inventory.Inventory(revision_id='revision-id')

            inv.root.revision = 'revision-id'

            repo.add_inventory('revision-id', inv, [])

            revision = _mod_revision.Revision('revision-id',

                committer='jrandom@example.com', timestamp=0,

                inventory_sha1='', timezone=0, message='message', parent_ids=[])

            repo.add_revision('revision-id',revision, inv)

        except:

            repo.abort_write_group()

            repo.unlock()

            raise

        else:

            repo.commit_write_group()

            repo.unlock()

        # Change the knit index's record of the parents for 'revision-id' to

        # claim it has a parent, 'incorrect-parent', that doesn't exist in this

        # knit at all.

        repo.lock_write()

        self.addCleanup(repo.unlock)

        rev_knit = repo._get_revision_vf()

        index_cache = rev_knit._index._cache

        cached_index_entry = list(index_cache['revision-id'])

        cached_index_entry[4] = ['incorrect-parent']

        index_cache['revision-id'] = tuple(cached_index_entry)

        return repo