/brz/remove-bazaar

# Copyright (C) 2011, 2016 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

"""Tests for InterBranch.fetch."""

from breezy.revision import NULL_REVISION

from breezy.tests.per_interbranch import (

    TestCaseWithInterBranch,

    )

class TestInterBranchFetch(TestCaseWithInterBranch):

    def test_fetch_revisions(self):

        """Test fetch-revision operation."""

        wt = self.make_from_branch_and_tree('b1')

        b1 = wt.branch

        self.build_tree_contents([('b1/foo', 'hello')])

        wt.add(['foo'])

        wt.commit('lala!', rev_id='revision-1', allow_pointless=False)

        b2 = self.make_to_branch('b2')

        b2.fetch(b1)

        # fetch does not update the last revision

        self.assertEqual(NULL_REVISION, b2.last_revision())

        rev = b2.repository.get_revision('revision-1')

        tree = b2.repository.revision_tree('revision-1')

        tree.lock_read()

        self.addCleanup(tree.unlock)

        self.assertEqual(

                tree.get_file_text(tree.path2id('foo'), 'foo'),

                'hello')

    def test_fetch_revisions_limit(self):

        """Test fetch-revision operation."""

        builder = self.make_branch_builder('b1',

            format=self.branch_format_from._matchingbzrdir)

        builder.start_series()

        builder.build_commit(rev_id='revision-1')

        builder.build_commit(rev_id='revision-2')

        builder.build_commit(rev_id='revision-3')

        builder.finish_series()

        b1 = builder.get_branch()

        b2 = self.make_to_branch('b2')

        b2.fetch(b1, limit=1)

        # fetch does not update the last revision

        self.assertEqual(NULL_REVISION, b2.last_revision())

        self.assertEqual(

            {'revision-1'},

            b2.repository.has_revisions(

                ['revision-1', 'revision-2', 'revision-3']))

    def test_fetch_revisions_limit_incremental(self):

        """Test incremental fetch-revision operation with limit."""

        wt = self.make_from_branch_and_tree('b1')

        b1 = wt.branch

        self.build_tree_contents([('b1/foo', 'hello')])

        wt.add(['foo'])

        wt.commit('lala!', rev_id='revision-1', allow_pointless=False)

        b2 = self.make_to_branch('b2')

        b2.fetch(b1, limit=1)

        self.assertEqual(

            {'revision-1'},

            b2.repository.has_revisions(

                ['revision-1', 'revision-2', 'revision-3']))

        wt.commit('hmm', rev_id='revision-2')

        wt.commit('hmmm', rev_id='revision-3')

        b2.fetch(b1, limit=1)

        # fetch does not update the last revision

        self.assertEqual(NULL_REVISION, b2.last_revision())

        self.assertEqual(

            {'revision-1', 'revision-2'},

            b2.repository.has_revisions(

                ['revision-1', 'revision-2', 'revision-3']))