/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: Breezy landing bot
  • Author(s): Jelmer Vernooij
  • Date: 2018-11-16 18:59:44 UTC
  • mfrom: (7143.15.15 more-cleanups)
  • Revision ID: breezy.the.bot@gmail.com-20181116185944-biefv1sub37qfybm
Sprinkle some PEP8iness.

Merged from https://code.launchpad.net/~jelmer/brz/more-cleanups/+merge/358611

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/tests/test_branchbuilder.py
40
40
        branch = builder.get_branch()
41
41
        self.assertIsInstance(branch, _mod_branch.Branch)
42
42
        self.assertEqual(self.get_transport().clone('foo').base,
43
 
            branch.base)
 
43
                         branch.base)
44
44
        self.assertEqual(
45
45
            (0, _mod_revision.NULL_REVISION),
46
46
            branch.last_revision_info())
117
117
    def build_a_rev(self):
118
118
        builder = BranchBuilder(self.get_transport().clone('foo'))
119
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')
 
120
                                         [('add', ('', b'a-root-id', 'directory', None)),
 
121
                                          ('add', ('a', b'a-id', 'file', b'contents'))],
 
122
                                         revision_id=b'A-id')
123
123
        self.assertEqual(b'A-id', rev_id1)
124
124
        return builder
125
125
 
137
137
    def test_add_second_file(self):
138
138
        builder = self.build_a_rev()
139
139
        rev_id2 = builder.build_snapshot(None,
140
 
            [('add', ('b', b'b-id', 'file', b'content_b'))],
141
 
            revision_id=b'B-id')
 
140
                                         [('add', ('b', b'b-id', 'file', b'content_b'))],
 
141
                                         revision_id=b'B-id')
142
142
        self.assertEqual(b'B-id', rev_id2)
143
143
        branch = builder.get_branch()
144
144
        self.assertEqual((2, rev_id2), branch.last_revision_info())
153
153
    def test_add_empty_dir(self):
154
154
        builder = self.build_a_rev()
155
155
        rev_id2 = builder.build_snapshot(None,
156
 
            [('add', ('b', b'b-id', 'directory', None))],
157
 
            revision_id=b'B-id')
 
156
                                         [('add', ('b', b'b-id', 'directory', None))],
 
157
                                         revision_id=b'B-id')
158
158
        rev_tree = builder.get_branch().repository.revision_tree(b'B-id')
159
159
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
160
160
                              (u'a', b'a-id', 'file'),
161
161
                              (u'b', b'b-id', 'directory'),
162
 
                             ], rev_tree)
 
162
                              ], rev_tree)
163
163
 
164
164
    def test_commit_timestamp(self):
165
165
        builder = self.make_branch_builder('foo')
166
166
        rev_id = builder.build_snapshot(None,
167
 
            [('add', (u'', None, 'directory', None))],
168
 
            timestamp=1234567890)
 
167
                                        [('add', (u'', None, 'directory', None))],
 
168
                                        timestamp=1234567890)
169
169
        rev = builder.get_branch().repository.get_revision(rev_id)
170
170
        self.assertEqual(
171
171
            1234567890,
174
174
    def test_commit_message_default(self):
175
175
        builder = BranchBuilder(self.get_transport().clone('foo'))
176
176
        rev_id = builder.build_snapshot(None,
177
 
            [('add', (u'', None, 'directory', None))])
 
177
                                        [('add', (u'', None, 'directory', None))])
178
178
        branch = builder.get_branch()
179
179
        rev = branch.repository.get_revision(rev_id)
180
180
        self.assertEqual(u'commit 1', rev.message)
182
182
    def test_commit_message_supplied(self):
183
183
        builder = BranchBuilder(self.get_transport().clone('foo'))
184
184
        rev_id = builder.build_snapshot(None,
185
 
            [('add', (u'', None, 'directory', None))],
186
 
            message=u'Foo')
 
185
                                        [('add', (u'', None, 'directory', None))],
 
186
                                        message=u'Foo')
187
187
        branch = builder.get_branch()
188
188
        rev = branch.repository.get_revision(rev_id)
189
189
        self.assertEqual(u'Foo', rev.message)
191
191
    def test_commit_message_callback(self):
192
192
        builder = BranchBuilder(self.get_transport().clone('foo'))
193
193
        rev_id = builder.build_snapshot(None,
194
 
            [('add', (u'', None, 'directory', None))],
195
 
            message_callback=lambda x:u'Foo')
 
194
                                        [('add', (u'', None, 'directory', None))],
 
195
                                        message_callback=lambda x: u'Foo')
196
196
        branch = builder.get_branch()
197
197
        rev = branch.repository.get_revision(rev_id)
198
198
        self.assertEqual(u'Foo', rev.message)
200
200
    def test_modify_file(self):
