← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_trace.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2008-04-07 07:52:50 UTC
  • mfrom: (3340.1.1 208418-1.4)
  • Revision ID: pqm@pqm.ubuntu.com-20080407075250-phs53xnslo8boaeo
Return the correct knit serialisation method in _StreamAccess.
        (Andrew Bennetts, Martin Pool, Robert Collins)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_trace.py
1
 
# Copyright (C) 2005, 2006 Canonical Ltd
 
1
# Copyright (C) 2005, 2006, 2007, 2008 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
21
21
from cStringIO import StringIO
22
22
import errno
23
23
import os
 
24
import re
24
25
import sys
 
26
import tempfile
25
27
 
26
28
from bzrlib import (
27
29
    errors,
28
30
    )
29
31
from bzrlib.tests import TestCaseInTempDir, TestCase
30
 
from bzrlib.trace import mutter, report_exception
 
32
from bzrlib.trace import (
 
33
    mutter, mutter_callsite, report_exception,
 
34
    set_verbosity_level, get_verbosity_level, is_quiet, is_verbose, be_quiet,
 
35
    pop_log_file,
 
36
    push_log_file,
 
37
    _rollover_trace_maybe,
 
38
    )
31
39
 
32
40
 
33
41
def _format_exception():
83
91
            pass
84
92
        msg = _format_exception()
85
93
        self.assertTrue(len(msg) > 0)
86
 
        self.assertEqualDiff(msg, 'bzr: ERROR: Not a branch: wibble\n')
 
94
        self.assertEqualDiff(msg, 'bzr: ERROR: Not a branch: \"wibble\".\n')
87
95
 
88
96
    def test_trace_unicode(self):
89
97
        """Write Unicode to trace log"""
113
121
        else:
114
122
            self.fail("expected error not raised")
115
123
 
 
124
    def assertLogStartsWith(self, log, string):
 
125
        """Like assertStartsWith, but skips the log timestamp."""
 
126
        self.assertContainsRe(log,
 
127
            '^\\d+\\.\\d+  ' + re.escape(string))
 
128
 
 
129
    def test_mutter_callsite_1(self):
 
130
        """mutter_callsite can capture 1 level of stack frame."""
 
131
        mutter_callsite(1, "foo %s", "a string")
 
132
        log = self._get_log(keep_log_file=True)
 
133
        # begin with the message
 
134
        self.assertLogStartsWith(log, 'foo a string\nCalled from:\n')
 
135
        # should show two frame: this frame and the one above
 
136
        self.assertContainsRe(log,
 
137
            'test_trace\\.py", line \\d+, in test_mutter_callsite_1\n')
 
138
        # this frame should be the final one
 
139
        self.assertEndsWith(log, ' "a string")\n')
 
140
 
 
141
    def test_mutter_callsite_2(self):
 
142
        """mutter_callsite can capture 2 levels of stack frame."""
 
143
        mutter_callsite(2, "foo %s", "a string")
 
144
        log = self._get_log(keep_log_file=True)
 
145
        # begin with the message
 
146
        self.assertLogStartsWith(log, 'foo a string\nCalled from:\n')
 
147
        # should show two frame: this frame and the one above
 
148
        self.assertContainsRe(log,
 
149
            'test_trace.py", line \d+, in test_mutter_callsite_2\n')
 
150
        # this frame should be the final one
 
151
        self.assertEndsWith(log, ' "a string")\n')
 
152
 
116
153
    def test_mutter_never_fails(self):
117
154
        # Even if the decode/encode stage fails, mutter should not
118
155
        # raise an exception
123
160
        self.assertContainsRe(log, 'Writing a greek mu')
124
161
        self.assertContainsRe(log, "But fails in an ascii string")
125
162
        self.assertContainsRe(log, u"ascii argument: \xb5")
 
163
 
 
164
    def test_push_log_file(self):
 
165
        """Can push and pop log file, and this catches mutter messages.
 
166
 
 
167
        This is primarily for use in the test framework. 
 
168
        """
 
169
        tmp1 = tempfile.NamedTemporaryFile()
 
170
        tmp2 = tempfile.NamedTemporaryFile()
 
171
        try:
 
172
            memento1 = push_log_file(tmp1)
 
173
            mutter("comment to file1")
 
174
            try:
 
175
                memento2 = push_log_file(tmp2)
 
176
                try:
 
177
                    mutter("comment to file2")
 
178
                finally:
 
179
                    pop_log_file(memento2)
 
180
                mutter("again to file1")
 
181
            finally:
 
182
                pop_log_file(memento1)
 
183
            # the files were opened in binary mode, so should have exactly
 
184
            # these bytes.  and removing the file as the log target should
 
185
            # have caused them to be flushed out.  need to match using regexps
 
186
            # as there's a timestamp at the front.
 
187
            tmp1.seek(0)
 
188
            self.assertContainsRe(tmp1.read(),
 
189
                r"\d+\.\d+  comment to file1\n\d+\.\d+  again to file1\n")
 
190
            tmp2.seek(0)
 
191
            self.assertContainsRe(tmp2.read(),
 
192
                r"\d+\.\d+  comment to file2\n")
 
193
        finally:
 
194
            tmp1.close()
 
195
            tmp2.close()
 
196
 
 
197
 
 
198
class TestVerbosityLevel(TestCase):
 
199
 
 
200
    def test_verbosity_level(self):
 
201
        set_verbosity_level(1)
 
202
        self.assertEqual(1, get_verbosity_level())
 
203
        self.assertTrue(is_verbose())
 
204
        self.assertFalse(is_quiet())
 
205
        set_verbosity_level(-1)
 
206
        self.assertEqual(-1, get_verbosity_level())
 
207
        self.assertFalse(is_verbose())
 
208
        self.assertTrue(is_quiet())
 
209
        set_verbosity_level(0)
 
210
        self.assertEqual(0, get_verbosity_level())
 
211
        self.assertFalse(is_verbose())
 
212
        self.assertFalse(is_quiet())
 
213
 
 
214
    def test_be_quiet(self):
 
215
        # Confirm the old API still works
 
216
        be_quiet(True)
 
217
        self.assertEqual(-1, get_verbosity_level())
 
218
        be_quiet(False)
 
219
        self.assertEqual(0, get_verbosity_level())
 
220
 
 
221
 
 
222
class TestBzrLog(TestCaseInTempDir):
 
223
 
 
224
    def test_log_rollover(self):
 
225
        temp_log_name = 'test-log'
 
226
        trace_file = open(temp_log_name, 'at')
 
227
        trace_file.write('test_log_rollover padding\n' * 1000000)
 
228
        trace_file.close()
 
229
        _rollover_trace_maybe(temp_log_name)
 
230
        # should have been rolled over
 
231
        self.assertFalse(os.access(temp_log_name, os.R_OK))