33
33
from bzrlib.branch import Branch
34
from bzrlib.clone import copy_branch
35
34
from bzrlib.errors import BzrCommandError
36
35
from bzrlib.osutils import has_symlinks
37
from bzrlib.selftest import TestCaseInTempDir, BzrTestBase
36
from bzrlib.tests import TestCaseWithTransport
38
37
from bzrlib.annotate import annotate_file
41
class TestAnnotate(TestCaseInTempDir):
40
class TestAnnotate(TestCaseWithTransport):
43
43
super(TestAnnotate, self).setUp()
44
b = Branch.initialize('.')
45
self.build_tree_contents([('hello.txt', 'my helicopter\n')])
48
committer='test@user')
44
wt = self.make_branch_and_tree('.')
46
self.build_tree_contents([('hello.txt', 'my helicopter\n'),
47
('nomail.txt', 'nomail\n')])
49
wt.commit('add hello', committer='test@user')
50
wt.add(['nomail.txt'])
51
wt.commit('add nomail', committer='no mail')
52
file('hello.txt', 'ab').write('your helicopter')
53
wt.commit('mod hello', committer='user@test')
50
55
def test_help_annotate(self):
51
56
"""Annotate command exists"""
56
61
self.assertEquals(err, '')
57
62
self.assertEqualDiff(out, '''\
58
63
1 test@us | my helicopter
64
3 user@te | your helicopter
67
def test_no_mail(self):
68
out, err = self.run_bzr_captured(['annotate', 'nomail.txt'])
69
self.assertEquals(err, '')
70
self.assertEqualDiff(out, '''\
74
def test_annotate_cmd_revision(self):
75
out, err = self.run_bzr_captured(['annotate', 'hello.txt', '-r 1'])
76
self.assertEquals(err, '')
77
self.assertEqualDiff(out, '''\
78
1 test@us | my helicopter
81
def test_annotate_cmd_revision3(self):
82
out, err = self.run_bzr_captured(['annotate', 'hello.txt', '-r 3'])
83
self.assertEquals(err, '')
84
self.assertEqualDiff(out, '''\
85
1 test@us | my helicopter
86
3 user@te | your helicopter
89
def test_annotate_cmd_unknown_revision(self):
90
out, err = self.run_bzr_captured(['annotate', 'hello.txt', '-r 10'],
92
self.assertEquals(out, '')
93
self.assertContainsRe(err, 'has no revision 10')
95
def test_annotate_cmd_two_revisions(self):
96
out, err = self.run_bzr_captured(['annotate', 'hello.txt', '-r 1..2'],
98
self.assertEquals(out, '')
99
self.assertEquals(err, 'bzr: ERROR: bzr annotate --revision takes'
100
' exactly 1 argument\n')