201
201
        builder = self.build_a_rev()
202
202
        rev_id2 = builder.build_snapshot(None,
203
 
            [('modify', ('a', b'new\ncontent\n'))],
204
 
            revision_id=b'B-id')
 
203
                                         [('modify', ('a', b'new\ncontent\n'))],
 
204
                                         revision_id=b'B-id')
205
205
        self.assertEqual(b'B-id', rev_id2)
206
206
        branch = builder.get_branch()
207
207
        rev_tree = branch.repository.revision_tree(rev_id2)
213
213
    def test_delete_file(self):
214
214
        builder = self.build_a_rev()
215
215
        rev_id2 = builder.build_snapshot(None,
216
 
            [('unversion', 'a')], revision_id=b'B-id')
 
216
                                         [('unversion', 'a')], revision_id=b'B-id')
217
217
        self.assertEqual(b'B-id', rev_id2)
218
218
        branch = builder.get_branch()
219
219
        rev_tree = branch.repository.revision_tree(rev_id2)
224
224
    def test_delete_directory(self):
225
225
        builder = self.build_a_rev()
226
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')
 
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',
 
230
                                                      'directory', None)),
 
231
                                             ('add', ('b/d/e', b'e-id',
 
232
                                                      'file', b'eff\n')),
 
233
                                          ], revision_id=b'B-id')
232
234
        rev_tree = builder.get_branch().repository.revision_tree(b'B-id')
233
235
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
234
236
                              (u'a', b'a-id', 'file'),
238
240
                              (u'b/d/e', b'e-id', 'file')], rev_tree)
239
241
        # Removing a directory removes all child dirs
240
242
        builder.build_snapshot(
241
 
                None, [('unversion', 'b')],
242
 
                revision_id=b'C-id')
 
243
            None, [('unversion', 'b')],
 
244
            revision_id=b'C-id')
243
245
        rev_tree = builder.get_branch().repository.revision_tree(b'C-id')
244
246
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
245
247
                              (u'a', b'a-id', 'file'),
246
 
                             ], rev_tree)
 
248
                              ], rev_tree)
247
249
 
248
250
    def test_unknown_action(self):
249
251
        builder = self.build_a_rev()
250
252
        e = self.assertRaises(ValueError,
251
 
            builder.build_snapshot, None, [('weirdo', ('foo',))],
252
 
            revision_id=b'B-id')
 
253
                              builder.build_snapshot, None, [
 
254
                                  ('weirdo', ('foo',))],
 
255
                              revision_id=b'B-id')
253
256
        self.assertEqual('Unknown build action: "weirdo"', str(e))
254
257
 
255
258
    def test_rename(self):
256
259
        builder = self.build_a_rev()
257
260
        builder.build_snapshot(None,
258
 
            [('rename', ('a', 'b'))], revision_id=b'B-id')
 
261
                               [('rename', ('a', 'b'))], revision_id=b'B-id')
259
262
        rev_tree = builder.get_branch().repository.revision_tree(b'B-id')
260
263
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
261
264
                              (u'b', b'a-id', 'file')], rev_tree)
263
266
    def test_rename_into_subdir(self):
264
267
        builder = self.build_a_rev()
265
268
        builder.build_snapshot(None,
266
 
            [('add', ('dir', b'dir-id', 'directory', None)),
267
 
             ('rename', ('a', 'dir/a'))], revision_id=b'B-id')
 
269
                               [('add', ('dir', b'dir-id', 'directory', None)),
 
270
                                ('rename', ('a', 'dir/a'))], revision_id=b'B-id')
268
271
        rev_tree = builder.get_branch().repository.revision_tree(b'B-id')
269
272
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
270
273
                              (u'dir', b'dir-id', 'directory'),
273
276
    def test_rename_out_of_unversioned_subdir(self):
274
277
        builder = self.build_a_rev()
275
278
        builder.build_snapshot(None,
276
 
            [('add', ('dir', b'dir-id', 'directory', None)),
277
 
             ('rename', ('a', 'dir/a'))],
278
 
            revision_id=b'B-id')
 
279
                               [('add', ('dir', b'dir-id', 'directory', None)),
 
280
                                ('rename', ('a', 'dir/a'))],
 
281
                               revision_id=b'B-id')
279
282
        builder.build_snapshot(None,
280
 
            [('rename', ('dir/a', 'a')),
281
 
             ('unversion', 'dir')], revision_id=b'C-id')
 
283
                               [('rename', ('dir/a', 'a')),
 
284
                                ('unversion', 'dir')], revision_id=b'C-id')
282
285
        rev_tree = builder.get_branch().repository.revision_tree(b'C-id')
283
286
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
284
287
                              (u'a', b'a-id', 'file')], rev_tree)
288
291
        builder.start_series()
