229 lines
8.1 KiB
Python
229 lines
8.1 KiB
Python
from __future__ import annotations
|
|
|
|
from unittest.mock import patch
|
|
|
|
from tests.helpers import PluginsTestCase, temporary_root
|
|
from tutor import config as tutor_config
|
|
from tutor import exceptions, fmt, hooks, plugins
|
|
from tutor.plugins import v0 as plugins_v0
|
|
from tutor.types import Config, get_typed
|
|
|
|
|
|
class PluginsTests(PluginsTestCase):
|
|
def test_iter_installed(self) -> None:
|
|
self.assertEqual([], list(plugins.iter_installed()))
|
|
|
|
def test_is_installed(self) -> None:
|
|
self.assertFalse(plugins.is_installed("dummy"))
|
|
|
|
def test_official_plugins(self) -> None:
|
|
# Create 2 official plugins
|
|
plugins_v0.OfficialPlugin("plugin1")
|
|
plugins_v0.OfficialPlugin("plugin2")
|
|
self.assertEqual(
|
|
["plugin1", "plugin2"],
|
|
list(plugins.iter_installed()),
|
|
)
|
|
|
|
def test_load(self) -> None:
|
|
config: Config = {tutor_config.PLUGINS_CONFIG_KEY: []}
|
|
plugins_v0.DictPlugin({"name": "plugin1"})
|
|
plugins_v0.DictPlugin({"name": "plugin2"})
|
|
plugins.load("plugin2")
|
|
plugins.load("plugin1")
|
|
tutor_config.save_enabled_plugins(config)
|
|
self.assertEqual(
|
|
["plugin1", "plugin2"], config[tutor_config.PLUGINS_CONFIG_KEY]
|
|
)
|
|
|
|
def test_enable_twice(self) -> None:
|
|
plugins_v0.DictPlugin({"name": "plugin1"})
|
|
plugins.load("plugin1")
|
|
plugins.load("plugin1")
|
|
config: Config = {tutor_config.PLUGINS_CONFIG_KEY: []}
|
|
tutor_config.save_enabled_plugins(config)
|
|
self.assertEqual(["plugin1"], config[tutor_config.PLUGINS_CONFIG_KEY])
|
|
|
|
def test_load_not_installed_plugin(self) -> None:
|
|
self.assertRaises(exceptions.TutorError, plugins.load, "plugin1")
|
|
|
|
def test_disable(self) -> None:
|
|
plugins_v0.DictPlugin(
|
|
{
|
|
"name": "plugin1",
|
|
"version": "1.0.0",
|
|
"config": {"set": {"KEY": "value"}},
|
|
}
|
|
)
|
|
plugins_v0.DictPlugin(
|
|
{
|
|
"name": "plugin2",
|
|
"version": "1.0.0",
|
|
}
|
|
)
|
|
config: Config = {"PLUGINS": ["plugin1", "plugin2"]}
|
|
tutor_config.enable_plugins(config)
|
|
with patch.object(fmt, "STDOUT"):
|
|
hooks.Actions.PLUGIN_UNLOADED.do("plugin1", "", config)
|
|
self.assertEqual(["plugin2"], config["PLUGINS"])
|
|
|
|
def test_disable_removes_set_config(self) -> None:
|
|
plugins_v0.DictPlugin(
|
|
{
|
|
"name": "plugin1",
|
|
"version": "1.0.0",
|
|
"config": {"set": {"KEY": "value"}},
|
|
}
|
|
)
|
|
config: Config = {"PLUGINS": ["plugin1"], "KEY": "value"}
|
|
tutor_config.enable_plugins(config)
|
|
with patch.object(fmt, "STDOUT"):
|
|
hooks.Actions.PLUGIN_UNLOADED.do("plugin1", "", config)
|
|
self.assertEqual([], config["PLUGINS"])
|
|
self.assertNotIn("KEY", config)
|
|
|
|
def test_patches(self) -> None:
|
|
plugins_v0.DictPlugin(
|
|
{"name": "plugin1", "patches": {"patch1": "Hello {{ ID }}"}}
|
|
)
|
|
plugins.load("plugin1")
|
|
patches = list(plugins.iter_patches("patch1"))
|
|
self.assertEqual(["Hello {{ ID }}"], patches)
|
|
|
|
def test_plugin_without_patches(self) -> None:
|
|
plugins_v0.DictPlugin({"name": "plugin1"})
|
|
plugins.load("plugin1")
|
|
patches = list(plugins.iter_patches("patch1"))
|
|
self.assertEqual([], patches)
|
|
|
|
def test_configure(self) -> None:
|
|
plugins_v0.DictPlugin(
|
|
{
|
|
"name": "plugin1",
|
|
"config": {
|
|
"add": {"PARAM1": "value1", "PARAM2": "value2"},
|
|
"set": {"PARAM3": "value3"},
|
|
"defaults": {"PARAM4": "value4"},
|
|
},
|
|
}
|
|
)
|
|
plugins.load("plugin1")
|
|
|
|
base = tutor_config.get_base()
|
|
defaults = tutor_config.get_defaults()
|
|
|
|
self.assertEqual(base["PARAM3"], "value3")
|
|
self.assertEqual(base["PLUGIN1_PARAM1"], "value1")
|
|
self.assertEqual(base["PLUGIN1_PARAM2"], "value2")
|
|
self.assertEqual(defaults["PLUGIN1_PARAM4"], "value4")
|
|
|
|
def test_configure_set_does_not_override(self) -> None:
|
|
config: Config = {"ID1": "oldid"}
|
|
|
|
plugins_v0.DictPlugin(
|
|
{"name": "plugin1", "config": {"set": {"ID1": "newid", "ID2": "id2"}}}
|
|
)
|
|
plugins.load("plugin1")
|
|
tutor_config.update_with_base(config)
|
|
|
|
self.assertEqual("oldid", config["ID1"])
|
|
self.assertEqual("id2", config["ID2"])
|
|
|
|
def test_configure_set_random_string(self) -> None:
|
|
plugins_v0.DictPlugin(
|
|
{
|
|
"name": "plugin1",
|
|
"config": {"set": {"PARAM1": "{{ 128|random_string }}"}},
|
|
}
|
|
)
|
|
plugins.load("plugin1")
|
|
config = tutor_config.get_base()
|
|
tutor_config.render_full(config)
|
|
|
|
self.assertEqual(128, len(get_typed(config, "PARAM1", str)))
|
|
|
|
def test_configure_default_value_with_previous_definition(self) -> None:
|
|
config: Config = {"PARAM1": "value"}
|
|
plugins_v0.DictPlugin(
|
|
{"name": "plugin1", "config": {"defaults": {"PARAM2": "{{ PARAM1 }}"}}}
|
|
)
|
|
plugins.load("plugin1")
|
|
tutor_config.update_with_defaults(config)
|
|
self.assertEqual("{{ PARAM1 }}", config["PLUGIN1_PARAM2"])
|
|
|
|
def test_config_load_from_plugins(self) -> None:
|
|
config: Config = {}
|
|
|
|
plugins_v0.DictPlugin(
|
|
{"name": "plugin1", "config": {"add": {"PARAM1": "{{ 10|random_string }}"}}}
|
|
)
|
|
plugins.load("plugin1")
|
|
|
|
tutor_config.update_with_base(config)
|
|
tutor_config.update_with_defaults(config)
|
|
tutor_config.render_full(config)
|
|
value1 = get_typed(config, "PLUGIN1_PARAM1", str)
|
|
|
|
self.assertEqual(10, len(value1))
|
|
|
|
def test_init_tasks(self) -> None:
|
|
plugins_v0.DictPlugin({"name": "plugin1", "hooks": {"init": ["myclient"]}})
|
|
with patch.object(
|
|
plugins_v0.env, "read_template_file", return_value="echo hello"
|
|
) as mock_read_template:
|
|
plugins.load("plugin1")
|
|
mock_read_template.assert_called_once_with(
|
|
"plugin1", "hooks", "myclient", "init"
|
|
)
|
|
|
|
self.assertIn(
|
|
("myclient", "echo hello"),
|
|
list(hooks.Filters.CLI_DO_INIT_TASKS.iterate()),
|
|
)
|
|
|
|
def test_plugins_are_updated_on_config_change(self) -> None:
|
|
config: Config = {}
|
|
plugins_v0.DictPlugin({"name": "plugin1"})
|
|
tutor_config.enable_plugins(config)
|
|
plugins1 = list(plugins.iter_loaded())
|
|
config["PLUGINS"] = ["plugin1"]
|
|
tutor_config.enable_plugins(config)
|
|
plugins2 = list(plugins.iter_loaded())
|
|
|
|
self.assertEqual([], plugins1)
|
|
self.assertEqual(1, len(plugins2))
|
|
|
|
def test_dict_plugin(self) -> None:
|
|
plugin = plugins_v0.DictPlugin(
|
|
{"name": "myplugin", "config": {"set": {"KEY": "value"}}, "version": "0.1"}
|
|
)
|
|
plugins.load("myplugin")
|
|
overriden_items = hooks.Filters.CONFIG_OVERRIDES.apply([])
|
|
versions = list(plugins.iter_info())
|
|
self.assertEqual("myplugin", plugin.name)
|
|
self.assertEqual([("myplugin", "0.1")], versions)
|
|
self.assertEqual([("KEY", "value")], overriden_items)
|
|
|
|
def test_config_disable_plugin(self) -> None:
|
|
plugins_v0.DictPlugin(
|
|
{"name": "plugin1", "config": {"set": {"KEY1": "value1"}}}
|
|
)
|
|
plugins_v0.DictPlugin(
|
|
{"name": "plugin2", "config": {"set": {"KEY2": "value2"}}}
|
|
)
|
|
plugins.load("plugin1")
|
|
plugins.load("plugin2")
|
|
|
|
with temporary_root() as root:
|
|
config = tutor_config.load_minimal(root)
|
|
config_pre = config.copy()
|
|
with patch.object(fmt, "STDOUT"):
|
|
hooks.Actions.PLUGIN_UNLOADED.do("plugin1", "", config)
|
|
config_post = tutor_config.load_minimal(root)
|
|
|
|
self.assertEqual("value1", config_pre["KEY1"])
|
|
self.assertEqual("value2", config_pre["KEY2"])
|
|
self.assertNotIn("KEY1", config)
|
|
self.assertNotIn("KEY1", config_post)
|
|
self.assertEqual("value2", config["KEY2"])
|