/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/tests/test_branchbuilder.py

  • Committer: Jelmer Vernooij
  • Date: 2018-07-08 14:45:27 UTC
  • mto: This revision was merged to the branch mainline in revision 7036.
  • Revision ID: jelmer@jelmer.uk-20180708144527-codhlvdcdg9y0nji
Fix a bunch of merge tests.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/tests/test_branchbuilder.py
1
 
# Copyright (C) 2007, 2009 Canonical Ltd
 
1
# Copyright (C) 2007-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
16
16
 
17
17
"""Tests for the BranchBuilder class."""
18
18
 
19
 
from bzrlib import (
 
19
from .. import (
20
20
    branch as _mod_branch,
21
 
    errors,
22
21
    revision as _mod_revision,
23
22
    tests,
24
23
    )
25
 
from bzrlib.branchbuilder import BranchBuilder
 
24
from ..bzr import (
 
25
    branch as _mod_bzrbranch,
 
26
    )
 
27
from ..branchbuilder import BranchBuilder
26
28
 
27
29
 
28
30
class TestBranchBuilder(tests.TestCaseWithMemoryTransport):
47
49
        """Making a BranchBuilder with a format option sets the branch type."""
48
50
        builder = BranchBuilder(self.get_transport(), format='dirstate-tags')
49
51
        branch = builder.get_branch()
50
 
        self.assertIsInstance(branch, _mod_branch.BzrBranch6)
 
52
        self.assertIsInstance(branch, _mod_bzrbranch.BzrBranch6)
51
53
 
52
54
    def test_build_one_commit(self):
53
55
        """doing build_commit causes a commit to happen."""
87
89
            [rev_id1],
88
90
            branch.repository.get_revision(branch.last_revision()).parent_ids)
89
91
 
 
92
    def test_build_commit_parent_ids(self):
 
93
        """build_commit() takes a parent_ids argument."""
 
94
        builder = BranchBuilder(self.get_transport().clone('foo'))
 
95
        rev_id1 = builder.build_commit(
 
96
            parent_ids=[b"ghost"], allow_leftmost_as_ghost=True)
 
97
        rev_id2 = builder.build_commit(parent_ids=[])
 
98
        branch = builder.get_branch()
 
99
        self.assertEqual((1, rev_id2), branch.last_revision_info())
 
100
        self.assertEqual(
 
101
            [b"ghost"],
 
102
            branch.repository.get_revision(rev_id1).parent_ids)
 
103
 
90
104
 
91
105
class TestBranchBuilderBuildSnapshot(tests.TestCaseWithMemoryTransport):
92
106
 
102
116
 
103
117
    def build_a_rev(self):
104
118
        builder = BranchBuilder(self.get_transport().clone('foo'))
105
 
        rev_id1 = builder.build_snapshot('A-id', None,
106
 
            [('add', ('', 'a-root-id', 'directory', None)),
107
 
             ('add', ('a', 'a-id', 'file', 'contents'))])
108
 
        self.assertEqual('A-id', rev_id1)
 
119
        rev_id1 = builder.build_snapshot(None,
 
120
            [('add', ('', b'a-root-id', 'directory', None)),
 
121
             ('add', ('a', b'a-id', 'file', b'contents'))],
 
122
            revision_id=b'A-id')
 
123
        self.assertEqual(b'A-id', rev_id1)
109
124
        return builder
110
125
 
111
126
    def test_add_one_file(self):
112
127
        builder = self.build_a_rev()
113
128
        branch = builder.get_branch()
114
 
        self.assertEqual((1, 'A-id'), branch.last_revision_info())
115
 
        rev_tree = branch.repository.revision_tree('A-id')
 
129
        self.assertEqual((1, b'A-id'), branch.last_revision_info())
 
130
        rev_tree = branch.repository.revision_tree(b'A-id')
116
131
        rev_tree.lock_read()
117
132
        self.addCleanup(rev_tree.unlock)
118
 
        self.assertTreeShape([(u'', 'a-root-id', 'directory'),
119
 
                              (u'a', 'a-id', 'file')], rev_tree)
120
 
        self.assertEqual('contents', rev_tree.get_file_text('a-id'))
 
133
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
 
