diff options
Diffstat (limited to 'mitmproxy/test/taddons.py')
| -rw-r--r-- | mitmproxy/test/taddons.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/mitmproxy/test/taddons.py b/mitmproxy/test/taddons.py index d966f1d5..12fc0986 100644 --- a/mitmproxy/test/taddons.py +++ b/mitmproxy/test/taddons.py @@ -59,7 +59,7 @@ class context: provides a number of helper methods for common testing scenarios. """ - def __init__(self, master=None, options=None): + def __init__(self, *addons, master=None, options=None): options = options or mitmproxy.options.Options() self.master = master or RecordingMaster( options @@ -67,6 +67,10 @@ class context: self.options = self.master.options self.wrapped = None + loader = addonmanager.Loader(self.master) + for a in addons: + self.master.addons.invoke_addon(a, "load", loader) + def ctx(self): """ Returns a new handler context. |