~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/selftest/testlog.py

  • Committer: Martin Pool
  • Date: 2005-08-29 13:51:52 UTC
  • Revision ID: mbp@sourcefrog.net-20050829135152-f38a289503a2006a
- lift out import statements within WorkingTree

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, InTempDir
 
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(InTempDir):
 
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
        ass = self.assert_
 
69
        
 
70
        b = Branch('.', init=True)
 
71
 
 
72
        lf = LogCatcher()
 
73
        show_log(b, lf)
 
74
        # no entries yet
 
75
        eq(lf.logs, [])
 
76
 
 
77
 
 
78
        b.commit('empty commit')
 
79
        lf = LogCatcher()
 
80
        show_log(b, lf, verbose=True)
 
81
        eq(len(lf.logs), 1)
 
82
        eq(lf.logs[0].revno, 1)
 
83
        eq(lf.logs[0].rev.message, 'empty commit')
 
84
        d = lf.logs[0].delta
 
85
        self.log('log delta: %r' % d)
 
86
        self.checkDelta(d)
 
87
 
 
88
 
 
89
        self.build_tree(['hello'])
 
90
        b.add('hello')
 
91
        b.commit('add one file')
 
92
 
 
93
        lf = StringIO()
 
94
        # log using regular thing
 
95
        show_log(b, LongLogFormatter(lf))
 
96
        lf.seek(0)
 
97
        for l in lf.readlines():
 
98
            self.log(l)
 
99
 
 
100
        # get log as data structure
 
101
        lf = LogCatcher()
 
102
        show_log(b, lf, verbose=True)
 
103
        eq(len(lf.logs), 2)
 
104
        self.log('log entries:')
 
105
        for logentry in lf.logs:
 
106
            self.log('%4d %s' % (logentry.revno, logentry.rev.message))
 
107
        
 
108
        # first one is most recent
 
109
        logentry = lf.logs[0]
 
110
        eq(logentry.revno, 2)
 
111
        eq(logentry.rev.message, 'add one file')
 
112
        d = logentry.delta
 
113
        self.log('log 2 delta: %r' % d)
 
114
        # self.checkDelta(d, added=['hello'])
 
115