134
                              (u'a', b'a-id', 'file')], rev_tree)
 
135
        self.assertEqual(b'contents', rev_tree.get_file_text('a'))
121
136
 
122
137
    def test_add_second_file(self):
123
138
        builder = self.build_a_rev()
124
 
        rev_id2 = builder.build_snapshot('B-id', None,
125
 
            [('add', ('b', 'b-id', 'file', 'content_b'))])
126
 
        self.assertEqual('B-id', rev_id2)
 
139
        rev_id2 = builder.build_snapshot(None,
 
140
            [('add', ('b', b'b-id', 'file', b'content_b'))],
 
141
            revision_id=b'B-id')
 
142
        self.assertEqual(b'B-id', rev_id2)
127
143
        branch = builder.get_branch()
128
144
        self.assertEqual((2, rev_id2), branch.last_revision_info())
129
145
        rev_tree = branch.repository.revision_tree(rev_id2)
130
146
        rev_tree.lock_read()
131
147
        self.addCleanup(rev_tree.unlock)
132
 
        self.assertTreeShape([(u'', 'a-root-id', 'directory'),
133
 
                              (u'a', 'a-id', 'file'),
134
 
                              (u'b', 'b-id', 'file')], rev_tree)
135
 
        self.assertEqual('content_b', rev_tree.get_file_text('b-id'))
 
148
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
 
149
                              (u'a', b'a-id', 'file'),
 
150
                              (u'b', b'b-id', 'file')], rev_tree)
 
151
        self.assertEqual(b'content_b', rev_tree.get_file_text('b'))
136
152
 
137
153
    def test_add_empty_dir(self):
138
154
        builder = self.build_a_rev()
139
 
        rev_id2 = builder.build_snapshot('B-id', None,
140
 
            [('add', ('b', 'b-id', 'directory', None))])
141
 
        rev_tree = builder.get_branch().repository.revision_tree('B-id')
142
 
        self.assertTreeShape([(u'', 'a-root-id', 'directory'),
143
 
                              (u'a', 'a-id', 'file'),
144
 
                              (u'b', 'b-id', 'directory'),
 
155
        rev_id2 = builder.build_snapshot(None,
 
156
            [('add', ('b', b'b-id', 'directory', None))],
 
157
            revision_id=b'B-id')
 
158
        rev_tree = builder.get_branch().repository.revision_tree(b'B-id')
 
159
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
 
160
                              (u'a', b'a-id', 'file'),
 
161
                              (u'b', b'b-id', 'directory'),
145
162
                             ], rev_tree)
146
163
 
147
164
    def test_commit_timestamp(self):
148
165
        builder = self.make_branch_builder('foo')
