~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/blackbox/test_ancestry.py

  • Committer: John Arbash Meinel
  • Date: 2007-05-04 18:59:36 UTC
  • mto: This revision was merged to the branch mainline in revision 2643.
  • Revision ID: john@arbash-meinel.com-20070504185936-1mjdoqmtz74xe5mg
A C implementation of _fields_to_entry_0_parents drops the time from 400ms to 330ms for a 21k-entry tree

Show diffs side-by-side

added added

removed removed

Lines of Context:
16
16
 
17
17
import os
18
18
 
 
19
from bzrlib.builtins import merge
19
20
from bzrlib.tests import TestCaseWithTransport
20
21
from bzrlib.workingtree import WorkingTree
21
22
from bzrlib.branch import Branch
25
26
 
26
27
    def _build_branches(self):
27
28
        a_wt = self.make_branch_and_tree('A')
28
 
        self.build_tree_contents([('A/foo', '1111\n')])
 
29
        open('A/foo', 'wb').write('1111\n')
29
30
        a_wt.add('foo')
30
31
        a_wt.commit('added foo',rev_id='A1')
31
 
 
32
 
        b_wt = a_wt.bzrdir.sprout('B').open_workingtree()
33
 
        self.build_tree_contents([('B/foo', '1111\n22\n')])
 
32
        self.run_bzr_captured(['branch', 'A', 'B'])
 
33
        b_wt = WorkingTree.open('B')
 
34
        open('B/foo','wb').write('1111\n22\n')
34
35
        b_wt.commit('modified B/foo',rev_id='B1')
35
 
 
36
 
        self.build_tree_contents([('A/foo', '000\n1111\n')])
 
36
        open('A/foo', 'wb').write('000\n1111\n')
37
37
        a_wt.commit('modified A/foo',rev_id='A2')
38
 
 
39
 
        a_wt.merge_from_branch(b_wt.branch, b_wt.last_revision(),
40
 
            b_wt.branch.get_rev_id(1))
 
38
        merge(['B',-1],['B',1],this_dir='A')
41
39
        a_wt.commit('merged B into A',rev_id='A3')
42
 
        return a_wt, b_wt
43
40
 
44
41
    def _check_ancestry(self, location='', result=None):
45
 
        out = self.run_bzr(['ancestry', location])[0]
 
42
        out = self.capture('ancestry ' + location)
46
43
        if result is None:
47
44
            result = "A1\nB1\nA2\nA3\n"
48
45
        self.assertEqualDiff(out, result)
61
58
    def test_ancestry_with_repo_branch(self):
62
59
        """Tests 'ancestry' command with a location that is a
63
60
        repository branch."""
64
 
        a_tree = self._build_branches()[0]
65
 
 
66
 
        self.make_repository('repo', shared=True)
67
 
 
68
 
        a_tree.bzrdir.sprout('repo/A')
 
61
        self._build_branches()
 
62
        self.run_bzr('init-repo', 'repo')
 
63
        self.run_bzr('branch', 'A', 'repo/A')
69
64
        self._check_ancestry('repo/A')
70
65
 
71
66
    def test_ancestry_with_checkout(self):
72
67
        """Tests 'ancestry' command with a location that is a
73
68
        checkout of a repository branch."""
74
 
        a_tree = self._build_branches()[0]
75
 
        self.make_repository('repo', shared=True)
76
 
        repo_branch = a_tree.bzrdir.sprout('repo/A').open_branch()
77
 
        repo_branch.create_checkout('A-checkout')
 
69
        self._build_branches()
 
70
        self.run_bzr('init-repo', 'repo')
 
71
        self.run_bzr('branch', 'A', 'repo/A')
 
72
        self.run_bzr('checkout', 'repo/A', 'A-checkout')
78
73
        self._check_ancestry('A-checkout')
79
74
 
80
75
    def test_ancestry_with_lightweight_checkout(self):
81
76
        """Tests 'ancestry' command with a location that is a
82
77
        lightweight checkout of a repository branch."""
83
 
        a_tree = self._build_branches()[0]
84
 
        self.make_repository('repo', shared=True)
85
 
        repo_branch = a_tree.bzrdir.sprout('repo/A').open_branch()
86
 
        repo_branch.create_checkout('A-checkout', lightweight=True)
 
78
        self._build_branches()
 
79
        self.run_bzr('init-repo', 'repo')
 
80
        self.run_bzr('branch', 'A', 'repo/A')
 
81
        self.run_bzr('checkout', '--lightweight', 'repo/A', 'A-checkout')
87
82
        self._check_ancestry('A-checkout')
88
83
 
89
84
    def test_ancestry_with_truncated_checkout(self):
90
85
        """Tests 'ancestry' command with a location that is a
91
86
        checkout of a repository branch with a shortened revision history."""
92
 
        a_tree = self._build_branches()[0]
93
 
        self.make_repository('repo', shared=True)
94
 
        repo_branch = a_tree.bzrdir.sprout('repo/A').open_branch()
95
 
        repo_branch.create_checkout('A-checkout',
96
 
                                    revision_id=repo_branch.get_rev_id(2))
 
87
        self._build_branches()
 
88
        self.run_bzr('init-repo', 'repo')
 
89
        self.run_bzr('branch', 'A', 'repo/A')
 
90
        self.run_bzr('checkout', '-r', '2', 'repo/A', 'A-checkout')
97
91
        self._check_ancestry('A-checkout', "A1\nA2\n")
98
92
 
99
93
    def test_ancestry_with_truncated_lightweight_checkout(self):
100
94
        """Tests 'ancestry' command with a location that is a lightweight
101
95
        checkout of a repository branch with a shortened revision history."""
102
 
        a_tree = self._build_branches()[0]
103
 
        self.make_repository('repo', shared=True)
104
 
        repo_branch = a_tree.bzrdir.sprout('repo/A').open_branch()
105
 
        repo_branch.create_checkout('A-checkout',
106
 
                                    revision_id=repo_branch.get_rev_id(2),
107
 
                                    lightweight=True)
 
96
        self._build_branches()
 
97
        self.run_bzr('init-repo', 'repo')
 
98
        self.run_bzr('branch', 'A', 'repo/A')
 
99
        self.run_bzr('checkout', '-r', '2', '--lightweight',
 
100
                'repo/A', 'A-checkout')
108
101
        self._check_ancestry('A-checkout', "A1\nA2\n")