50
50
self.tree = self.make_branch_and_tree('tree')
51
51
self.build_tree(['tree/a'])
52
self.tree.lock_write()
53
self.addCleanup(self.tree.unlock)
54
52
self.tree.add(['a'])
55
53
self.tree.commit('a', rev_id='r1')
57
55
self.tree2 = self.tree.bzrdir.sprout('tree2').open_workingtree()
58
56
self.tree2.commit('alt', rev_id='alt_r2')
60
self.tree.merge_from_branch(self.tree2.branch)
58
self.tree.branch.repository.fetch(self.tree2.branch.repository,
60
self.tree.set_pending_merges(['alt_r2'])
61
61
self.tree.commit('second', rev_id='r2')
63
63
def get_in_history(self, revision_spec):
74
74
% (revision_spec, exp_revision_id, rev_info.rev_id))
76
def assertInvalid(self, revision_spec, extra='',
77
invalid_as_revision_id=True):
76
def assertInvalid(self, revision_spec, extra=''):
79
78
self.get_in_history(revision_spec)
80
79
except errors.InvalidRevisionSpec, e:
81
80
self.assertEqual(revision_spec, e.spec)
82
81
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)
83
self.fail('Expected InvalidRevisionSpec to be raised for %s'
133
87
class TestOddRevisionSpec(TestRevisionSpec):
134
88
"""Test things that aren't normally thought of as revision specs"""
136
90
def test_none(self):
137
self.assertInHistoryIs(None, None, None)
91
self.assertInHistoryIs(0, None, None)
139
93
def test_object(self):
140
94
self.assertRaises(TypeError, RevisionSpec.from_string, object())
292
246
self.assertEqual((2, 'b@r-0-2'),
293
247
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
251
class TestRevisionSpec_revid(TestRevisionSpec):
336
283
self.assertInHistoryIs(3, revision_id, u'revid:\N{SNOWMAN}')
337
284
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
287
class TestRevisionSpec_last(TestRevisionSpec):
407
345
revision_id='new_r1')
408
346
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')
417
349
class TestRevisionSpec_tag(TestRevisionSpec):
419
351
def make_branch_and_tree(self, relpath):
420
352
# override format as the default one may not support tags
421
return TestRevisionSpec.make_branch_and_tree(
422
self, relpath, format='dirstate-tags')
353
control = bzrdir.BzrDir.create(relpath)
354
control.create_repository()
355
branch.BzrBranchExperimental.initialize(control)
356
return control.create_workingtree()
424
358
def test_from_string_tag(self):
425
359
spec = RevisionSpec.from_string('tag:bzr-0.14')
429
363
def test_lookup_tag(self):
430
364
self.tree.branch.tags.set_tag('bzr-0.14', 'r1')
431
365
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
367
def test_failed_lookup(self):
436
368
# tags that don't exist give a specific message: arguably we should