diff options
Diffstat (limited to 'mitmproxy/test/taddons.py')
| -rw-r--r-- | mitmproxy/test/taddons.py | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/mitmproxy/test/taddons.py b/mitmproxy/test/taddons.py index 3dbccba2..39ebb2e6 100644 --- a/mitmproxy/test/taddons.py +++ b/mitmproxy/test/taddons.py @@ -107,14 +107,16 @@ class context: self.master.addons.invoke_addon( addon, "configure", - self.options, kwargs.keys() ) def script(self, path): + """ + Loads a script from path, and returns the enclosed addon. + """ sc = script.Script(path) loader = addonmanager.Loader(self.master) - sc.load(loader) - for a in addonmanager.traverse(sc.addons): - getattr(a, "load", lambda x: None)(loader) - return sc + self.master.addons.invoke_addon(sc, "load", loader) + self.configure(sc) + self.master.addons.invoke_addon(sc, "tick") + return sc.addons[0] if sc.addons else None |