← 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: Martin Pool
  • Date: 2006-06-20 05:32:16 UTC
  • mfrom: (1797 +trunk)
  • mto: This revision was merged to the branch mainline in revision 1798.
  • Revision ID: mbp@sourcefrog.net-20060620053216-817857d7ca3e9d1f
[merge] bzr.dev

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_trace.py
1
 
# Copyright (C) 2005 by Canonical Ltd
2
 
#   Authors: Robert Collins <robert.collins@canonical.com>
 
1
# Copyright (C) 2005, 2006 by Canonical Ltd
3
2
#
4
3
# This program is free software; you can redistribute it and/or modify
5
4
# it under the terms of the GNU General Public License as published by
19
18
 
20
19
"""Tests for trace library"""
21
20
 
 
21
import errno
22
22
import os
23
23
import sys
 
24
from StringIO import StringIO
24
25
 
25
26
from bzrlib.tests import TestCaseInTempDir, TestCase
26
 
from bzrlib.trace import format_exception_short, mutter
27
 
from bzrlib.errors import NotBranchError, BzrError, BzrNewError
 
27
from bzrlib.trace import mutter, report_exception
 
28
from bzrlib.errors import NotBranchError
 
29
 
 
30
 
 
31
def _format_exception():
 
32
    """Format an exception as it would normally be displayed to the user"""
 
33
    buf = StringIO()
 
34
    report_exception(sys.exc_info(), buf)
 
35
    return buf.getvalue()
 
36
 
28
37
 
29
38
class TestTrace(TestCase):
 
39
 
30
40
    def test_format_sys_exception(self):
31
 
        """Short formatting of exceptions"""
32
41
        try:
33
42
            raise NotImplementedError, "time travel"
34
43
        except NotImplementedError:
35
44
            pass
36
 
        error_lines = format_exception_short(sys.exc_info()).splitlines()
37
 
        self.assertEqualDiff(error_lines[0], 
38
 
                'exceptions.NotImplementedError: time travel')
39
 
        self.assertContainsRe(error_lines[1], 
40
 
                r'^  at .*trace\.py line \d+$')  
41
 
        self.assertContainsRe(error_lines[2], 
42
 
                r'^  in test_format_sys_exception$')
 
45
        err = _format_exception()
 
46
        self.assertEqualDiff(err.splitlines()[0],
 
47
                'bzr: ERROR: exceptions.NotImplementedError: time travel')
 
48
        self.assertContainsRe(err,
 
49
                r'File.*test_trace.py')
 
50
 
 
51
    def test_format_interrupt_exception(self):
 
52
        try:
 
53
            raise KeyboardInterrupt()
 
54
        except KeyboardInterrupt:
 
55
            # XXX: Some risk that a *real* keyboard interrupt won't be seen
 
56
            pass
 
57
        msg = _format_exception()
 
58
        self.assertTrue(len(msg) > 0)
 
59
        self.assertEqualDiff(msg, 'bzr: interrupted\n')
 
60
 
 
61
    def test_format_os_error(self):
 
62
        try:
 
63
            file('nosuchfile22222')
 
64
        except (OSError, IOError):
 
65
            pass
 
66
        msg = _format_exception()
 
67
        self.assertContainsRe(msg, r'^bzr: ERROR: \[Errno .*\] No such file.*nosuchfile')
 
68
 
43
69
 
44
70
    def test_format_exception(self):
45
 
        """Short formatting of exceptions"""
 
71
        """Short formatting of bzr exceptions"""
46
72
        try:
47
73
            raise NotBranchError, 'wibble'
48
74
        except NotBranchError:
49
75
            pass
50
 
        msg = format_exception_short(sys.exc_info())
51
 
        self.assertEqualDiff(msg, 'Not a branch: wibble')
52
 
 
53
 
    def test_format_old_exception(self):
54
 
        # format a class that doesn't descend from BzrNewError; 
55
 
        # remove this test when everything is unified there
56
 
        self.assertFalse(issubclass(BzrError, BzrNewError))
57
 
        try:
58
 
            raise BzrError('some old error')
59
 
        except BzrError:
60
 
            pass
61
 
        msg = format_exception_short(sys.exc_info())
62
 
        self.assertEqualDiff(msg, 'some old error')
 
76
        msg = _format_exception()
 
77
        self.assertTrue(len(msg) > 0)
 
78
        self.assertEqualDiff(msg, 'bzr: ERROR: Not a branch: wibble\n')
63
79
 
64
80
    def test_trace_unicode(self):
65
81
        """Write Unicode to trace log"""
67
83
        self.assertContainsRe('the unicode character',
68
84
                self._get_log())
69
85
 
 
86
    def test_report_broken_pipe(self):
 
87
        try:
 
88
            raise IOError(errno.EPIPE, 'broken pipe foofofo')
 
89
        except IOError, e:
 
90
            msg = _format_exception()
 
91
            self.assertEquals(msg, "bzr: broken pipe\n")
 
92
        else:
 
93
            self.fail("expected error not raised")
 
94
 
70
95
    def test_mutter_never_fails(self):
71
96
        # Even if the decode/encode stage fails, mutter should not
72
97
        # raise an exception
81
106
        self.assertContainsRe(log, 'Writing a greek mu')
82
107
        self.assertContainsRe(log, 'UnicodeError')
83
108
        self.assertContainsRe(log, "'But fails in an ascii string")
84