~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/selftest/log.py

[merge] Added the uncommit plugin

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2005 by Canonical Ltd
 
2
 
 
3
# This program is free software; you can redistribute it and/or modify
 
4
# it under the terms of the GNU General Public License as published by
 
5
# the Free Software Foundation; either version 2 of the License, or
 
6
# (at your option) any later version.
 
7
 
 
8
# This program is distributed in the hope that it will be useful,
 
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
# GNU General Public License for more details.
 
12
 
 
13
# You should have received a copy of the GNU General Public License
 
14
# along with this program; if not, write to the Free Software
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
 
 
17
import os
 
18
from cStringIO import StringIO
 
19
 
 
20
from bzrlib.selftest import BzrTestBase, TestCaseInTempDir
 
21
from bzrlib.log import LogFormatter, show_log, LongLogFormatter, ShortLogFormatter
 
22
from bzrlib.branch import Branch
 
23
from bzrlib.errors import InvalidRevisionNumber
 
24
 
 
25
class _LogEntry(object):
 
26
    # should probably move into bzrlib.log?
 
27
    pass
 
28
 
 
29
 
 
30
class LogCatcher(LogFormatter):
 
31
    """Pull log messages into list rather than displaying them.
 
32
 
 
33
    For ease of testing we save log messages here rather than actually
 
34
    formatting them, so that we can precisely check the result without
 
35
    being too dependent on the exact formatting.
 
36
 
 
37
    We should also test the LogFormatter.
 
38
    """
 
39
    def __init__(self):
 
40
        super(LogCatcher, self).__init__(to_file=None)
 
41
        self.logs = []
 
42
        
 
43
        
 
44
    def show(self, revno, rev, delta):
 
45
        le = _LogEntry()
 
46
        le.revno = revno
 
47
        le.rev = rev
 
48
        le.delta = delta
 
49
        self.logs.append(le)
 
50
 
 
51
 
 
52
class SimpleLogTest(TestCaseInTempDir):
 
53
 
 
54
    def checkDelta(self, delta, **kw):
 
55
        """Check the filenames touched by a delta are as expected."""
 
56
        for n in 'added', 'removed', 'renamed', 'modified', 'unchanged':
 
57
            expected = kw.get(n, [])
 
58
 
 
59
            # tests are written with unix paths; fix them up for windows
 
60
            if os.sep != '/':
 
61
                expected = [x.replace('/', os.sep) for x in expected]
 
62
 
 
63
            # strip out only the path components
 
64
            got = [x[0] for x in getattr(delta, n)]
 
65
            self.assertEquals(expected, got)
 
66
 
 
67
    def test_cur_revno(self):
 
68
        b = Branch('.', init=True)
 
69
 
 
70
        lf = LogCatcher()
 
71
        b.working_tree().commit('empty commit')
 
72
        show_log(b, lf, verbose=True, start_revision=1, end_revision=1)
 
73
        self.assertRaises(InvalidRevisionNumber, show_log, b, lf,
 
74
                          start_revision=2, end_revision=1) 
 
75
        self.assertRaises(InvalidRevisionNumber, show_log, b, lf,
 
76
                          start_revision=1, end_revision=2) 
 
77
        self.assertRaises(InvalidRevisionNumber, show_log, b, lf,
 
78
                          start_revision=0, end_revision=2) 
 
79
        self.assertRaises(InvalidRevisionNumber, show_log, b, lf,
 
80
                          start_revision=1, end_revision=0) 
 
81
        self.assertRaises(InvalidRevisionNumber, show_log, b, lf,
 
82
                          start_revision=-1, end_revision=1) 
 
83
        self.assertRaises(InvalidRevisionNumber, show_log, b, lf,
 
84
                          start_revision=1, end_revision=-1) 
 
85
 
 
86
    def test_cur_revno(self):
 
87
        b = Branch.initialize('.')
 
88
 
 
89
        lf = LogCatcher()
 
90
        b.working_tree().commit('empty commit')
 
91
        show_log(b, lf, verbose=True, start_revision=1, end_revision=1)
 
92
        self.assertRaises(InvalidRevisionNumber, show_log, b, lf,
 
93
                          start_revision=2, end_revision=1) 
 
94
        self.assertRaises(InvalidRevisionNumber, show_log, b, lf,
 
95
                          start_revision=1, end_revision=2) 
 
96
        self.assertRaises(InvalidRevisionNumber, show_log, b, lf,
 
97
                          start_revision=0, end_revision=2) 
 
98
        self.assertRaises(InvalidRevisionNumber, show_log, b, lf,
 
99
                          start_revision=1, end_revision=0) 
 
100
        self.assertRaises(InvalidRevisionNumber, show_log, b, lf,
 
101
                          start_revision=-1, end_revision=1) 
 
102
        self.assertRaises(InvalidRevisionNumber, show_log, b, lf,
 
103
                          start_revision=1, end_revision=-1) 
 
104
 
 
105
    def test_simple_log(self):
 
106
        eq = self.assertEquals
 
107
        
 
108
        b = Branch.initialize('.')
 
109
 
 
110
        lf = LogCatcher()
 
111
        show_log(b, lf)
 
112
        # no entries yet
 
113
        eq(lf.logs, [])
 
114
 
 
115
        b.working_tree().commit('empty commit')
 
116
        lf = LogCatcher()
 
117
        show_log(b, lf, verbose=True)
 
118
        eq(len(lf.logs), 1)
 
119
        eq(lf.logs[0].revno, 1)
 