149
 
        rev_id = builder.build_snapshot(None, None,
 
166
        rev_id = builder.build_snapshot(None,
150
167
            [('add', (u'', None, 'directory', None))],
151
168
            timestamp=1234567890)
152
169
        rev = builder.get_branch().repository.get_revision(rev_id)
156
173
 
157
174
    def test_commit_message_default(self):
158
175
        builder = BranchBuilder(self.get_transport().clone('foo'))
159
 
        rev_id = builder.build_snapshot(None, None,
 
176
        rev_id = builder.build_snapshot(None,
160
177
            [('add', (u'', None, 'directory', None))])
161
178
        branch = builder.get_branch()
162
179
        rev = branch.repository.get_revision(rev_id)
164
181
 
165
182
    def test_commit_message_supplied(self):
166
183
        builder = BranchBuilder(self.get_transport().clone('foo'))
167
 
        rev_id = builder.build_snapshot(None, None,
 
184
        rev_id = builder.build_snapshot(None,
168
185
            [('add', (u'', None, 'directory', None))],
169
186
            message=u'Foo')
170
187
        branch = builder.get_branch()
173
190
 
174
191
    def test_commit_message_callback(self):
175
192
        builder = BranchBuilder(self.get_transport().clone('foo'))
176
 
        rev_id = builder.build_snapshot(None, None,
 
193
        rev_id = builder.build_snapshot(None,
177
194
            [('add', (u'', None, 'directory', None))],
178
195
            message_callback=lambda x:u'Foo')
179
196
        branch = builder.get_branch()
182
199
 
183
200
    def test_modify_file(self):
184
201
        builder = self.build_a_rev()
185
 
        rev_id2 = builder.build_snapshot('B-id', None,
186
 
            [('modify', ('a-id', 'new\ncontent\n'))])
187
 
        self.assertEqual('B-id', rev_id2)
 
202
        rev_id2 = builder.build_snapshot(None,
 
203
            [('modify', ('a', b'new\ncontent\n'))],
 
204
            revision_id=b'B-id')
 
205
        self.assertEqual(b'B-id', rev_id2)
188
206
        branch = builder.get_branch()
189
207
        rev_tree = branch.repository.revision_tree(rev_id2)
190
208
        rev_tree.lock_read()
191
209
        self.addCleanup(rev_tree.unlock)
192
 
        self.assertEqual('new\ncontent\n', rev_tree.get_file_text('a-id'))
 
210
        self.assertEqual(b'new\ncontent\n',
 
211
                         rev_tree.get_file_text(rev_tree.id2path(b'a-id')))
193
212
 
194
213
    def test_delete_file(self):
195
214
        builder = self.build_a_rev()
196
 
        rev_id2 = builder.build_snapshot('B-id', None,
197
 
            [('unversion', 'a-id')])
198
 
        self.assertEqual('B-id', rev_id2)
 
215
        rev_id2 = builder.build_snapshot(None,
 
216
            [('unversion', 'a')], revision_id=b'B-id')
 
217
        self.assertEqual(b'B-id', rev_id2)
199
218
        branch = builder.get_branch()
200
219
        rev_tree = branch.repository.revision_tree(rev_id2)
201
220
        rev_tree.lock_read()
202
221
        self.addCleanup(rev_tree.unlock)
203
 
        self.assertTreeShape([(u'', 'a-root-id', 'directory')], rev_tree)
 
222
        self.assertTreeShape([(u'', b'a-root-id', 'directory')], rev_tree)
204
223
 
205
224
    def test_delete_directory(self):
206
225
        builder = self.build_a_rev()
207
 
        rev_id2 = builder.build_snapshot('B-id', None,
208
 
            [('add', ('b', 'b-id', 'directory', None)),
209
 
             ('add', ('b/c', 'c-id', 'file', 'foo\n')),
210
 
             ('add', ('b/d', 'd-id', 'directory', None)),
211
 
             ('add', ('b/d/e', 'e-id', 'file', 'eff\n')),
212
 
            ])
213
 
        rev_tree = builder.get_branch().repository.revision_tree('B-id')
214
 
        self.assertTreeShape([(u'', 'a-root-id', 'directory'),
215
 
                              (u'a', 'a-id', 'file'),
216
 
                              (u'b', 'b-id', 'directory'),
217
 
                              (u'b/c', 'c-id', 'file'),
218
 
                              (u'b/d', 'd-id', 'directory'),
219
 
                              (u'b/d/e', 'e-id', 'file')], rev_tree)
 
226
        rev_id2 = builder.build_snapshot(None,
 
227
            [('add', ('b', b'b-id', 'directory', None)),
 
228
             ('add', ('b/c', b'c-id', 'file', b'foo\n')),
 
229
             ('add', ('b/d', b'd-id', 'directory', None)),
 
230
             ('add', ('b/d/e', b'e-id', 'file', b'eff\n')),
 
231
            ], revision_id=b'B-id')
 
232
        rev_tree = builder.get_branch().repository.revision_tree(b'B-id')
 
233
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
 
234
                              (u'a', b'a-id', 'file'),
 
235
                              (u'b', b'b-id', 'directory'),
 
236
                              (u'b/c', b'c-id', 'file'),
 
237
                              (u'b/d', b'd-id', 'directory'),
 
238
                              (u'b/d/e', b'e-id', 'file')], rev_tree)
220
239
        # Removing a directory removes all child dirs
221
 
        builder.build_snapshot('C-id', None, [('unversion', 'b-id')])
222
 
        rev_tree = builder.get_branch().repository.revision_tree('C-id')
223
 
        self.assertTreeShape([(u'', 'a-root-id', 'directory'),
224
 
                              (u'a', 'a-id', 'file'),
 
240
        builder.build_snapshot(
 
241
                None, [('unversion', 'b')],
 
242
                revision_id=b'C-id')
 
243
        rev_tree = builder.get_branch().repository.revision_tree(b'C-id')
 
244
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
 
245
                              (u'a', b'a-id', 'file'),
225
246
                             ], rev_tree)
226
247
 
227
248
    def test_unknown_action(self):
228
249
        builder = self.build_a_rev()
229
250
        e = self.assertRaises(ValueError,
230
 
            builder.build_snapshot, 'B-id', None, [('weirdo', ('foo',))])
 
251
            builder.build_snapshot, None, [('weirdo', ('foo',))],
 
252
            revision_id=b'B-id')
231
253
        self.assertEqual('Unknown build action: "weirdo"', str(e))
232
254
 
233
255
    def test_rename(self):
234
256
        builder = self.build_a_rev()
235
 
        builder.build_snapshot('B-id', None,
236
 
            [('rename', ('a', 'b'))])
237
 
        rev_tree = builder.get_branch().repository.revision_tree('B-id')
238
 
        self.assertTreeShape([(u'', 'a-root-id', 'directory'),
239
 
                              (u'b', 'a-id', 'file')], rev_tree)
 
257
        builder.build_snapshot(None,
 
258
            [('rename', ('a', 'b'))], revision_id=b'B-id')
 
259
        rev_tree = builder.get_branch().repository.revision_tree(b'B-id')
 
260
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
 
261
                              (u'b', b'a-id', 'file')], rev_tree)
