/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-02-28 13:52:00 UTC
  • mfrom: (776.2.12 ui-factory)
  • Revision ID: sinzui.is@verizon.net-20120228135200-xzqb6qhtzck01doo
Merged GtkUIFactory additions and tweaks.

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/__init__.py
23
23
import os
24
24
 
25
25
 
26
 
def discover_test_names(module_or_name):
27
 
    if isinstance(module_or_name, basestring):
28
 
        match = module_or_name
 
26
def load_tests(basic_tests, module, loader):
 
27
    if module == 'discover':
 
28
        here = os.path.abspath(os.path.dirname(__file__))
 
29
        basic_tests.addTest(loader.discover(here))
29
30
    else:
30
 
        match = ''
31
 
    file_names = os.listdir(os.path.dirname(__file__))
32
 
    test_names = set()
33
 
    for file_name in file_names:
34
 
        name, ext = os.path.splitext(file_name)
35
 
        if name.startswith('test_') and ext == '.py' and match in name:
36
 
            test_names.add("%s.%s" % (__name__, name))
37
 
    return test_names
38
 
 
39
 
 
40
 
def load_tests(basic_tests, module, loader):
41
 
    test_names = discover_test_names(module)
42
 
    basic_tests.addTest(loader.loadTestsFromModuleNames(test_names))
 
31
        full_name = "%s.%s" % (__name__, module)
 
32
        basic_tests.addTest(loader.loadTestsFromModuleNames([full_name]))
43
33
    return basic_tests
44
34
 
45
35
 
46
36
class MockMethod(object):
47
37
 
48
38
    @classmethod
49
 
    def bind(klass, test_instance, obj, method_name,
50
 
             return_value=None, raise_error=None, raise_on=1):
 
39
    def bind(klass, test_instance, obj, method_name, return_value=None):
51
40
        original_method = getattr(obj, method_name)
52
41
        test_instance.addCleanup(setattr, obj, method_name, original_method)
53
 
        setattr(obj, method_name, klass(return_value, raise_error, raise_on))
 
42
        setattr(obj, method_name, klass(return_value))
54
43
 
55
 
    def __init__(self, return_value=None, raise_error=None, raise_on=1):
 
44
    def __init__(self, return_value=None):
56
45
        self.called = False
57
46
        self.call_count = 0
58
47
        self.args = None
59
48
        self.kwargs = None
60
49
        self.return_value = return_value
61
 
        self.raise_error = raise_error
62
 
        self.raise_on = raise_on
63
50
 
64
51
    def __call__(self, *args, **kwargs):
65
52
        self.called = True
66
53
        self.call_count += 1
67
54
        self.args = args
68
55
        self.kwargs = kwargs
69
 
        if self.raise_error is not None and self.call_count == self.raise_on:
70
 
            raise self.raise_error
71
56
        return self.return_value
72
57
 
73
58