~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/selftest/testlog.py

  • Committer: Martin Pool
  • Date: 2005-06-22 09:08:43 UTC
  • Revision ID: mbp@sourcefrog.net-20050622090843-78fe9c62da9ed167
- add john's changeset 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
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)