~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/selftest/testlog.py

  • Committer: mbp at sourcefrog
  • Date: 2005-03-09 06:49:00 UTC
  • Revision ID: mbp@sourcefrog.net-20050309064900-74935ffb7350b24b
import more files from baz

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
 
 
24
 
class _LogEntry(object):
25
 
    # should probably move into bzrlib.log?
26
 
    pass
27
 
 
28
 
 
29
 
class LogCatcher(LogFormatter):
30
 
    """Pull log messages into list rather than displaying them.
31
 
 
32
 
    For ease of testing we save log messages here rather than actually
33
 
    formatting them, so that we can precisely check the result without
34
 
    being too dependent on the exact formatting.
35
 
 
36
 
    We should also test the LogFormatter.
37
 
    """
38
 
    def __init__(self):
39
 
        super(LogCatcher, self).__init__(to_file=None)
40
 
        self.logs = []
41
 
        
42
 
        
43
 
    def show(self, revno, rev, delta):
44
 
        le = _LogEntry()
45
 
        le.revno = revno
46
 
        le.rev = rev
47
 
        le.delta = delta
48
 
        self.logs.append(le)
49
 
 
50
 
 
51
 
class SimpleLogTest(TestCaseInTempDir):
52
 
 
53
 
    def checkDelta(self, delta, **kw):
54
 
        """Check the filenames touched by a delta are as expected."""
55
 
        for n in 'added', 'removed', 'renamed', 'modified', 'unchanged':
56
 
            expected = kw.get(n, [])
57
 
 
58
 
            # tests are written with unix paths; fix them up for windows
59
 
            if os.sep != '/':
60
 
                expected = [x.replace('/', os.sep) for x in expected]
61
 
 
62
 
            # strip out only the path components
63
 
            got = [x[0] for x in getattr(delta, n)]
64
 
            self.assertEquals(expected, got)
65
 
 
66
 
    def test_simple_log(self):
67
 
        eq = self.assertEquals
68
 
        
69
 
        b = Branch('.', init=True)
70
 
 
71
 
        lf = LogCatcher()
72
 
        show_log(b, lf)
73
 
        # no entries yet
74
 
        eq(lf.logs, [])
75
 
 
76
 
 
77
 
        b.commit('empty commit')
78
 
        lf = LogCatcher()
79
 
        show_log(b, lf, verbose=True)
80
 
        eq(len(lf.logs), 1)
81
 
        eq(lf.logs[0].revno, 1)
82
 
        eq(lf.logs[0].rev.message, 'empty commit')
83
 
        d = lf.logs[0].delta
84
 
        self.log('log delta: %r' % d)
85
 
        self.checkDelta(d)
86
 
 
87
 
 
88
 
        self.build_tree(['hello'])
89
 
        b.add('hello')
90
 
        b.commit('add one file')
91
 
 
92
 
        lf = StringIO()
93
 
        # log using regular thing
94
 
        show_log(b, LongLogFormatter(lf))
95
 
        lf.seek(0)
96
 
        for l in lf.readlines():
97
 
            self.log(l)
98
 
 
99
 
        # get log as data structure
100
 
        lf = LogCatcher()
101
 
        show_log(b, lf, verbose=True)
102
 
        eq(len(lf.logs), 2)
103
 
        self.log('log entries:')
104
 
        for logentry in lf.logs:
105
 
            self.log('%4d %s' % (logentry.revno, logentry.rev.message))
106
 
        
107
 
        # first one is most recent
108
 
        logentry = lf.logs[0]
109
 
        eq(logentry.revno, 2)
110
 
        eq(logentry.rev.message, 'add one file')
111
 
        d = logentry.delta
112
 
        self.log('log 2 delta: %r' % d)
113
 
        # self.checkDelta(d, added=['hello'])
114