44
40
        merger_tree = self.make_branch_and_tree('merger')
 
45
41
        merger = merger_tree.branch
 
46
42
        self.assertUnmerged(([], []), original, puller)
 
47
 
        original_tree.commit('a', rev_id='a')
 
48
 
        self.assertUnmerged(([('1', 'a')], []), original, puller)
 
 
43
        original_tree.commit('a', rev_id=b'a')
 
 
44
        self.assertUnmerged(([('1', b'a', 0)], []), original, puller)
 
49
45
        puller_tree.pull(original)
 
50
46
        self.assertUnmerged(([], []), original, puller)
 
51
47
        merger_tree.pull(original)
 
52
 
        original_tree.commit('b', rev_id='b')
 
53
 
        original_tree.commit('c', rev_id='c')
 
54
 
        self.assertUnmerged(([('2', 'b'), ('3', 'c')], []),
 
 
48
        original_tree.commit('b', rev_id=b'b')
 
 
49
        original_tree.commit('c', rev_id=b'c')
 
 
50
        self.assertUnmerged(([('2', b'b', 0), ('3', b'c', 0)], []),
 
56
 
        self.assertUnmerged(([('3', 'c'), ('2', 'b')], []),
 
 
52
        self.assertUnmerged(([('3', b'c', 0), ('2', b'b', 0)], []),
 
57
53
                            original, puller, backward=True)
 
59
55
        puller_tree.pull(original)
 
60
56
        self.assertUnmerged(([], []), original, puller)
 
61
 
        self.assertUnmerged(([('2', 'b'), ('3', 'c')], []),
 
 
57
        self.assertUnmerged(([('2', b'b', 0), ('3', b'c', 0)], []),
 
63
59
        merger_tree.merge_from_branch(original)
 
64
 
        self.assertUnmerged(([('2', 'b'), ('3', 'c')], []),
 
 
60
        self.assertUnmerged(([('2', b'b', 0), ('3', b'c', 0)], []),
 
66
 
        merger_tree.commit('d', rev_id='d')
 
67
 
        self.assertUnmerged(([], [('2', 'd')]), original, merger)
 
 
62
        merger_tree.commit('d', rev_id=b'd')
 
 
63
        self.assertUnmerged(([], [('2', b'd', 0)]), original, merger)
 
69
65
    def test_iter_log_revisions(self):
 
70
66
        base_tree = self.make_branch_and_tree('base')
 
71
67
        self.build_tree(['base/a'])
 
72
 
        base_tree.add(['a'], ['a-id'])
 
73
 
        base_tree.commit('add a', rev_id='b-1')
 
 
68
        base_tree.add(['a'], [b'a-id'])
 
 
69
        base_tree.commit('add a', rev_id=b'b-1')
 
75
 
        child_tree = base_tree.bzrdir.sprout('child').open_workingtree()
 
 
71
        child_tree = base_tree.controldir.sprout('child').open_workingtree()
 
77
73
        self.build_tree(['child/b'])
 
78
 
        child_tree.add(['b'], ['b-id'])
 
79
 
        child_tree.commit('adding b', rev_id='c-2')
 
 
74
        child_tree.add(['b'], [b'b-id'])
 
 
75
        child_tree.commit('adding b', rev_id=b'c-2')
 
81
77
        child_tree.remove(['a'])
 
82
 
        child_tree.commit('removing a', rev_id='c-3')
 
 
78
        child_tree.commit('removing a', rev_id=b'c-3')
 
84
 
        self.build_tree_contents([('child/b', 'new contents for b\n')])
 
85
 
        child_tree.commit('modifying b', rev_id='c-4')
 
 
80
        self.build_tree_contents([('child/b', b'new contents for b\n')])
 
 
81
        child_tree.commit('modifying b', rev_id=b'c-4')
 
87
83
        child_tree.rename_one('b', 'c')
 
88
 
        child_tree.commit('rename b=>c', rev_id='c-5')
 
 
84
        child_tree.commit('rename b=>c', rev_id=b'c-5')
 
90
86
        base_extra, child_extra = missing.find_unmerged(base_tree.branch,
 
92
88
        results = list(iter_log_revisions(base_extra,
 
93
 
                            base_tree.branch.repository,
 
 
89
                                          base_tree.branch.repository,
 
95
91
        self.assertEqual([], results)
 
97
93
        results = list(iter_log_revisions(child_extra,
 
98
 
                            child_tree.branch.repository,
 
 
94
                                          child_tree.branch.repository,
 
100
96
        self.assertEqual(4, len(results))
 
102
 
        r0,r1,r2,r3 = results
 
 
98
        r0, r1, r2, r3 = results
 
104
 
        self.assertEqual([('2', 'c-2'), ('3', 'c-3'),
 
105
 
                          ('4', 'c-4'), ('5', 'c-5'),],
 
 
100
        self.assertEqual([('2', b'c-2'), ('3', b'c-3'),
 
 
101
                          ('4', b'c-4'), ('5', b'c-5'), ],
 
106
102
                         [(r.revno, r.rev.revision_id) for r in results])
 
108
104
        delta0 = r0.delta
 
109
105
        self.assertNotEqual(None, delta0)
 
110
 
        self.assertEqual([('b', 'b-id', 'file')], delta0.added)
 
 
106
        self.assertEqual([('b', b'b-id', 'file')], delta0.added)
 
111
107
        self.assertEqual([], delta0.removed)
 
112
108
        self.assertEqual([], delta0.renamed)
 
113
109
        self.assertEqual([], delta0.modified)
 
 
156
153
        self.addCleanup(tree.unlock)
 
157
154
        self.assertUnmerged([], [], tree.branch, tree.branch)
 
158
155
        self.assertUnmerged([], [], tree.branch, tree.branch,
 
159
 
            local_revid_range=(rev1, rev1))
 
 
156
                            local_revid_range=(rev1, rev1))
 
161
158
    def test_one_ahead(self):
 
162
159
        tree = self.make_branch_and_tree('tree')
 
163
 
        rev1 = tree.commit('one')
 
164
 
        tree2 = tree.bzrdir.sprout('tree2').open_workingtree()
 
 
161
        tree2 = tree.controldir.sprout('tree2').open_workingtree()
 
165
162
        rev2 = tree2.commit('two')
 
166
 
        self.assertUnmerged([], [('2', rev2)], tree.branch, tree2.branch)
 
167
 
        self.assertUnmerged([('2', rev2)], [], tree2.branch, tree.branch)
 
 
163
        self.assertUnmerged([], [('2', rev2, 0)], tree.branch, tree2.branch)
 
 
164
        self.assertUnmerged([('2', rev2, 0)], [], tree2.branch, tree.branch)
 
169
166
    def test_restrict(self):
 
170
167
        tree = self.make_branch_and_tree('tree')
 
171
 
        rev1 = tree.commit('one')
 
172
 
        tree2 = tree.bzrdir.sprout('tree2').open_workingtree()
 
 
169
        tree2 = tree.controldir.sprout('tree2').open_workingtree()
 
173
170
        rev2 = tree2.commit('two')
 
174
 
        self.assertUnmerged([], [('2', rev2)], tree.branch, tree2.branch)
 
 
171
        self.assertUnmerged([], [('2', rev2, 0)], tree.branch, tree2.branch)
 
175
172
        self.assertUnmerged([], None, tree.branch, tree2.branch, 'local')
 
176
 
        self.assertUnmerged(None, [('2', rev2)], tree.branch, tree2.branch,
 
 
173
        self.assertUnmerged(None, [('2', rev2, 0)], tree.branch, tree2.branch,
 
179
176
    def test_merged(self):
 
180
177
        tree = self.make_branch_and_tree('tree')
 
181
178
        rev1 = tree.commit('one')
 
182
 
        tree2 = tree.bzrdir.sprout('tree2').open_workingtree()
 
183
 
        rev2 = tree2.commit('two')
 
184
 
        rev3 = tree2.commit('three')
 
 
179
        tree2 = tree.controldir.sprout('tree2').open_workingtree()
 
 
181
        tree2.commit('three')
 
185
182
        tree.merge_from_branch(tree2.branch)
 
186
183
        rev4 = tree.commit('four')
 
188
 
        self.assertUnmerged([('2', rev4)], [], tree.branch, tree2.branch)
 
189
 
        self.assertUnmerged([('2', rev4)], [], tree.branch, tree2.branch,
 
190
 
            local_revid_range=(rev4, rev4))
 
 
185
        self.assertUnmerged([('2', rev4, 0)], [], tree.branch, tree2.branch)
 
 
186
        self.assertUnmerged([('2', rev4, 0)], [], tree.branch, tree2.branch,
 
 
187
                            local_revid_range=(rev4, rev4))
 
191
188
        self.assertUnmerged([], [], tree.branch, tree2.branch,
 
192
 
            local_revid_range=(rev1, rev1))
 
 
189
                            local_revid_range=(rev1, rev1))
 
194
 
    def test_include_merges(self):
 
 
191
    def test_include_merged(self):
 
195
192
        tree = self.make_branch_and_tree('tree')
 
196
 
        rev1 = tree.commit('one', rev_id='rev1')
 
198
 
        tree2 = tree.bzrdir.sprout('tree2').open_workingtree()
 
199
 
        rev2 = tree2.commit('two', rev_id='rev2')
 
200
 
        rev3 = tree2.commit('three', rev_id='rev3')
 
202
 
        tree3 = tree2.bzrdir.sprout('tree3').open_workingtree()
 
203
 
        rev4 = tree3.commit('four', rev_id='rev4')
 
204
 
        rev5 = tree3.commit('five', rev_id='rev5')
 
 
193
        tree.commit('one', rev_id=b'rev1')
 
 
195
        tree2 = tree.controldir.sprout('tree2').open_workingtree()
 
 
196
        tree2.commit('two', rev_id=b'rev2')
 
 
197
        rev3 = tree2.commit('three', rev_id=b'rev3')
 
 
199
        tree3 = tree2.controldir.sprout('tree3').open_workingtree()
 
 
200
        rev4 = tree3.commit('four', rev_id=b'rev4')
 
 
201
        rev5 = tree3.commit('five', rev_id=b'rev5')
 
206
203
        tree2.merge_from_branch(tree3.branch)
 
207
 
        rev6 = tree2.commit('six', rev_id='rev6')
 
209
 
        self.assertUnmerged([], [('2', 'rev2', 0), ('3', 'rev3', 0),
 
211
 
                                 ('3.1.1', 'rev4', 1), ('3.1.2', 'rev5', 1),
 
213
 
                            tree.branch, tree2.branch,
 
216
 
        self.assertUnmerged([], [('4', 'rev6', 0),
 
217
 
                                 ('3.1.2', 'rev5', 1), ('3.1.1', 'rev4', 1),
 
218
 
                                 ('3', 'rev3', 0), ('2', 'rev2', 0),
 
220
 
                            tree.branch, tree2.branch,
 
 
204
        rev6 = tree2.commit('six', rev_id=b'rev6')
 
 
206
        self.assertUnmerged([], [('2', b'rev2', 0), ('3', b'rev3', 0),
 
 
208
                                 ('3.1.1', b'rev4', 1), ('3.1.2', b'rev5', 1),
 
 
210
                            tree.branch, tree2.branch,
 
 
213
        self.assertUnmerged([], [('4', b'rev6', 0),
 
 
214
                                 ('3.1.2', b'rev5', 1), ('3.1.1', b'rev4', 1),
 
 
215
                                 ('3', b'rev3', 0), ('2', b'rev2', 0),
 
 
217
                            tree.branch, tree2.branch,
 
224
 
        self.assertUnmerged([], [('4', 'rev6', 0)],
 
225
 
            tree.branch, tree2.branch,
 
226
 
            include_merges=True, remote_revid_range=(rev6, rev6))
 
228
 
        self.assertUnmerged([], [('3', 'rev3', 0), ('3.1.1', 'rev4', 1)],
 
229
 
                    tree.branch, tree2.branch,
 
230
 
                    include_merges=True, remote_revid_range=(rev3, rev4))
 
232
 
        self.assertUnmerged([], [('4', 'rev6', 0), ('3.1.2', 'rev5', 1)],
 
233
 
                    tree.branch, tree2.branch,
 
234
 
                    include_merges=True, remote_revid_range=(rev5, rev6))
 
 
222
            [], [('4', b'rev6', 0)],
 
 
223
            tree.branch, tree2.branch,
 
 
224
            include_merged=True, remote_revid_range=(rev6, rev6))
 
 
227
            [], [('3', b'rev3', 0), ('3.1.1', b'rev4', 1)],
 
 
228
            tree.branch, tree2.branch,
 
 
229
            include_merged=True, remote_revid_range=(rev3, rev4))
 
 
232
            [], [('4', b'rev6', 0), ('3.1.2', b'rev5', 1)],
 
 
233
            tree.branch, tree2.branch,
 
 
234
            include_merged=True, remote_revid_range=(rev5, rev6))
 
236
236
    def test_revision_range(self):
 
237
237
        local = self.make_branch_and_tree('local')
 
238
238
        lrevid1 = local.commit('one')
 
239
 
        remote = local.bzrdir.sprout('remote').open_workingtree()
 
 
239
        remote = local.controldir.sprout('remote').open_workingtree()
 
240
240
        rrevid2 = remote.commit('two')
 
241
241
        rrevid3 = remote.commit('three')
 
242
242
        rrevid4 = remote.commit('four')
 
243
243
        lrevid2 = local.commit('two')
 
244
244
        lrevid3 = local.commit('three')
 
245
245
        lrevid4 = local.commit('four')
 
246
 
        local_extra = [('2', lrevid2), ('3', lrevid3), ('4', lrevid4)]
 
247
 
        remote_extra = [('2', rrevid2), ('3', rrevid3), ('4', rrevid4)]
 
 
246
        local_extra = [('2', lrevid2, 0), ('3', lrevid3, 0), ('4', lrevid4, 0)]
 
 
247
        remote_extra = [('2', rrevid2, 0), ('3', rrevid3, 0),
 
250
251
        self.assertUnmerged(local_extra, remote_extra,
 
251
 
            local.branch, remote.branch)
 
 
252
                            local.branch, remote.branch)
 
252
253
        self.assertUnmerged(local_extra, remote_extra,
 
253
 
            local.branch, remote.branch, local_revid_range=(None, None),
 
254
 
            remote_revid_range=(None, None))
 
 
254
                            local.branch, remote.branch, local_revid_range=(
 
 
256
                            remote_revid_range=(None, None))
 
256
258
        # exclude local revisions
 
257
 
        self.assertUnmerged([('2', lrevid2)], remote_extra,
 
 
260
            [('2', lrevid2, 0)], remote_extra,
 
258
261
            local.branch, remote.branch, local_revid_range=(lrevid2, lrevid2))
 
259
 
        self.assertUnmerged([('2', lrevid2), ('3', lrevid3)], remote_extra,
 
 
263
            [('2', lrevid2, 0), ('3', lrevid3, 0)], remote_extra,
 
260
264
            local.branch, remote.branch, local_revid_range=(lrevid2, lrevid3))
 
261
 
        self.assertUnmerged([('2', lrevid2), ('3', lrevid3)], None,
 
 
266
            [('2', lrevid2, 0), ('3', lrevid3, 0)], None,
 
262
267
            local.branch, remote.branch, 'local',
 
263
268
            local_revid_range=(lrevid2, lrevid3))
 
265
270
        # exclude remote revisions
 
266
 
        self.assertUnmerged(local_extra, [('2', rrevid2)],
 
 
272
            local_extra, [('2', rrevid2, 0)],
 
267
273
            local.branch, remote.branch, remote_revid_range=(None, rrevid2))
 
268
 
        self.assertUnmerged(local_extra, [('2', rrevid2)],
 
 
275
            local_extra, [('2', rrevid2, 0)],
 
269
276
            local.branch, remote.branch, remote_revid_range=(lrevid1, rrevid2))
 
270
 
        self.assertUnmerged(local_extra, [('2', rrevid2)],
 
 
278
            local_extra, [('2', rrevid2, 0)],
 
271
279
            local.branch, remote.branch, remote_revid_range=(rrevid2, rrevid2))
 
272
 
        self.assertUnmerged(local_extra, [('2', rrevid2), ('3', rrevid3)],
 
 
281
            local_extra, [('2', rrevid2, 0), ('3', rrevid3, 0)],
 
273
282
            local.branch, remote.branch, remote_revid_range=(None, rrevid3))
 
274
 
        self.assertUnmerged(local_extra, [('2', rrevid2), ('3', rrevid3)],
 
 
284
            local_extra, [('2', rrevid2, 0), ('3', rrevid3, 0)],
 
275
285
            local.branch, remote.branch, remote_revid_range=(rrevid2, rrevid3))
 
276
 
        self.assertUnmerged(local_extra, [('3', rrevid3)],
 
 
287
            local_extra, [('3', rrevid3, 0)],
 
277
288
            local.branch, remote.branch, remote_revid_range=(rrevid3, rrevid3))
 
278
 
        self.assertUnmerged(None, [('2', rrevid2), ('3', rrevid3)],
 
279
 
            local.branch, remote.branch, 'remote',
 
280
 
            remote_revid_range=(rrevid2, rrevid3))
 
 
289
        self.assertUnmerged(None, [('2', rrevid2, 0), ('3', rrevid3, 0)],
 
 
290
                            local.branch, remote.branch, 'remote',
 
 
291
                            remote_revid_range=(rrevid2, rrevid3))
 
282
293
        # exclude local and remote revisions
 
283
 
        self.assertUnmerged([('3', lrevid3)], [('3', rrevid3)],
 
284
 
            local.branch, remote.branch, local_revid_range=(lrevid3, lrevid3),
 
285
 
            remote_revid_range=(rrevid3, rrevid3))
 
 
294
        self.assertUnmerged([('3', lrevid3, 0)], [('3', rrevid3, 0)],
 
 
295
                            local.branch, remote.branch, local_revid_range=(
 
 
297
                            remote_revid_range=(rrevid3, rrevid3))