/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_trace.py

  • Committer: Aaron Bentley
  • Date: 2008-02-24 16:42:13 UTC
  • mfrom: (3234 +trunk)
  • mto: This revision was merged to the branch mainline in revision 3235.
  • Revision ID: aaron@aaronbentley.com-20080224164213-eza1lzru5bwuwmmj
Merge with bzr.dev

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_trace.py
1
 
# Copyright (C) 2005, 2006, 2007 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,
30
32
from bzrlib.trace import (
31
33
    mutter, mutter_callsite, report_exception,
32
34
    set_verbosity_level, get_verbosity_level, is_quiet, is_verbose, be_quiet,
 
35
    pop_log_file,
 
36
    push_log_file,
33
37
    _rollover_trace_maybe,
34
38
    )
35
39
 
117
121
        else:
118
122
            self.fail("expected error not raised")
119
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
 
120
129
    def test_mutter_callsite_1(self):
121
130
        """mutter_callsite can capture 1 level of stack frame."""
122
131
        mutter_callsite(1, "foo %s", "a string")
123
132
        log = self._get_log(keep_log_file=True)
124
133
        # begin with the message
125
 
        self.assertStartsWith(log, 'foo a string\nCalled from:\n')
 
134
        self.assertLogStartsWith(log, 'foo a string\nCalled from:\n')
126
135
        # should show two frame: this frame and the one above
127
136
        self.assertContainsRe(log,
128
 
            'test_trace\.py", line \d+, in test_mutter_callsite_1\n')
 
137
            'test_trace\\.py", line \\d+, in test_mutter_callsite_1\n')
129
138
        # this frame should be the final one
130
139
        self.assertEndsWith(log, ' "a string")\n')
131
140
 
134
143
        mutter_callsite(2, "foo %s", "a string")
135
144
        log = self._get_log(keep_log_file=True)
136
145
        # begin with the message
137
 
        self.assertStartsWith(log, 'foo a string\nCalled from:\n')
 
146
        self.assertLogStartsWith(log, 'foo a string\nCalled from:\n')
138
147
        # should show two frame: this frame and the one above
139
148
        self.assertContainsRe(log,
140
149
            'test_trace.py", line \d+, in test_mutter_callsite_2\n')
152
161
        self.assertContainsRe(log, "But fails in an ascii string")
153
162
        self.assertContainsRe(log, u"ascii argument: \xb5")
154
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
 
155
197
 
156
198
class TestVerbosityLevel(TestCase):
157
199