/
test_plugin_registry.py
123 lines (90 loc) · 3.53 KB
/
test_plugin_registry.py
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
from ..plugin_registry import PluginRegistry
from typing import Callable
class TypedCallableRegistry(PluginRegistry[Callable[[int], int]]):
pass
class GeneralCallableRegistry(PluginRegistry):
_global_settings = {"global_setting": None}
@property
def global_setting(self):
return self._global_settings["global_setting"]
@global_setting.setter
def global_setting(self, val):
self._global_settings["global_setting"] = val
def test_plugin_registry():
plugins = TypedCallableRegistry()
assert plugins.names() == []
assert plugins.active == ""
assert plugins.get() is None
assert repr(plugins) == "TypedCallableRegistry(active='', registered=[])"
plugins.register("new_plugin", lambda x: x**2)
assert plugins.names() == ["new_plugin"]
assert plugins.active == ""
assert plugins.get() is None
assert repr(plugins) == (
"TypedCallableRegistry(active='', " "registered=['new_plugin'])"
)
plugins.enable("new_plugin")
assert plugins.names() == ["new_plugin"]
assert plugins.active == "new_plugin"
assert plugins.get()(3) == 9
assert repr(plugins) == (
"TypedCallableRegistry(active='new_plugin', " "registered=['new_plugin'])"
)
def test_plugin_registry_extra_options():
plugins = GeneralCallableRegistry()
plugins.register("metadata_plugin", lambda x, p=2: x**p)
plugins.enable("metadata_plugin")
assert plugins.get()(3) == 9
plugins.enable("metadata_plugin", p=3)
assert plugins.active == "metadata_plugin"
assert plugins.get()(3) == 27
# enabling without changing name
plugins.enable(p=2)
assert plugins.active == "metadata_plugin"
assert plugins.get()(3) == 9
def test_plugin_registry_global_settings():
plugins = GeneralCallableRegistry()
# we need some default plugin, but we won't do anything with it
plugins.register("default", lambda x: x)
plugins.enable("default")
# default value of the global flag
assert plugins.global_setting is None
# enabling changes the global state, not the options
plugins.enable(global_setting=True)
assert plugins.global_setting is True
assert plugins._options == {}
# context manager changes global state temporarily
with plugins.enable(global_setting="temp"):
assert plugins.global_setting == "temp"
assert plugins._options == {}
assert plugins.global_setting is True
assert plugins._options == {}
def test_plugin_registry_context():
plugins = GeneralCallableRegistry()
plugins.register("default", lambda x, p=2: x**p)
# At first there is no plugin enabled
assert plugins.active == ""
assert plugins.options == {}
# Make sure the context is set and reset correctly
with plugins.enable("default", p=6):
assert plugins.active == "default"
assert plugins.options == {"p": 6}
assert plugins.active == ""
assert plugins.options == {}
# Make sure the context is reset even if there is an error
try:
with plugins.enable("default", p=6):
assert plugins.active == "default"
assert plugins.options == {"p": 6}
raise ValueError()
except ValueError:
pass
assert plugins.active == ""
assert plugins.options == {}
# Enabling without specifying name uses current name
plugins.enable("default", p=2)
with plugins.enable(p=6):
assert plugins.active == "default"
assert plugins.options == {"p": 6}
assert plugins.active == "default"
assert plugins.options == {"p": 2}