← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_missing.py

  • Committer: John Arbash Meinel
  • Date: 2008-09-26 22:14:42 UTC
  • mto: This revision was merged to the branch mainline in revision 3747.
  • Revision ID: john@arbash-meinel.com-20080926221442-3r67j99sr9rwe9w0
Make message optional, don't check the memory flag directly.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_missing.py
17
17
import os
18
18
 
19
19
 
20
 
from bzrlib.builtins import merge
 
20
from bzrlib import (
 
21
    missing,
 
22
    tests,
 
23
    )
21
24
from bzrlib.missing import (
22
 
    find_unmerged,
23
 
    iter_log_data,
24
25
    iter_log_revisions,
25
26
    )
26
 
from bzrlib.symbol_versioning import zero_seventeen
27
27
from bzrlib.tests import TestCaseWithTransport
28
28
from bzrlib.workingtree import WorkingTree
29
29
 
30
30
 
31
31
class TestMissing(TestCaseWithTransport):
32
32
 
 
33
    def assertUnmerged(self, expected, source, target, restrict='all'):
 
34
        unmerged = missing.find_unmerged(source, target, restrict=restrict)
 
35
        self.assertEqual(expected, unmerged)
 
36
            
33
37
    def test_find_unmerged(self):
34
38
        original_tree = self.make_branch_and_tree('original')
35
39
        original = original_tree.branch
37
41
        puller = puller_tree.branch
38
42
        merger_tree = self.make_branch_and_tree('merger')
39
43
        merger = merger_tree.branch
40
 
        self.assertEqual(find_unmerged(original, puller), ([], []))
 
44
        self.assertUnmerged(([], []), original, puller)
41
45
        original_tree.commit('a', rev_id='a')
42
 
        self.assertEqual(find_unmerged(original, puller), ([(1, u'a')], []))
 
46
        self.assertUnmerged(([(1, 'a')], []), original, puller)
43
47
        puller_tree.pull(original)
44
 
        self.assertEqual(find_unmerged(original, puller), ([], []))
 
48
        self.assertUnmerged(([], []), original, puller)
45
49
        merger_tree.pull(original)
46
50
        original_tree.commit('b', rev_id='b')
47
51
        original_tree.commit('c', rev_id='c')
48
 
        self.assertEqual(find_unmerged(original, puller), ([(2, u'b'), 
49
 
                                                            (3, u'c')], []))
 
52
        self.assertUnmerged(([(2, 'b'), (3, 'c')], []),
 
53
                            original, puller)
50
54
 
51
55
        puller_tree.pull(original)
52
 
        self.assertEqual(find_unmerged(original, puller), ([], []))
53
 
        self.assertEqual(find_unmerged(original, merger), ([(2, u'b'), 
54
 
                                                            (3, u'c')], []))
55
 
        merge(['original', -1], [None, None], this_dir='merger')
56
 
        self.assertEqual(find_unmerged(original, merger), ([(2, u'b'), 
57
 
                                                            (3, u'c')], []))
 
56
        self.assertUnmerged(([], []), original, puller)
 
57
        self.assertUnmerged(([(2, 'b'), (3, 'c')], []),
 
58
                            original, merger)
 
59
        merger_tree.merge_from_branch(original)
 
60
        self.assertUnmerged(([(2, 'b'), (3, 'c')], []),
 
61
                            original, merger)
58
62
        merger_tree.commit('d', rev_id='d')
59
 
        self.assertEqual(find_unmerged(original, merger), ([], [(2, 'd')]))
60
 
 
61
 
    def test_iter_log_data(self):
62
 
        base_tree = self.make_branch_and_tree('base')
63
 
        self.build_tree(['base/a'])
64
 
        base_tree.add(['a'], ['a-id'])
65
 
        base_tree.commit('add a', rev_id='b-1')
66
 
 
67
 
        child_tree = base_tree.bzrdir.sprout('child').open_workingtree()
68
 
 
69
 
        self.build_tree(['child/b'])
70
 
        child_tree.add(['b'], ['b-id'])
71
 
        child_tree.commit('adding b', rev_id='c-2')
72
 
 
73
 
        child_tree.remove(['a'])
74
 
        child_tree.commit('removing a', rev_id='c-3')
75
 
 
76
 
        self.build_tree_contents([('child/b', 'new contents for b\n')])
77
 
        child_tree.commit('modifying b', rev_id='c-4')
78
 
 
79
 
        child_tree.rename_one('b', 'c')
80
 
        child_tree.commit('rename b=>c', rev_id='c-5')
81
 
 
82
 
        base_extra, child_extra = find_unmerged(base_tree.branch,
83
 
                                                child_tree.branch)
84
 
        results = list(self.applyDeprecated(zero_seventeen, iter_log_data,
85
 
                            base_extra,
86
 
                            base_tree.branch.repository,
87
 
                            verbose=True))
88
 
        self.assertEqual([], results)
89
 
 
90
 
        results = list(self.applyDeprecated(zero_seventeen, iter_log_data,
91
 
                            child_extra,
92
 
                            child_tree.branch.repository,
93
 
                            verbose=True))
94
 
        self.assertEqual(4, len(results))
95
 
 
96
 
        r0,r1,r2,r3 = results
97
 
 
98
 
        self.assertEqual((2, 'c-2'), (r0[0], r0[1].revision_id))
99
 
        self.assertEqual((3, 'c-3'), (r1[0], r1[1].revision_id))
100
 
        self.assertEqual((4, 'c-4'), (r2[0], r2[1].revision_id))
101
 
        self.assertEqual((5, 'c-5'), (r3[0], r3[1].revision_id))
