← 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: Aaron Bentley
  • Date: 2008-03-03 16:52:41 UTC
  • mfrom: (3144.3.11 fix-conflict-handling)
  • mto: This revision was merged to the branch mainline in revision 3250.
  • Revision ID: aaron@aaronbentley.com-20080303165241-0k2c7ggs6kr9q6hf
Merge with fix-conflict-handling

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
23
23
import os
24
24
import re
25
25
import sys
 
26
import tempfile
26
27
 
27
28
from bzrlib import (
28
29
    errors,
31
32
from bzrlib.trace import (
32
33
    mutter, mutter_callsite, report_exception,
33
34
    set_verbosity_level, get_verbosity_level, is_quiet, is_verbose, be_quiet,
 
35
    pop_log_file,
 
36
    push_log_file,
34
37
    _rollover_trace_maybe,
35
38
    )
36
39
 
158
161
        self.assertContainsRe(log, "But fails in an ascii string")
159
162
        self.assertContainsRe(log, u"ascii argument: \xb5")
160
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
 
161
197
 
162
198
class TestVerbosityLevel(TestCase):
163
199