50
44
self.tree = self.make_branch_and_tree('tree')
51
45
self.build_tree(['tree/a'])
52
self.tree.lock_write()
53
self.addCleanup(self.tree.unlock)
54
46
self.tree.add(['a'])
55
47
self.tree.commit('a', rev_id='r1')
57
49
self.tree2 = self.tree.bzrdir.sprout('tree2').open_workingtree()
58
50
self.tree2.commit('alt', rev_id='alt_r2')
60
self.tree.merge_from_branch(self.tree2.branch)
52
self.tree.branch.repository.fetch(self.tree2.branch.repository,
54
self.tree.set_pending_merges(['alt_r2'])
61
55
self.tree.commit('second', rev_id='r2')
63
57
def get_in_history(self, revision_spec):
66
60
def assertInHistoryIs(self, exp_revno, exp_revision_id, revision_spec):
67
61
rev_info = self.get_in_history(revision_spec)
68
62
self.assertEqual(exp_revno, rev_info.revno,
69
'Revision spec: %r returned wrong revno: %r != %r'
63
'Revision spec: %s returned wrong revno: %s != %s'
70
64
% (revision_spec, exp_revno, rev_info.revno))
71
65
self.assertEqual(exp_revision_id, rev_info.rev_id,
72
'Revision spec: %r returned wrong revision id:'
66
'Revision spec: %s returned wrong revision id:'
74
68
% (revision_spec, exp_revision_id, rev_info.rev_id))
76
def assertInvalid(self, revision_spec, extra='',
77
invalid_as_revision_id=True):
70
def assertInvalid(self, revision_spec, extra=''):
79
72
self.get_in_history(revision_spec)
80
73
except errors.InvalidRevisionSpec, e:
81
74
self.assertEqual(revision_spec, e.spec)
82
75
self.assertEqual(extra, e.extra)
84
self.fail('Expected InvalidRevisionSpec to be raised for'
85
' %r.in_history' % (revision_spec,))
86
if invalid_as_revision_id:
88
spec = RevisionSpec.from_string(revision_spec)
89
spec.as_revision_id(self.tree.branch)
90
except errors.InvalidRevisionSpec, e:
91
self.assertEqual(revision_spec, e.spec)
92
self.assertEqual(extra, e.extra)
94
self.fail('Expected InvalidRevisionSpec to be raised for'
95
' %r.as_revision_id' % (revision_spec,))
97
def assertAsRevisionId(self, revision_id, revision_spec):
98
"""Calling as_revision_id() should return the specified id."""
99
spec = RevisionSpec.from_string(revision_spec)
100
self.assertEqual(revision_id,
101
spec.as_revision_id(self.tree.branch))
104
class RevisionSpecMatchOnTrap(RevisionSpec):
106
def _match_on(self, branch, revs):
107
self.last_call = (branch, revs)
108
return super(RevisionSpecMatchOnTrap, self)._match_on(branch, revs)
111
class TestRevisionSpecBase(TestRevisionSpec):
113
def test_wants_revision_history(self):
114
# If wants_revision_history = True, then _match_on should get the
115
# branch revision history
116
spec = RevisionSpecMatchOnTrap('foo', _internal=True)
117
spec.in_history(self.tree.branch)
119
self.assertEqual((self.tree.branch, ['r1' ,'r2']),
122
def test_wants_no_revision_history(self):
123
# If wants_revision_history = False, then _match_on should get None for
124
# the branch revision history
125
spec = RevisionSpecMatchOnTrap('foo', _internal=True)
126
spec.wants_revision_history = False
127
spec.in_history(self.tree.branch)
129
self.assertEqual((self.tree.branch, None), spec.last_call)
77
self.fail('Expected InvalidRevisionSpec to be raised for %s'
133
81
class TestOddRevisionSpec(TestRevisionSpec):
134
82
"""Test things that aren't normally thought of as revision specs"""
136
84
def test_none(self):
137
self.assertInHistoryIs(None, None, None)
85
self.assertInHistoryIs(0, None, None)
139
87
def test_object(self):
140
88
self.assertRaises(TypeError, RevisionSpec.from_string, object())
292
240
self.assertEqual((2, 'b@r-0-2'),
293
241
spec_in_history('revno:2:b/', None))
295
def test_as_revision_id(self):
296
self.assertAsRevisionId('null:', '0')
297
self.assertAsRevisionId('r1', '1')
298
self.assertAsRevisionId('r2', '2')
299
self.assertAsRevisionId('r1', '-2')
300
self.assertAsRevisionId('r2', '-1')
301
self.assertAsRevisionId('alt_r2', '1.1.1')
304
245
class TestRevisionSpec_revid(TestRevisionSpec):
329
270
revision_id='alt_r3')
330
271
self.assertInHistoryIs(None, 'alt_r3', 'revid:alt_r3')
332
def test_unicode(self):
333
"""We correctly convert a unicode ui string to an encoded revid."""
334
revision_id = u'\N{SNOWMAN}'.encode('utf-8')
335
self.tree.commit('unicode', rev_id=revision_id)
336
self.assertInHistoryIs(3, revision_id, u'revid:\N{SNOWMAN}')
337
self.assertInHistoryIs(3, revision_id, 'revid:' + revision_id)
339
def test_as_revision_id(self):
340
self.assertAsRevisionId('r1', 'revid:r1')
341
self.assertAsRevisionId('r2', 'revid:r2')
342
self.assertAsRevisionId('alt_r2', 'revid:alt_r2')
345
274
class TestRevisionSpec_last(TestRevisionSpec):
347
276
def test_positive(self):
348
277
self.assertInHistoryIs(2, 'r2', 'last:1')
349
278
self.assertInHistoryIs(1, 'r1', 'last:2')
350
self.assertInHistoryIs(0, 'null:', 'last:3')
279
self.assertInHistoryIs(0, None, 'last:3')
352
281
def test_empty(self):
353
282
self.assertInHistoryIs(2, 'r2', 'last:')
405
330
new_tree.commit('first', rev_id='new_r1')
406
331
self.tree.branch.repository.fetch(new_tree.branch.repository,
407
332
revision_id='new_r1')
408
self.assertInHistoryIs(0, 'null:', 'before:revid:new_r1')
410
def test_as_revision_id(self):
411
self.assertAsRevisionId('r1', 'before:revid:r2')
412
self.assertAsRevisionId('r1', 'before:2')
413
self.assertAsRevisionId('r1', 'before:1.1.1')
414
self.assertAsRevisionId('r1', 'before:revid:alt_r2')
333
self.assertInHistoryIs(0, None, 'before:revid:new_r1')
417
336
class TestRevisionSpec_tag(TestRevisionSpec):
419
def make_branch_and_tree(self, relpath):
420
# override format as the default one may not support tags
421
return TestRevisionSpec.make_branch_and_tree(
422
self, relpath, format='dirstate-tags')
424
def test_from_string_tag(self):
425
spec = RevisionSpec.from_string('tag:bzr-0.14')
426
self.assertIsInstance(spec, RevisionSpec_tag)
427
self.assertEqual(spec.spec, 'bzr-0.14')
429
def test_lookup_tag(self):
430
self.tree.branch.tags.set_tag('bzr-0.14', 'r1')
431
self.assertInHistoryIs(1, 'r1', 'tag:bzr-0.14')
432
self.tree.branch.tags.set_tag('null_rev', 'null:')
433
self.assertInHistoryIs(0, 'null:', 'tag:null_rev')
435
def test_failed_lookup(self):
436
# tags that don't exist give a specific message: arguably we should
437
# just give InvalidRevisionSpec but I think this is more helpful
438
self.assertRaises(errors.NoSuchTag,
440
'tag:some-random-tag')
442
def test_as_revision_id(self):
443
self.tree.branch.tags.set_tag('my-tag', 'r2')
444
self.tree.branch.tags.set_tag('null_rev', 'null:')
445
self.assertAsRevisionId('r2', 'tag:my-tag')
446
self.assertAsRevisionId('null:', 'tag:null_rev')
447
self.assertAsRevisionId('r1', 'before:tag:my-tag')
338
def test_invalid(self):
339
self.assertInvalid('tag:foo', extra='\ntag: namespace registered,'
340
' but not implemented')
450
343
class TestRevisionSpec_date(TestRevisionSpec):
564
451
new_tree = self.make_branch_and_tree('new_tree')
565
452
self.assertRaises(errors.NoCommits,
566
453
self.get_in_history, 'branch:new_tree')
568
def test_as_revision_id(self):
569
self.assertAsRevisionId('alt_r2', 'branch:tree2')
572
class TestRevisionSpec_submit(TestRevisionSpec):
574
def test_submit_branch(self):
575
# Common ancestor of trees is 'alt_r2'
576
self.assertRaises(errors.NoSubmitBranch, self.get_in_history,
578
self.tree.branch.set_parent('../tree2')
579
self.assertInHistoryIs(None, 'alt_r2', 'submit:')
580
self.tree.branch.set_parent('bogus')
581
self.assertRaises(errors.NotBranchError, self.get_in_history,
583
# submit branch overrides parent branch
584
self.tree.branch.set_submit_branch('tree2')
585
self.assertInHistoryIs(None, 'alt_r2', 'submit:')
587
def test_as_revision_id(self):
588
self.tree.branch.set_submit_branch('tree2')
589
self.assertAsRevisionId('alt_r2', 'branch:tree2')