102
 
 
103
 
        delta0 = r0[2]
104
 
        self.assertNotEqual(None, delta0)
105
 
        self.assertEqual([('b', 'b-id', 'file')], delta0.added)
106
 
        self.assertEqual([], delta0.removed)
107
 
        self.assertEqual([], delta0.renamed)
108
 
        self.assertEqual([], delta0.modified)
109
 
 
110
 
        delta1 = r1[2]
111
 
        self.assertNotEqual(None, delta1)
112
 
        self.assertEqual([], delta1.added)
113
 
        self.assertEqual([('a', 'a-id', 'file')], delta1.removed)
114
 
        self.assertEqual([], delta1.renamed)
115
 
        self.assertEqual([], delta1.modified)
116
 
 
117
 
        delta2 = r2[2]
118
 
        self.assertNotEqual(None, delta2)
119
 
        self.assertEqual([], delta2.added)
120
 
        self.assertEqual([], delta2.removed)
121
 
        self.assertEqual([], delta2.renamed)
122
 
        self.assertEqual([('b', 'b-id', 'file', True, False)], delta2.modified)
123
 
 
124
 
        delta3 = r3[2]
125
 
        self.assertNotEqual(None, delta3)
126
 
        self.assertEqual([], delta3.added)
127
 
        self.assertEqual([], delta3.removed)
128
 
        self.assertEqual([('b', 'c', 'b-id', 'file', False, False)],
129
 
                         delta3.renamed)
130
 
        self.assertEqual([], delta3.modified)
 
63
        self.assertUnmerged(([], [(2, 'd')]), original, merger)
131
64
 
132
65
    def test_iter_log_revisions(self):
133
66
        base_tree = self.make_branch_and_tree('base')
150
83
        child_tree.rename_one('b', 'c')
151
84
        child_tree.commit('rename b=>c', rev_id='c-5')
152
85
 
153
 
        base_extra, child_extra = find_unmerged(base_tree.branch,
154
 
                                                child_tree.branch)
 
86
        base_extra, child_extra = missing.find_unmerged(base_tree.branch,
 
87
                                                        child_tree.branch)
155
88
        results = list(iter_log_revisions(base_extra, 
156
89
                            base_tree.branch.repository,
157
90
                            verbose=True))
197
130
        self.assertEqual([('b', 'c', 'b-id', 'file', False, False)],
198
131
                         delta3.renamed)
199
132
        self.assertEqual([], delta3.modified)
 
133
 
 
134
 
 
135
class TestFindUnmerged(tests.TestCaseWithTransport):
 
136
 
 
137
    def assertUnmerged(self, local, remote, local_branch, remote_branch,
 
138
                       restrict):
 
139
        """Check the output of find_unmerged_mainline_revisions"""
 
140
        local_extra, remote_extra = missing.find_unmerged(
 
141
                                        local_branch, remote_branch, restrict)
 
142
        self.assertEqual(local, local_extra)
 
143
        self.assertEqual(remote, remote_extra)
 
144
 
 
145
    def test_same_branch(self):
 
146
        tree = self.make_branch_and_tree('tree')
 
147
        rev1 = tree.commit('one')
 
148
        tree.lock_read()
 
149
        self.addCleanup(tree.unlock)
 
150
        self.assertUnmerged([], [], tree.branch, tree.branch, 'all')
 
151
 
 
152
    def test_one_ahead(self):
 
153
        tree = self.make_branch_and_tree('tree')
 
154
        rev1 = tree.commit('one')
 
155
        tree2 = tree.bzrdir.sprout('tree2').open_workingtree()
 
156
        rev2 = tree2.commit('two')
 
157
        tree.lock_read()
 
158
        self.addCleanup(tree.unlock)
 
159
        tree2.lock_read()
 
160
        self.addCleanup(tree2.unlock)
 
161
        self.assertUnmerged([], [(2, rev2)], tree.branch, tree2.branch, 'all')
 
162
        self.assertUnmerged([(2, rev2)], [], tree2.branch, tree.branch, 'all')
 
163
 
 
164
    def test_restrict(self):
 
165
        tree = self.make_branch_and_tree('tree')
 
166
        rev1 = tree.commit('one')
 
167
        tree2 = tree.bzrdir.sprout('tree2').open_workingtree()
 
168
        rev2 = tree2.commit('two')
 
169
        tree.lock_read()
 
170
        self.addCleanup(tree.unlock)
 
171
        tree2.lock_read()
 
172
        self.addCleanup(tree2.unlock)
 
173
        self.assertUnmerged([], [(2, rev2)], tree.branch, tree2.branch, 'all')
 
174
        self.assertUnmerged([], None, tree.branch, tree2.branch, 'local')
 
175
        self.assertUnmerged(None, [(2, rev2)], tree.branch, tree2.branch,
 
176
                                               'remote')
 
177
 
 
178
    def test_merged(self):
 
179
        tree = self.make_branch_and_tree('tree')
 
180
        rev1 = tree.commit('one')
 
181
        tree2 = tree.bzrdir.sprout('tree2').open_workingtree()
 
182
        rev2 = tree2.commit('two')
 
183
        rev3 = tree2.commit('three')
 
184
        tree.merge_from_branch(tree2.branch)
 
185
        rev4 = tree.commit('four')
 
186
 
 
187
        tree.lock_read()
 
188
        self.addCleanup(tree.unlock)
 
189
        tree2.lock_read()
 
190
        self.addCleanup(tree2.unlock)
 
191
 
 
192
        self.assertUnmerged([(2, rev4)], [], tree.branch, tree2.branch, 'all')