158
158
timezone = 5 * 3600
159
159
self.assertRaises(errors.NoMergeSource, self.make_merge_directive,
160
b'example:', b'sha', time, timezone, 'http://example.com')
160
b'example:', b'sha', time, timezone, 'http://example.com')
161
161
self.assertRaises(errors.NoMergeSource, self.make_merge_directive,
162
b'example:', b'sha', time, timezone, 'http://example.com',
162
b'example:', b'sha', time, timezone, 'http://example.com',
164
164
self.make_merge_directive(b'example:', b'sha', time, timezone,
165
'http://example.com', source_branch='http://example.org')
165
'http://example.com', source_branch='http://example.org')
166
166
md = self.make_merge_directive(b'null:', b'sha', time, timezone,
167
'http://example.com', patch=b'blah', patch_type='bundle')
167
'http://example.com', patch=b'blah', patch_type='bundle')
168
168
self.assertIs(None, md.source_branch)
169
169
md2 = self.make_merge_directive(b'null:', b'sha', time, timezone,
170
'http://example.com', patch=b'blah', patch_type='bundle',
170
'http://example.com', patch=b'blah', patch_type='bundle',
172
172
self.assertEqual('bar', md2.source_branch)
174
174
def test_serialization(self):
177
177
md = self.make_merge_directive(b'example:', b'sha', time, timezone,
178
'http://example.com', patch=b'booga', patch_type='bundle')
178
'http://example.com', patch=b'booga', patch_type='bundle')
179
179
self.assertEqualDiff(self.OUTPUT1, b''.join(md.to_lines()))
180
180
md = self.make_merge_directive(b'example:', b'sha', time, timezone,
181
'http://example.com', source_branch="http://example.org",
182
patch=b'booga', patch_type='diff', message="Hi mom!")
181
'http://example.com', source_branch="http://example.org",
182
patch=b'booga', patch_type='diff', message="Hi mom!")
183
183
self.assertEqualDiff(self.OUTPUT2, b''.join(md.to_lines()))
185
185
def test_deserialize_junk(self):
208
208
timezone = 7.5 * 3600
209
209
md = self.make_merge_directive(b'example:', b'sha', time, timezone,
210
'http://example.com', source_branch="http://example.org",
211
patch=b'booga', patch_type='diff')
210
'http://example.com', source_branch="http://example.org",
211
patch=b'booga', patch_type='diff')
212
212
md2 = merge_directive.MergeDirective.from_lines(md.to_lines())
213
213
self.assertEqual(b'example:', md2.revision_id)
214
214
self.assertIsInstance(md2.revision_id, bytes)
245
245
OUTPUT2 = OUTPUT2
247
247
def make_merge_directive(self, revision_id, testament_sha1, time, timezone,
248
target_branch, patch=None, patch_type=None,
249
source_branch=None, message=None):
248
target_branch, patch=None, patch_type=None,
249
source_branch=None, message=None):
250
250
return merge_directive.MergeDirective(revision_id, testament_sha1,
251
time, timezone, target_branch, patch, patch_type,
252
source_branch, message)
251
time, timezone, target_branch, patch, patch_type,
252
source_branch, message)
255
255
def set_bundle(md, value):
261
261
self.assertRaises(errors.PatchMissing, merge_directive.MergeDirective,
262
b'example:', b'sha', time, timezone, 'http://example.com',
262
b'example:', b'sha', time, timezone, 'http://example.com',
264
264
md = merge_directive.MergeDirective(b'example:', b'sha1', time, timezone,
265
'http://example.com', source_branch="http://example.org",
266
patch=b'', patch_type='diff')
265
'http://example.com', source_branch="http://example.org",
266
patch=b'', patch_type='diff')
267
267
self.assertEqual(md.patch, b'')
277
277
OUTPUT2 = OUTPUT2_2
279
279
def make_merge_directive(self, revision_id, testament_sha1, time, timezone,
280
target_branch, patch=None, patch_type=None,
281
source_branch=None, message=None, base_revision_id=b'null:'):
280
target_branch, patch=None, patch_type=None,
281
source_branch=None, message=None, base_revision_id=b'null:'):
282
282
if patch_type == 'bundle':
287
287
return merge_directive.MergeDirective2(revision_id, testament_sha1,
288
time, timezone, target_branch, patch, source_branch, message,
289
bundle, base_revision_id)
288
time, timezone, target_branch, patch, source_branch, message,
289
bundle, base_revision_id)
292
292
def set_bundle(md, value):
371
372
tree_a, tree_b, branch_c = self.make_trees()
372
373
tree_d = self.make_branch_and_tree('tree_d')
373
374
md2 = self.from_objects(tree_a.branch.repository, b'rev2a', 500, 120,
374
tree_d.branch.base, patch_type='diff',
375
public_branch=tree_a.branch.base)
375
tree_d.branch.base, patch_type='diff',
376
public_branch=tree_a.branch.base)
377
378
def test_disk_name(self):
378
379
tree_a, tree_b, branch_c = self.make_trees()
379
380
tree_a.branch.nick = 'fancy <name>'
380
381
md = self.from_objects(tree_a.branch.repository, b'rev2a', 500, 120,
382
383
self.assertEqual('fancy-name-2', md.get_disk_name(tree_a.branch))
384
385
def test_disk_name_old_revno(self):
385
386
tree_a, tree_b, branch_c = self.make_trees()
386
387
tree_a.branch.nick = 'fancy-name'
387
388
md = self.from_objects(tree_a.branch.repository, b'rev1', 500, 120,
389
390
self.assertEqual('fancy-name-1', md.get_disk_name(tree_a.branch))
391
392
def test_generate_patch(self):
392
393
tree_a, tree_b, branch_c = self.make_trees()
393
394
md2 = self.from_objects(tree_a.branch.repository, b'rev2a', 500, 120,
394
tree_b.branch.base, patch_type='diff',
395
public_branch=tree_a.branch.base)
395
tree_b.branch.base, patch_type='diff',
396
public_branch=tree_a.branch.base)
396
397
self.assertNotContainsRe(md2.patch, b'Bazaar revision bundle')
397
398
self.assertContainsRe(md2.patch, b'\\+content_c')
398
399
self.assertNotContainsRe(md2.patch, b'\\+\\+\\+ b/')
401
402
def test_public_branch(self):
402
403
tree_a, tree_b, branch_c = self.make_trees()
403
404
self.assertRaises(errors.PublicBranchOutOfDate,
404
self.from_objects, tree_a.branch.repository, b'rev2a', 500, 144,
405
tree_b.branch.base, public_branch=branch_c.base, patch_type='diff')
405
self.from_objects, tree_a.branch.repository, b'rev2a', 500, 144,
406
tree_b.branch.base, public_branch=branch_c.base, patch_type='diff')
406
407
self.assertRaises(errors.PublicBranchOutOfDate,
407
self.from_objects, tree_a.branch.repository, b'rev2a', 500, 144,
408
tree_b.branch.base, public_branch=branch_c.base, patch_type=None)
408
self.from_objects, tree_a.branch.repository, b'rev2a', 500, 144,
409
tree_b.branch.base, public_branch=branch_c.base, patch_type=None)
409
410
# public branch is not checked if patch format is bundle.
410
411
md1 = self.from_objects(tree_a.branch.repository, b'rev2a', 500, 144,
411
tree_b.branch.base, public_branch=branch_c.base)
412
tree_b.branch.base, public_branch=branch_c.base)
412
413
# public branch is provided with a bundle, despite possibly being out
413
414
# of date, because it's not required if a bundle is present.
414
415
self.assertEqual(md1.source_branch, branch_c.base)
415
416
# Once we update the public branch, we can generate a diff.
416
417
branch_c.pull(tree_a.branch)
417
418
md3 = self.from_objects(tree_a.branch.repository, b'rev2a', 500, 144,
418
tree_b.branch.base, patch_type=None, public_branch=branch_c.base)
419
tree_b.branch.base, patch_type=None, public_branch=branch_c.base)
420
421
def test_use_public_submit_branch(self):
421
422
tree_a, tree_b, branch_c = self.make_trees()
422
423
branch_c.pull(tree_a.branch)
423
424
md = self.from_objects(tree_a.branch.repository, b'rev2a', 500, 144,
424
tree_b.branch.base, patch_type=None, public_branch=branch_c.base)
425
tree_b.branch.base, patch_type=None, public_branch=branch_c.base)
425
426
self.assertEqual(md.target_branch, tree_b.branch.base)
426
427
tree_b.branch.set_public_branch('http://example.com')
427
428
md2 = self.from_objects(
428
tree_a.branch.repository, b'rev2a', 500, 144, tree_b.branch.base,
429
patch_type=None, public_branch=branch_c.base)
429
tree_a.branch.repository, b'rev2a', 500, 144, tree_b.branch.base,
430
patch_type=None, public_branch=branch_c.base)
430
431
self.assertEqual(md2.target_branch, 'http://example.com')
432
433
def test_message(self):
433
434
tree_a, tree_b, branch_c = self.make_trees()
434
435
md3 = self.from_objects(tree_a.branch.repository, b'rev1', 500, 120,
435
tree_b.branch.base, patch_type=None, public_branch=branch_c.base,
436
message='Merge message')
436
tree_b.branch.base, patch_type=None, public_branch=branch_c.base,
437
message='Merge message')
438
439
self.assertIs(None, md3.patch)
439
440
self.assertEqual('Merge message', md3.message)
461
462
def test_signing(self):
464
466
class FakeBranch(object):
465
467
def get_config_stack(self):
467
469
md = self.make_merge_directive(b'example:', b'sha', time, timezone,
468
'http://example.com', source_branch="http://example.org",
469
patch=b'booga', patch_type='diff')
470
'http://example.com', source_branch="http://example.org",
471
patch=b'booga', patch_type='diff')
470
472
old_strategy = gpg.GPGStrategy
471
473
gpg.GPGStrategy = gpg.LoopbackGPGStrategy
480
482
def test_email(self):
481
483
tree_a, tree_b, branch_c = self.make_trees()
482
484
md = self.from_objects(tree_a.branch.repository, b'rev2a', 476, 60,
483
tree_b.branch.base, patch_type=None,
484
public_branch=tree_a.branch.base)
485
tree_b.branch.base, patch_type=None,
486
public_branch=tree_a.branch.base)
485
487
message = md.to_email('pqm@example.com', tree_a.branch)
486
488
self.assertContainsRe(message.as_string(), self.EMAIL1)
487
489
md.message = 'Commit of rev2a with special message'
491
493
def test_install_revisions_branch(self):
492
494
tree_a, tree_b, branch_c = self.make_trees()
493
495
md = self.from_objects(tree_a.branch.repository, b'rev2a', 500, 36,
494
tree_b.branch.base, patch_type=None,
495
public_branch=tree_a.branch.base)
496
tree_b.branch.base, patch_type=None,
497
public_branch=tree_a.branch.base)
496
498
self.assertFalse(tree_b.branch.repository.has_revision(b'rev2a'))
497
499
revision = md.install_revisions(tree_b.branch.repository)
498
500
self.assertEqual(b'rev2a', revision)
501
503
def test_get_merge_request(self):
502
504
tree_a, tree_b, branch_c = self.make_trees()
503
505
md = self.from_objects(tree_a.branch.repository, b'rev2a', 500, 36,
504
tree_b.branch.base, patch_type='bundle',
505
public_branch=tree_a.branch.base)
506
tree_b.branch.base, patch_type='bundle',
507
public_branch=tree_a.branch.base)
506
508
self.assertFalse(tree_b.branch.repository.has_revision(b'rev2a'))
507
509
md.install_revisions(tree_b.branch.repository)
508
510
base, revision, verified = md.get_merge_request(
516
518
self.assertEqual(b'rev2a', revision)
517
519
self.assertTrue(tree_b.branch.repository.has_revision(b'rev2a'))
518
520
md = self.from_objects(tree_a.branch.repository, b'rev2a', 500, 36,
519
tree_b.branch.base, patch_type=None,
520
public_branch=tree_a.branch.base)
521
tree_b.branch.base, patch_type=None,
522
public_branch=tree_a.branch.base)
521
523
base, revision, verified = md.get_merge_request(
522
524
tree_b.branch.repository)
523
525
if isinstance(md, merge_directive.MergeDirective):
527
529
self.assertEqual(b'rev1', base)
528
530
self.assertEqual('inapplicable', verified)
529
531
md = self.from_objects(tree_a.branch.repository, b'rev2a', 500, 36,
530
tree_b.branch.base, patch_type='diff',
531
public_branch=tree_a.branch.base)
532
tree_b.branch.base, patch_type='diff',
533
public_branch=tree_a.branch.base)
532
534
base, revision, verified = md.get_merge_request(
533
535
tree_b.branch.repository)
534
536
if isinstance(md, merge_directive.MergeDirective):
550
552
def test_install_revisions_bundle(self):
551
553
tree_a, tree_b, branch_c = self.make_trees()
552
554
md = self.from_objects(tree_a.branch.repository, b'rev2a', 500, 36,
553
tree_b.branch.base, patch_type='bundle',
554
public_branch=tree_a.branch.base)
555
tree_b.branch.base, patch_type='bundle',
556
public_branch=tree_a.branch.base)
555
557
self.assertFalse(tree_b.branch.repository.has_revision(b'rev2a'))
556
558
revision = md.install_revisions(tree_b.branch.repository)
557
559
self.assertEqual(b'rev2a', revision)
560
562
def test_get_target_revision_nofetch(self):
561
563
tree_a, tree_b, branch_c = self.make_trees()
562
564
tree_b.branch.fetch(tree_a.branch)
563
md = self.from_objects( tree_a.branch.repository, b'rev2a', 500, 36,
564
tree_b.branch.base, patch_type=None,
565
public_branch=tree_a.branch.base)
565
md = self.from_objects(tree_a.branch.repository, b'rev2a', 500, 36,
566
tree_b.branch.base, patch_type=None,
567
public_branch=tree_a.branch.base)
566
568
md.source_branch = '/dev/null'
567
569
revision = md.install_revisions(tree_b.branch.repository)
568
570
self.assertEqual(b'rev2a', revision)
582
584
self.build_tree_contents([('tree_a/file', b'content_q\ncontent_r\n')])
583
585
tree_a.commit('rev3a', rev_id=b'rev3a')
584
586
md = self.from_objects(tree_a.branch.repository, b'rev3a', 500, 36,
585
branch_c.base, base_revision_id=b'rev2a')
587
branch_c.base, base_revision_id=b'rev2a')
586
588
md.target_branch = self.get_url('not-a-branch')
587
589
self.assertRaises(errors.TargetNotBranch, md.install_revisions,
588
tree_b.branch.repository)
590
tree_b.branch.repository)
591
593
class TestMergeDirective1Branch(tests.TestCaseWithTransport,
592
TestMergeDirectiveBranch):
594
TestMergeDirectiveBranch):
593
595
"""Test merge directive format 1 with a branch"""
599
601
def from_objects(self, repository, revision_id, time, timezone,
600
target_branch, patch_type='bundle', local_target_branch=None,
601
public_branch=None, message=None, base_revision_id=None):
602
target_branch, patch_type='bundle', local_target_branch=None,
603
public_branch=None, message=None, base_revision_id=None):
602
604
if base_revision_id is not None:
603
605
raise tests.TestNotApplicable('This format does not support'
604
606
' explicit bases.')
605
607
with repository.lock_write():
606
608
return merge_directive.MergeDirective.from_objects(repository,
607
revision_id, time, timezone, target_branch, patch_type,
608
local_target_branch, public_branch, message)
609
revision_id, time, timezone, target_branch, patch_type,
610
local_target_branch, public_branch, message)
610
612
def make_merge_directive(self, revision_id, testament_sha1, time, timezone,
611
target_branch, patch=None, patch_type=None,
612
source_branch=None, message=None):
613
target_branch, patch=None, patch_type=None,
614
source_branch=None, message=None):
613
615
return merge_directive.MergeDirective(revision_id, testament_sha1,
614
time, timezone, target_branch, patch, patch_type,
615
source_branch, message)
616
time, timezone, target_branch, patch, patch_type,
617
source_branch, message)
618
620
class TestMergeDirective2Branch(tests.TestCaseWithTransport,
619
TestMergeDirectiveBranch):
621
TestMergeDirectiveBranch):
620
622
"""Test merge directive format 2 with a branch"""
622
624
EMAIL1 = EMAIL1_2
624
626
EMAIL2 = EMAIL2_2
626
628
def from_objects(self, repository, revision_id, time, timezone,
627
target_branch, patch_type='bundle', local_target_branch=None,
628
public_branch=None, message=None, base_revision_id=None):
629
target_branch, patch_type='bundle', local_target_branch=None,
630
public_branch=None, message=None, base_revision_id=None):
629
631
include_patch = (patch_type in ('bundle', 'diff'))
630
632
include_bundle = (patch_type == 'bundle')
631
633
self.assertTrue(patch_type in ('bundle', 'diff', None))
635
637
message, base_revision_id)
637
639
def make_merge_directive(self, revision_id, testament_sha1, time, timezone,
638
target_branch, patch=None, patch_type=None,
639
source_branch=None, message=None, base_revision_id=b'null:'):
640
target_branch, patch=None, patch_type=None,
641
source_branch=None, message=None, base_revision_id=b'null:'):
640
642
if patch_type == 'bundle':
645
647
return merge_directive.MergeDirective2(revision_id, testament_sha1,
646
time, timezone, target_branch, patch, source_branch, message,
647
bundle, base_revision_id)
648
time, timezone, target_branch, patch, source_branch, message,
649
bundle, base_revision_id)
649
651
def test_base_revision(self):
650
652
tree_a, tree_b, branch_c = self.make_trees()
651
653
md = self.from_objects(tree_a.branch.repository, b'rev2a', 500, 60,
652
tree_b.branch.base, patch_type='bundle',
653
public_branch=tree_a.branch.base, base_revision_id=None)
654
tree_b.branch.base, patch_type='bundle',
655
public_branch=tree_a.branch.base, base_revision_id=None)
654
656
self.assertEqual(b'rev1', md.base_revision_id)
655
657
md = self.from_objects(tree_a.branch.repository, b'rev2a', 500, 60,
656
tree_b.branch.base, patch_type='bundle',
657
public_branch=tree_a.branch.base, base_revision_id=b'null:')
658
tree_b.branch.base, patch_type='bundle',
659
public_branch=tree_a.branch.base, base_revision_id=b'null:')
658
660
self.assertEqual(b'null:', md.base_revision_id)
659
661
lines = md.to_lines()
660
662
md2 = merge_directive.MergeDirective.from_lines(lines)
663
665
def test_patch_verification(self):
664
666
tree_a, tree_b, branch_c = self.make_trees()
665
667
md = self.from_objects(tree_a.branch.repository, b'rev2a', 500, 60,
666
tree_b.branch.base, patch_type='bundle',
667
public_branch=tree_a.branch.base)
668
tree_b.branch.base, patch_type='bundle',
669
public_branch=tree_a.branch.base)
668
670
lines = md.to_lines()
669
671
md2 = merge_directive.MergeDirective.from_lines(lines)
670
672
md2._verify_patch(tree_a.branch.repository)