← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/blackbox/test_exceptions.py

Add bzrlib.pyutils, which has get_named_object, a wrapper around __import__.

This is used to replace various ad hoc implementations of the same logic,
notably the version used in registry's _LazyObjectGetter which had a bug when
getting a module without also getting a member.  And of course, this new
function has unit tests, unlike the replaced code.

This also adds a KnownHooksRegistry subclass to provide a more natural home for
some other logic.

I'm not thrilled about the name of the new module or the new functions, but it's
hard to think of good names for such generic functionality.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/blackbox/test_exceptions.py
1
 
# Copyright (C) 2006 Canonical Ltd
 
1
# Copyright (C) 2006, 2007, 2009, 2010 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
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
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
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
"""Tests for display of exceptions."""
18
18
 
 
19
from cStringIO import StringIO
19
20
import os
20
21
import sys
21
22
 
22
 
from bzrlib import bzrdir, repository
 
23
from bzrlib import (
 
24
    bzrdir,
 
25
    config,
 
26
    errors,
 
27
    osutils,
 
28
    repository,
 
29
    tests,
 
30
    trace,
 
31
    )
23
32
 
24
33
from bzrlib.tests import TestCaseInTempDir, TestCase
25
34
from bzrlib.errors import NotBranchError
26
35
 
 
36
 
27
37
class TestExceptionReporting(TestCase):
28
38
 
29
 
    def test_report_exception(self):
30
 
        """When an error occurs, display bug report details to stderr"""
31
 
        out, err = self.run_bzr("assert-fail", retcode=3)
 
39
    def test_exception_exitcode(self):
 
40
        # we must use a subprocess, because the normal in-memory mechanism
 
41
        # allows errors to propagate up through the test suite
 
42
        out, err = self.run_bzr_subprocess(['assert-fail'],
 
43
            universal_newlines=True,
 
44
            retcode=errors.EXIT_INTERNAL_ERROR)
 
45
        self.assertEqual(4, errors.EXIT_INTERNAL_ERROR)
32
46
        self.assertContainsRe(err,
33
 
                r'bzr: ERROR: exceptions\.AssertionError: always fails\n')
34
 
        self.assertContainsRe(err, r'please send this report to')
35
 
 
36
 
    # TODO: assert-fail doesn't need to always be present; we could just
37
 
    # register (and unregister) it from tests that want to touch it.
38
 
    #
39
 
    # TODO: Some kind of test for the feature of invoking pdb
40
 
    
41
 
 
42
 
class TestDeprecationWarning(TestCaseInTempDir):
 
47
                r'exceptions\.AssertionError: always fails\n')
 
48
        self.assertContainsRe(err, r'Bazaar has encountered an internal error')
 
49
 
 
50
 
 
51
class TestDeprecationWarning(tests.TestCaseWithTransport):
 
52
    """The deprecation warning is controlled via a global variable:
 
53
    repository._deprecation_warning_done. As such, it can be emitted only once
 
54
    during a bzr invocation, no matter how many repositories are involved.
 
55
 
 
56
    It would be better if it was a repo attribute instead but that's far more
 
57
    work than I want to do right now -- vila 20091215.
 
58
    """
 
59
 
 
60
    def setUp(self):
 
61
        super(TestDeprecationWarning, self).setUp()
 
62
        self.disable_deprecation_warning()
 
63
 
 
64
    def enable_deprecation_warning(self, repo=None):
 
65
        """repo is not used yet since _deprecation_warning_done is a global"""
 
66
        repository._deprecation_warning_done = False
 
67
 
 
68
    def disable_deprecation_warning(self, repo=None):
 
69
        """repo is not used yet since _deprecation_warning_done is a global"""
 
70
        repository._deprecation_warning_done = True
 
71
 
 
72
    def make_obsolete_repo(self, path):
 
73
        # We don't want the deprecation raising during the repo creation
 
74
        tree = self.make_branch_and_tree(path, format=bzrdir.BzrDirFormat5())
 
75
        return tree
 
76
 
 
77
    def check_warning(self, present):
 
78
        if present:
 
79
            check = self.assertContainsRe
 
80
        else:
 
81
            check = self.assertNotContainsRe
 
82
        check(self._get_log(keep_log_file=True), 'WARNING.*bzr upgrade')
43
83
 
44
84
    def test_repository_deprecation_warning(self):
45
85
        """Old formats give a warning"""
46
 
        # the warning's normally off for testing but we reenable it
47
 
        repository._deprecation_warning_done = False
48
 
        try:
49
 
            os.mkdir('foo')
50
 
            bzrdir.BzrDirFormat5().initialize('foo')
51
 
            out, err = self.run_bzr("status", "foo")
52
 
            self.assertContainsRe(self._get_log(keep_log_file=True),
53
 
                                  "bzr upgrade")
54
 
        finally:
55
 
            repository._deprecation_warning_done = True
56
 
 
 
86
        self.make_obsolete_repo('foo')
 
87
        self.enable_deprecation_warning()
 
88
        out, err = self.run_bzr('status', working_dir='foo')
 
89
        self.check_warning(True)
 
90
 
 
91
    def test_repository_deprecation_warning_suppressed_global(self):
 
92
        """Old formats give a warning"""
 
93
        conf = config.GlobalConfig()
 
94
        conf.set_user_option('suppress_warnings', 'format_deprecation')
 
95
        self.make_obsolete_repo('foo')
 
96
        self.enable_deprecation_warning()
 
97
        out, err = self.run_bzr('status', working_dir='foo')
 
98
        self.check_warning(False)
 
99
 
 
100
    def test_repository_deprecation_warning_suppressed_locations(self):
 
101
        """Old formats give a warning"""
 
102
        self.make_obsolete_repo('foo')
 
103
        conf = config.LocationConfig(osutils.pathjoin(self.test_dir, 'foo'))
 
104
        conf.set_user_option('suppress_warnings', 'format_deprecation')
 
105
        self.enable_deprecation_warning()
 
106
        out, err = self.run_bzr('status', working_dir='foo')
 
107
        self.check_warning(False)
 
108
 
 
109
    def test_repository_deprecation_warning_suppressed_branch(self):
 
110
        """Old formats give a warning"""
 
111
        tree = self.make_obsolete_repo('foo')
 
112
        conf = tree.branch.get_config()
 
113
        conf.set_user_option('suppress_warnings', 'format_deprecation')
 
114
        self.enable_deprecation_warning()
 
115
        out, err = self.run_bzr('status', working_dir='foo')
 
116
        self.check_warning(False)