1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
# Copyright (C) 2012- Takafumi Arakaki
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
from ..core import EPCDispatcher
from .utils import BaseTestCase
class Dummy(object):
pass
class TestEPCDispatcher(BaseTestCase):
def setUp(self):
self.dispatcher = EPCDispatcher()
def test_register_module(self):
import os
self.dispatcher.register_instance(os)
self.assertIs(self.dispatcher.get_method('chmod'), os.chmod)
def test_register_module_with_dotted_names(self):
import os
self.dispatcher.register_instance(os, allow_dotted_names=True)
self.assertIs(self.dispatcher.get_method('path.join'), os.path.join)
def test_error_on_private_method_access(self):
obj = Dummy()
obj._private_method = lambda: None
obj.sub = Dummy()
obj.sub._private_attribute = Dummy()
obj.sub._private_attribute.some_method = lambda: None
self.dispatcher.register_instance(obj, allow_dotted_names=True)
self.assertRaises(AttributeError, self.dispatcher.get_method,
'_private_method')
self.assertRaises(AttributeError, self.dispatcher.get_method,
'obj.sub._private_attribute.some_method')
def test_instance_get_method(self):
always_me = lambda: None
obj = Dummy()
obj._get_method = lambda _: always_me
self.dispatcher.register_instance(obj)
self.assertIs(self.dispatcher.get_method('x'), always_me)
self.assertIs(self.dispatcher.get_method('y'), always_me)
|