120
        eq(lf.logs[0].rev.message, 'empty commit')
 
121
        d = lf.logs[0].delta
 
122
        self.log('log delta: %r' % d)
 
123
        self.checkDelta(d)
 
124
 
 
125
        self.build_tree(['hello'])
 
126
        b.add('hello')
 
127
        b.working_tree().commit('add one file')
 
128
 
 
129
        lf = StringIO()
 
130
        # log using regular thing
 
131
        show_log(b, LongLogFormatter(lf))
 
132
        lf.seek(0)
 
133
        for l in lf.readlines():
 
134
            self.log(l)
 
135
 
 
136
        # get log as data structure
 
137
        lf = LogCatcher()
 
138
        show_log(b, lf, verbose=True)
 
139
        eq(len(lf.logs), 2)
 
140
        self.log('log entries:')
 
141
        for logentry in lf.logs:
 
142
            self.log('%4d %s' % (logentry.revno, logentry.rev.message))
 
143
        
 
144
        # first one is most recent
 
145
        logentry = lf.logs[0]
 
146
        eq(logentry.revno, 2)
 
147
        eq(logentry.rev.message, 'add one file')
 
148
        d = logentry.delta
 
149
        self.log('log 2 delta: %r' % d)
 
150
        # self.checkDelta(d, added=['hello'])
 
151
        
 
152
        # commit a log message with control characters
 
153
        msg = "All 8-bit chars: " +  ''.join([unichr(x) for x in range(256)])
 
154
        self.log("original commit message: %r", msg)
 
155
        b.working_tree().commit(msg)
 
156
        lf = LogCatcher()
 
157
        show_log(b, lf, verbose=True)
 
158
        committed_msg = lf.logs[0].rev.message
 
159
        self.log("escaped commit message: %r", committed_msg)
 
160
        self.assert_(msg != committed_msg)
 
161
        self.assert_(len(committed_msg) > len(msg))
 
162
 
 
163
        # Check that log message with only XML-valid characters isn't
 
164
        # escaped.  As ElementTree apparently does some kind of
 
165
        # newline conversion, neither LF (\x0A) nor CR (\x0D) are
 
166
        # included in the test commit message, even though they are
 
167
        # valid XML 1.0 characters.
 
168
        msg = "\x09" + ''.join([unichr(x) for x in range(0x20, 256)])
 
169
        self.log("original commit message: %r", msg)
 
170
        b.working_tree().commit(msg)
 
171
        lf = LogCatcher()
 
172
        show_log(b, lf, verbose=True)
 
173
        committed_msg = lf.logs[0].rev.message
 
174
        self.log("escaped commit message: %r", committed_msg)
 
175
        self.assert_(msg == committed_msg)
 
176
 
 
177
    def test_trailing_newlines(self):
 
178
        b = Branch.initialize('.')
 
179
        b.nick='test'
 
180
        wt = b.working_tree()
 
181
        open('a', 'wb').write('hello moto\n')
 
182
        b.add('a')
 
183
        wt.commit('simple log message', rev_id='a1'
 
184
                , timestamp=1132586655.459960938, timezone=-6*3600
 
185
                , committer='Joe Foo <joe@foo.com>')
 
186
        open('b', 'wb').write('goodbye\n')
 
187
        b.add('b')
 
188
        wt.commit('multiline\nlog\nmessage\n', rev_id='a2'
 
189
                , timestamp=1132586842.411175966, timezone=-6*3600
 
190
                , committer='Joe Foo <joe@foo.com>')
 
191
 
 
192
        open('c', 'wb').write('just another manic monday\n')
 
193
        b.add('c')
 
194
        wt.commit('single line with trailing newline\n', rev_id='a3'
 
195
                , timestamp=1132587176.835228920, timezone=-6*3600
 
196
                , committer = 'Joe Foo <joe@foo.com>')
 
197
 
 
198
        sio = StringIO()
 
199
        lf = ShortLogFormatter(to_file=sio)
 
200
        show_log(b, lf)
 
201
        self.assertEquals(sio.getvalue(), """\
 
202
    3 Joe Foo\t2005-11-21
 
203
      single line with trailing newline
 
204
 
 
205
    2 Joe Foo\t2005-11-21
 
206
      multiline
 
207
      log
 
208
      message
 
209
 
 
210
    1 Joe Foo\t2005-11-21
 
211
      simple log message
 
212
 
 
213
""")
 
214
 
 
215
        sio = StringIO()
 
216
        lf = LongLogFormatter(to_file=sio)
 
217
        show_log(b, lf)
 
218
        self.assertEquals(sio.getvalue(), """\
 
219
------------------------------------------------------------
 
220
revno: 3
 
221
committer: Joe Foo <joe@foo.com>
 
222
branch nick: test
 
223
timestamp: Mon 2005-11-21 09:32:56 -0600
 
224
message:
 
225
  single line with trailing newline
 
226
------------------------------------------------------------
 
227
revno: 2
 
228
committer: Joe Foo <joe@foo.com>
 
229
branch nick: test
 
230
timestamp: Mon 2005-11-21 09:27:22 -0600
 
231
message:
 
232
  multiline
 
233
  log
 
234
  message
 
235
------------------------------------------------------------
 
236
revno: 1
 
237
committer: Joe Foo <joe@foo.com>
 
238
branch nick: test
 
239
timestamp: Mon 2005-11-21 09:24:15 -0600
 
240
message:
 
241
  simple log message
 
242
""")
 
243