240
262
 
241
263
    def test_rename_into_subdir(self):
242
264
        builder = self.build_a_rev()
243
 
        builder.build_snapshot('B-id', None,
244
 
            [('add', ('dir', 'dir-id', 'directory', None)),
245
 
             ('rename', ('a', 'dir/a'))])
246
 
        rev_tree = builder.get_branch().repository.revision_tree('B-id')
247
 
        self.assertTreeShape([(u'', 'a-root-id', 'directory'),
248
 
                              (u'dir', 'dir-id', 'directory'),
249
 
                              (u'dir/a', 'a-id', 'file')], rev_tree)
 
265
        builder.build_snapshot(None,
 
266
            [('add', ('dir', b'dir-id', 'directory', None)),
 
267
             ('rename', ('a', 'dir/a'))], revision_id=b'B-id')
 
268
        rev_tree = builder.get_branch().repository.revision_tree(b'B-id')
 
269
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
 
270
                              (u'dir', b'dir-id', 'directory'),
 
271
                              (u'dir/a', b'a-id', 'file')], rev_tree)
 
272
 
 
273
    def test_rename_out_of_unversioned_subdir(self):
 
274
        builder = self.build_a_rev()
 
275
        builder.build_snapshot(None,
 
276
            [('add', ('dir', b'dir-id', 'directory', None)),
 
277
             ('rename', ('a', 'dir/a'))],
 
278
            revision_id=b'B-id')
 
279
        builder.build_snapshot(None,
 
280
            [('rename', ('dir/a', 'a')),
 
281
             ('unversion', 'dir')], revision_id=b'C-id')
 
282
        rev_tree = builder.get_branch().repository.revision_tree(b'C-id')
 
283
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
 
284
                              (u'a', b'a-id', 'file')], rev_tree)
250
285
 
251
286
    def test_set_parent(self):
252
287
        builder = self.build_a_rev()
253
288
        builder.start_series()
254
289
        self.addCleanup(builder.finish_series)
255
 
        builder.build_snapshot('B-id', ['A-id'],
256
 
            [('modify', ('a-id', 'new\ncontent\n'))])
257
 
        builder.build_snapshot('C-id', ['A-id'],
258
 
            [('add', ('c', 'c-id', 'file', 'alt\ncontent\n'))])
 
290
        builder.build_snapshot([b'A-id'],
 
291
            [('modify', ('a', b'new\ncontent\n'))],
 
292
            revision_id=b'B-id')
 
293
        builder.build_snapshot([b'A-id'],
 
294
            [('add', ('c', b'c-id', 'file', b'alt\ncontent\n'))],
 
295
            revision_id=b'C-id')
259
296
        # We should now have a graph:
