← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_ancestry.py

  • Committer: Martin Packman
  • Date: 2011-12-23 19:38:22 UTC
  • mto: This revision was merged to the branch mainline in revision 6405.
  • Revision ID: martin.packman@canonical.com-20111223193822-hesheea4o8aqwexv
Accept and document passing the medium rather than transport for smart connections

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_ancestry.py
1
 
# Copyright (C) 2005, 2007 Canonical Ltd
 
1
# Copyright (C) 2005-2009, 2011 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
 
 
17
 
 
18
 
from cStringIO import StringIO
19
 
import os
20
 
import sys
21
 
 
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
 
 
17
 
 
18
from bzrlib.branchbuilder import BranchBuilder
22
19
from bzrlib.tests import TestCaseWithMemoryTransport
23
 
from bzrlib.branch import Branch
24
 
from bzrlib.branchbuilder import BranchBuilder
25
 
from bzrlib.revision import is_ancestor
 
20
from bzrlib.tests.matchers import MatchesAncestry
26
21
 
27
22
 
28
23
class TestAncestry(TestCaseWithMemoryTransport):
29
24
 
30
 
    def assertAncestryEqual(self, expected, revision_id, branch):
31
 
        """Assert that the ancestry of revision_id in branch is as expected."""
32
 
        ancestry = branch.repository.get_ancestry(revision_id)
33
 
        self.assertEqual(expected, ancestry)
34
 
 
35
25
    def test_straightline_ancestry(self):
36
26
        """Test ancestry file when just committing."""
37
27
        builder = BranchBuilder(self.get_transport())
38
28
        rev_id_one = builder.build_commit()
39
29
        rev_id_two = builder.build_commit()
40
30
        branch = builder.get_branch()
41
 
        self.assertAncestryEqual([None, rev_id_one, rev_id_two],
42
 
            rev_id_two, branch)
43
 
        self.assertAncestryEqual([None, rev_id_one], rev_id_one, branch)
44
 
 
45
 
    def test_none_is_ancestor_empty_branch(self):
46
 
        branch = self.make_branch('.')
47
 
        self.assertTrue(is_ancestor(None, None, branch))
48
 
 
49
 
    def test_none_is_ancestor_non_empty_branch(self):
50
 
        builder = BranchBuilder(self.get_transport())
51
 
        rev_id = builder.build_commit()
52
 
        branch = builder.get_branch()
53
 
        self.assertTrue(is_ancestor(None, None, branch))
54
 
        self.assertTrue(is_ancestor(rev_id, None, branch))
55
 
        self.assertFalse(is_ancestor(None, rev_id, branch))
56
 
 
 
31
        self.assertThat([rev_id_one, rev_id_two],
 
32
            MatchesAncestry(branch.repository, rev_id_two))
 
33
        self.assertThat([rev_id_one],
 
34
            MatchesAncestry(branch.repository, rev_id_one))
57
35
 
58
36
# TODO: check that ancestry is updated to include indirectly merged revisions