Hopiu/llm
1
0
Fork 0
mirror of https://github.com/Hopiu/llm.git synced 2026-05-13 08:13:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix plugins tests to account for default plugins

Browse source
This commit is contained in:
Simon Willison 2023-07-01 11:06:28 -07:00
parent 8f7450bd74
commit 14ce371007
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
tests/test_plugins.py
View file

@ -7,7 +7,11 @@ import pytest
def test_register_commands():
importlib.reload(cli)
assert plugins.get_plugins() == []
def plugin_names():
return [plugin["name"] for plugin in plugins.get_plugins()]
assert "HelloWorldPlugin" not in plugin_names()
class HelloWorldPlugin:
__name__ = "HelloWorldPlugin"
@ -23,9 +27,7 @@ def test_register_commands():
plugins.pm.register(HelloWorldPlugin(), name="HelloWorldPlugin")
importlib.reload(cli)
assert plugins.get_plugins() == [
{"name": "HelloWorldPlugin", "hooks": ["register_commands"]}
]
assert "HelloWorldPlugin" in plugin_names()
runner = CliRunner()
result = runner.invoke(cli.cli, ["hello-world"])
@ -35,4 +37,4 @@ def test_register_commands():
finally:
plugins.pm.unregister(name="HelloWorldPlugin")
importlib.reload(cli)
assert plugins.get_plugins() == []
assert "HelloWorldPlugin" not in plugin_names()