260
297
        #   A
261
298
        #   |\
262
299
        #   C B
263
300
        # And not A => B => C
264
301
        repo = builder.get_branch().repository
265
 
        self.assertEqual({'B-id': ('A-id',), 'C-id': ('A-id',)},
266
 
                         repo.get_parent_map(['B-id', 'C-id']))
267
 
        b_tree = repo.revision_tree('B-id')
268
 
        self.assertTreeShape([(u'', 'a-root-id', 'directory'),
269
 
                              (u'a', 'a-id', 'file'),
 
302
        self.assertEqual({b'B-id': (b'A-id',), b'C-id': (b'A-id',)},
 
303
                         repo.get_parent_map([b'B-id', b'C-id']))
 
304
        b_tree = repo.revision_tree(b'B-id')
 
305
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
 
306
                              (u'a', b'a-id', 'file'),
270
307
                             ], b_tree)
271
 
        self.assertEqual('new\ncontent\n', b_tree.get_file_text('a-id'))
 
308
        self.assertEqual(b'new\ncontent\n', b_tree.get_file_text('a'))
272
309
 
273
310
        # We should still be using the content from A in C, not from B
274
 
        c_tree = repo.revision_tree('C-id')
275
 
        self.assertTreeShape([(u'', 'a-root-id', 'directory'),
276
 
                              (u'a', 'a-id', 'file'),
277
 
                              (u'c', 'c-id', 'file'),
 
311
        c_tree = repo.revision_tree(b'C-id')
 
312
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
 
313
                              (u'a', b'a-id', 'file'),
 
314
                              (u'c', b'c-id', 'file'),
278
315
                             ], c_tree)
279
 
        self.assertEqual('contents', c_tree.get_file_text('a-id'))
280
 
        self.assertEqual('alt\ncontent\n', c_tree.get_file_text('c-id'))
 
316
        self.assertEqual(b'contents', c_tree.get_file_text('a'))
 
317
        self.assertEqual(b'alt\ncontent\n', c_tree.get_file_text('c'))
281
318
 
282
319
    def test_set_merge_parent(self):
283
320
        builder = self.build_a_rev()
284
321
        builder.start_series()
285
322
        self.addCleanup(builder.finish_series)
286
 
        builder.build_snapshot('B-id', ['A-id'],
287
 
            [('add', ('b', 'b-id', 'file', 'b\ncontent\n'))])
288
 
        builder.build_snapshot('C-id', ['A-id'],
289
 
            [('add', ('c', 'c-id', 'file', 'alt\ncontent\n'))])
290
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'], [])
 
323
        builder.build_snapshot([b'A-id'],
 
324
            [('add', ('b', b'b-id', 'file', b'b\ncontent\n'))],
 
325
            revision_id=b'B-id')
 
326
        builder.build_snapshot([b'A-id'],
 
327
            [('add', ('c', b'c-id', 'file', b'alt\ncontent\n'))],
 
328
            revision_id=b'C-id')
 
329
        builder.build_snapshot([b'B-id', b'C-id'], [], revision_id=b'D-id')
291
330
        repo = builder.get_branch().repository
292
 
        self.assertEqual({'B-id': ('A-id',), 'C-id': ('A-id',),
293
 
                          'D-id': ('B-id', 'C-id')},
294
 
                         repo.get_parent_map(['B-id', 'C-id', 'D-id']))
295
 
        d_tree = repo.revision_tree('D-id')
 
331
        self.assertEqual({b'B-id': (b'A-id',), b'C-id': (b'A-id',),
 
332
                          b'D-id': (b'B-id', b'C-id')},
 
333
                         repo.get_parent_map([b'B-id', b'C-id', b'D-id']))
 
334
        d_tree = repo.revision_tree(b'D-id')
296
335
        # Note: by default a merge node does *not* pull in the changes from the
297
336
        #       merged tree, you have to supply it yourself.
298
 
        self.assertTreeShape([(u'', 'a-root-id', 'directory'),
299
 
                              (u'a', 'a-id', 'file'),
300
 
                              (u'b', 'b-id', 'file'),
 
337
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
 
338
                              (u'a', b'a-id', 'file'),
 
339
                              (u'b', b'b-id', 'file'),
301
340
                             ], d_tree)