289
292
        self.addCleanup(builder.finish_series)
290
293
        builder.build_snapshot([b'A-id'],
291
 
            [('modify', ('a', b'new\ncontent\n'))],
292
 
            revision_id=b'B-id')
 
294
                               [('modify', ('a', b'new\ncontent\n'))],
 
295
                               revision_id=b'B-id')
293
296
        builder.build_snapshot([b'A-id'],
294
 
            [('add', ('c', b'c-id', 'file', b'alt\ncontent\n'))],
295
 
            revision_id=b'C-id')
 
297
                               [('add', ('c', b'c-id', 'file', b'alt\ncontent\n'))],
 
298
                               revision_id=b'C-id')
296
299
        # We should now have a graph:
297
300
        #   A
298
301
        #   |\
304
307
        b_tree = repo.revision_tree(b'B-id')
305
308
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
306
309
                              (u'a', b'a-id', 'file'),
307
 
                             ], b_tree)
 
310
                              ], b_tree)
308
311
        self.assertEqual(b'new\ncontent\n', b_tree.get_file_text('a'))
309
312
 
310
313
        # We should still be using the content from A in C, not from B
312
315
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
313
316
                              (u'a', b'a-id', 'file'),
314
317
                              (u'c', b'c-id', 'file'),
315
 
                             ], c_tree)
 
318
                              ], c_tree)
316
319
        self.assertEqual(b'contents', c_tree.get_file_text('a'))
317
320
        self.assertEqual(b'alt\ncontent\n', c_tree.get_file_text('c'))
318
321
 
321
324
        builder.start_series()
322
325
        self.addCleanup(builder.finish_series)
323
326
        builder.build_snapshot([b'A-id'],
324
 
            [('add', ('b', b'b-id', 'file', b'b\ncontent\n'))],
325
 
            revision_id=b'B-id')
 
327
                               [('add', ('b', b'b-id', 'file', b'b\ncontent\n'))],
 
328
                               revision_id=b'B-id')
326
329
        builder.build_snapshot([b'A-id'],
327
 
            [('add', ('c', b'c-id', 'file', b'alt\ncontent\n'))],
328
 
            revision_id=b'C-id')
 
330
                               [('add', ('c', b'c-id', 'file', b'alt\ncontent\n'))],
 
331
                               revision_id=b'C-id')
329
332
        builder.build_snapshot([b'B-id', b'C-id'], [], revision_id=b'D-id')
330
333
        repo = builder.get_branch().repository
331
334
        self.assertEqual({b'B-id': (b'A-id',), b'C-id': (b'A-id',),
337
340
        self.assertTreeShape([(u'', b'a-root-id', 'directory'),
338
341
                              (u'a', b'a-id', 'file'),
339
342
                              (u'b', b'b-id', 'file'),
340
 
                             ], d_tree)
 
343
                              ], d_tree)
341
344
 
342
345
    def test_set_merge_parent_and_contents(self):
343
346
        builder = self.build_a_rev()
344
347
        builder.start_series()
345
348
        self.addCleanup(builder.finish_series)
346
349
        builder.build_snapshot([b'A-id'],
347
 
            [('add', ('b', b'b-id', 'file', b'b\ncontent\n'))],
348
 
            revision_id=b'B-id')
 
350
                               [('add', ('b', b'b-id', 'file', b'b\ncontent\n'))],
 
351
                               revision_id=b'B-id')
349
352
        builder.build_snapshot([b'A-id'],
350
 
            [('add', ('c', b'c-id', 'file', b'alt\ncontent\n'))],
351
 
            revision_id=b'C-id')
 
353
                               [('add', ('c', b'c-id', 'file', b'alt\ncontent\n'))],
 
354
                               revision_id=b'C-id')
352
355
        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')
 
356
                               [('add', ('c', b'c-id', 'file', b'alt\ncontent\n'))],
 
357
                               revision_id=b'D-id')
355
358
        repo = builder.get_branch().repository
356
359
        self.assertEqual({b'B-id': (b'A-id',), b'C-id': (b'A-id',),
357
360
                          b'D-id': (b'B-id', b'C-id')},
361
364
                              (u'a', b'a-id', 'file'),
362
365
                              (u'b', b'b-id', 'file'),
363
366
                              (u'c', b'c-id', 'file'),
364
 
                             ], d_tree)
 
367
                              ], d_tree)
365
368
        # Because we copied the exact text into *this* tree, the 'c' file
366
369
        # should look like it was not modified in the merge
367
370
        self.assertEqual(b'C-id', d_tree.get_file_revision('c'))
371
374
        builder.start_series()
372
375
        self.addCleanup(builder.finish_series)
373
376
        builder.build_snapshot([],
374
 
            [('add', ('', None, 'directory', None))],
375
 
            revision_id=b'B-id')
 
