/b-gtk/fix-viz

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/b-gtk/fix-viz

« back to all changes in this revision

Viewing changes to tests/__init__.py

  • Committer: Curtis Hovey
  • Date: 2012-03-04 00:09:11 UTC
  • mto: This revision was merged to the branch mainline in revision 782.
  • Revision ID: sinzui.is@verizon.net-20120304000911-iaw196xcw4zfta9m
Added test to verify that DivergedBranches is handled.

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/__init__.py
46
46
class MockMethod(object):
47
47
 
48
48
    @classmethod
49
 
    def bind(klass, test_instance, obj, method_name, return_value=None):
 
49
    def bind(klass, test_instance, obj, method_name,
 
50
             return_value=None, raise_error=None, raise_on=1):
50
51
        original_method = getattr(obj, method_name)
51
52
        test_instance.addCleanup(setattr, obj, method_name, original_method)
52
 
        setattr(obj, method_name, klass(return_value))
 
53
        setattr(obj, method_name, klass(return_value, raise_error, raise_on))
53
54
 
54
 
    def __init__(self, return_value=None):
 
55
    def __init__(self, return_value=None, raise_error=None, raise_on=1):
55
56
        self.called = False
56
57
        self.call_count = 0
57
58
        self.args = None
58
59
        self.kwargs = None
59
60
        self.return_value = return_value
 
61
        self.raise_error = raise_error
 
62
        self.raise_on = raise_on
60
63
 
61
64
    def __call__(self, *args, **kwargs):
62
65
        self.called = True
63
66
        self.call_count += 1
64
67
        self.args = args
65
68
        self.kwargs = kwargs
 
69
        if self.raise_error is not None and self.call_count == self.raise_on:
 
70
            raise self.raise_error
66
71
        return self.return_value
67
72
 
68
73