302
341
 
303
342
    def test_set_merge_parent_and_contents(self):
304
343
        builder = self.build_a_rev()
305
344
        builder.start_series()
306
345
        self.addCleanup(builder.finish_series)
307
 
        builder.build_snapshot('B-id', ['A-id'],
308
 
            [('add', ('b', 'b-id', 'file', 'b\ncontent\n'))])
309
 
        builder.build_snapshot('C-id', ['A-id'],
310
 
            [('add', ('c', 'c-id', 'file', 'alt\ncontent\n'))])
311
 
        builder.build_snapshot('D-id', ['B-id', 'C-id'],
312
 
            [('add', ('c', 'c-id', 'file', 'alt\ncontent\n'))])
 
346
        builder.build_snapshot([b'A-id'],
 
347
            [('add', ('b', b'b-id', 'file', b'b\ncontent\n'))],
 
348
            revision_id=b'B-id')
 
349
        builder.build_snapshot([b'A-id'],
 
350
            [('add', ('c', b'c-id', 'file', b'alt\ncontent\n'))],
 
351
            revision_id=b'C-id')
 
352
        builder.build_snapshot([b'B-id', b'C-id'],
 
353
            [('add', ('c', b'c-id', 'file', b'alt\ncontent\n'))],
 
354
            revision_id=b'D-id')
313
355
        repo = builder.get_branch().repository
314
 
        self.assertEqual({'B-id': ('A-id',), 'C-id': ('A-id',),
315
 
                          'D-id': ('B-id', 'C-id')},
316
 
                         repo.get_parent_map(['B-id', 'C-id', 'D-id']))
317
 
        d_tree = repo.revision_tree('D-id')
318
 
        self.assertTreeShape([(u'', 'a-root-id', 'directory'),
319
 
                              (u'a', 'a-id', 'file'),
320
 
                              (u'b', 'b-id', 'file'),
321
 
                              (u'c', 'c-id', 'file'),
 
356
        self.assertEqual({b'B-id': (b'A-id',), b'C-id': (b'A-id',),
 
357
                          b'D-id': (b'B-id', b'C-id')},
 
358
                         repo.get_parent_map([b'B-id', b'C-id', b'D-id']))
 
359
        d_tree = repo.revision_tree(b'D-id')
 
360
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
 
361
                              (u'a', b'a-id', 'file'),
 
362
                              (u'b', b'b-id', 'file'),
 
363
                              (u'c', b'c-id', 'file'),
322
364
                             ], d_tree)
323
365
        # Because we copied the exact text into *this* tree, the 'c' file
324
366
        # should look like it was not modified in the merge
325
 
        self.assertEqual('C-id', d_tree.inventory['c-id'].revision)
 
367
        self.assertEqual(b'C-id', d_tree.get_file_revision('c'))
 
368
 
 
369
    def test_set_parent_to_null(self):
 
370
        builder = self.build_a_rev()
 
371
        builder.start_series()
 
372
        self.addCleanup(builder.finish_series)
 
373
        builder.build_snapshot([],
 
374
            [('add', ('', None, 'directory', None))],
 
375
            revision_id=b'B-id')
 
376
        # We should now have a graph:
 
377
        #   A B
 
378
        # And not A => B
 
379
        repo = builder.get_branch().repository
 
380
        self.assertEqual({b'A-id': (_mod_revision.NULL_REVISION,),
 
381
                          b'B-id': (_mod_revision.NULL_REVISION,),},
 
382
                         repo.get_parent_map([b'A-id', b'B-id']))
326
383
 
327
384
    def test_start_finish_series(self):
328
385
        builder = BranchBuilder(self.get_transport().clone('foo'))
340
397
        builder = BranchBuilder(self.get_transport().clone('foo'))
341
398
        builder.start_series()
342
399
        try:
343
 
            builder.build_snapshot('tip', ['ghost'],
344
 
                [('add', ('', 'ROOT_ID', 'directory', ''))],
345
 
                allow_leftmost_as_ghost=True)
 
400
            builder.build_snapshot([b'ghost'],
 
401
                [('add', ('', b'ROOT_ID', 'directory', ''))],
 
402
                allow_leftmost_as_ghost=True, revision_id=b'tip')