377
                               [('add', ('', None, 'directory', None))],
 
378
                               revision_id=b'B-id')
376
379
        # We should now have a graph:
377
380
        #   A B
378
381
        # And not A => B
379
382
        repo = builder.get_branch().repository
380
383
        self.assertEqual({b'A-id': (_mod_revision.NULL_REVISION,),
381
 
                          b'B-id': (_mod_revision.NULL_REVISION,),},
 
384
                          b'B-id': (_mod_revision.NULL_REVISION,), },
382
385
                         repo.get_parent_map([b'A-id', b'B-id']))
383
386
 
384
387
    def test_start_finish_series(self):
398
401
        builder.start_series()
399
402
        try:
400
403
            builder.build_snapshot([b'ghost'],
401
 
                [('add', ('', b'ROOT_ID', 'directory', ''))],
402
 
                allow_leftmost_as_ghost=True, revision_id=b'tip')
 
404
                                   [('add', ('', b'ROOT_ID', 'directory', ''))],
 
405
                                   allow_leftmost_as_ghost=True, revision_id=b'tip')
403
406
        finally:
404
407
            builder.finish_series()
405
408
        b = builder.get_branch()
406
409
        b.lock_read()
407
410
        self.addCleanup(b.unlock)
408
411
        self.assertEqual((b'ghost',),
409
 
            b.repository.get_graph().get_parent_map([b'tip'])[b'tip'])
 
412
                         b.repository.get_graph().get_parent_map([b'tip'])[b'tip'])
410
413
 
411
414
    def test_unversion_root_add_new_root(self):
412
415
        builder = BranchBuilder(self.get_transport().clone('foo'))
413
416
        builder.start_series()
414
417
        builder.build_snapshot(None,
415
 
            [('add', ('', b'TREE_ROOT', 'directory', ''))],
416
 
            revision_id=b'rev-1')
 
418
                               [('add', ('', b'TREE_ROOT', 'directory', ''))],
 
419
                               revision_id=b'rev-1')
417
420
        builder.build_snapshot(None,
418
 
            [('unversion', ''),
419
 
             ('add', ('', b'my-root', 'directory', ''))],
420
 
            revision_id=b'rev-2')
 
421
                               [('unversion', ''),
 
422
                                ('add', ('', b'my-root', 'directory', ''))],
 
423
                               revision_id=b'rev-2')
421
424
        builder.finish_series()
422
425
        rev_tree = builder.get_branch().repository.revision_tree(b'rev-2')
423
426
        self.assertTreeShape([(u'', b'my-root', 'directory')], rev_tree)
427
430
        builder = BranchBuilder(self.get_transport().clone('foo'))
428
431
        builder.start_series()
429
432
        builder.build_snapshot(None,
430
 
            [('add', ('', b'TREE_ROOT', 'directory', ''))],
431
 
            revision_id=b'rev-1')
 
433
                               [('add', ('', b'TREE_ROOT', 'directory', ''))],
 
434
                               revision_id=b'rev-1')
432
435
        builder.build_snapshot(None, [('flush', None)], revision_id=b'rev-2')
433
436
        builder.finish_series()
434
437
        rev_tree = builder.get_branch().repository.revision_tree(b'rev-2')
441
444
        builder = BranchBuilder(self.get_transport().clone('foo'))
442
445
        builder.start_series()
443
446
        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
                               [('add', (u'', b'a-root-id', 'directory', None)),
 
448
                                ('add', (u'a', b'a-id', 'file', b'content\n'))],
 
449
                               revision_id=b'A-id')
447
450
        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')
 
451
                               [('unversion', 'a'),
 
452
                                ('flush', None),
 
453
                                   ('add', (u'a', b'a-id', 'directory', None))],
 
454
                               revision_id=b'B-id')
452
455
        builder.finish_series()
453
456
        rev_tree = builder.get_branch().repository.revision_tree(b'B-id')
454
457
        self.assertTreeShape(
462
465
        builder = BranchBuilder(self.get_transport().clone('foo'))
463
466
        builder.start_series()
464
467
        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
                               [('add', (u'', b'orig-root', 'directory', None)),
 
469
                                ('add', (u'dir', b'dir-id', 'directory', None))],
 
470
                               revision_id=b'A-id')
468
471
        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')
 
472
                               [('unversion', ''),  # implicitly unversions all children
 
473
                                ('flush', None),
 
474
                                   ('add', (u'', b'dir-id', 'directory', None))],
 
475
                               revision_id=b'B-id')
473
476
        builder.finish_series()
474
477
        rev_tree = builder.get_branch().repository.revision_tree(b'B-id')
475
478
        self.assertTreeShape([(u'', b'dir-id', 'directory')], rev_tree)
476