346
403
        finally:
347
404
            builder.finish_series()
348
405
        b = builder.get_branch()
349
406
        b.lock_read()
350
407
        self.addCleanup(b.unlock)
351
 
        self.assertEqual(('ghost',),
352
 
            b.repository.get_graph().get_parent_map(['tip'])['tip'])
 
408
        self.assertEqual((b'ghost',),
 
409
            b.repository.get_graph().get_parent_map([b'tip'])[b'tip'])
 
410
 
 
411
    def test_unversion_root_add_new_root(self):
 
412
        builder = BranchBuilder(self.get_transport().clone('foo'))
 
413
        builder.start_series()
 
414
        builder.build_snapshot(None,
 
415
            [('add', ('', b'TREE_ROOT', 'directory', ''))],
 
416
            revision_id=b'rev-1')
 
417
        builder.build_snapshot(None,
 
418
            [('unversion', ''),
 
419
             ('add', ('', b'my-root', 'directory', ''))],
 
420
            revision_id=b'rev-2')
 
421
        builder.finish_series()
 
422
        rev_tree = builder.get_branch().repository.revision_tree(b'rev-2')
 
423
        self.assertTreeShape([(u'', b'my-root', 'directory')], rev_tree)
 
424
 
 
425
    def test_empty_flush(self):
 
426
        """A flush with no actions before it is a no-op."""
 
427
        builder = BranchBuilder(self.get_transport().clone('foo'))
 
428
        builder.start_series()
 
429
        builder.build_snapshot(None,
 
430
            [('add', ('', b'TREE_ROOT', 'directory', ''))],
 
431
            revision_id=b'rev-1')
 
432
        builder.build_snapshot(None, [('flush', None)], revision_id=b'rev-2')
 
433
        builder.finish_series()
 
434
        rev_tree = builder.get_branch().repository.revision_tree(b'rev-2')
 
435
        self.assertTreeShape([(u'', b'TREE_ROOT', 'directory')], rev_tree)
 
436
 
 
437
    def test_kind_change(self):
 
438
        """It's possible to change the kind of an entry in a single snapshot
 
439
        with a bit of help from the 'flush' action.
 
440
        """
 
441
        builder = BranchBuilder(self.get_transport().clone('foo'))
 
442
        builder.start_series()
 
443
        builder.build_snapshot(None,
 
444
            [('add', (u'', b'a-root-id', 'directory', None)),
 
445
             ('add', (u'a', b'a-id', 'file', b'content\n'))],
 
446
            revision_id=b'A-id')
 
447
        builder.build_snapshot(None,
 
448
            [('unversion', 'a'),
 
449
             ('flush', None),
 
450
             ('add', (u'a', b'a-id', 'directory', None))],
 
451
            revision_id=b'B-id')
 
452
        builder.finish_series()
 
453
        rev_tree = builder.get_branch().repository.revision_tree(b'B-id')
 
454
        self.assertTreeShape(
 
455
            [(u'', b'a-root-id', 'directory'), (u'a', b'a-id', 'directory')],
 
456
            rev_tree)
 
457
 
 
458
    def test_pivot_root(self):
 
459
        """It's possible (albeit awkward) to move an existing dir to the root
 
460
        in a single snapshot by using unversion then flush then add.
 
461
        """
 
462
        builder = BranchBuilder(self.get_transport().clone('foo'))
 
463
        builder.start_series()
 
464
        builder.build_snapshot(None,
 
465
            [('add', (u'', b'orig-root', 'directory', None)),
 
466
             ('add', (u'dir', b'dir-id', 'directory', None))],
 
467
            revision_id=b'A-id')
 
468
        builder.build_snapshot(None,
 
469
            [('unversion', ''),  # implicitly unversions all children
 
470
             ('flush', None),
 
471
             ('add', (u'', b'dir-id', 'directory', None))],
 
472
            revision_id=b'B-id')
 
473
        builder.finish_series()
 
474
        rev_tree = builder.get_branch().repository.revision_tree(b'B-id')
 
475
        self.assertTreeShape([(u'', b'dir-id', 'directory')], rev_tree)
 
476