76 lines
1.9 KiB
Go
76 lines
1.9 KiB
Go
package catalog
|
|
|
|
import (
|
|
testing "github.com/mitchellh/go-testing-interface"
|
|
|
|
"github.com/hashicorp/nomad/helper/pluginutils/loader"
|
|
"github.com/hashicorp/nomad/helper/testlog"
|
|
"github.com/hashicorp/nomad/nomad/structs/config"
|
|
)
|
|
|
|
// TestPluginLoader returns a plugin loader populated only with internal plugins
|
|
func TestPluginLoader(t testing.T) loader.PluginCatalog {
|
|
driverConfigs := []*config.PluginConfig{
|
|
{
|
|
Name: "raw_exec",
|
|
Config: map[string]interface{}{
|
|
"enabled": true,
|
|
},
|
|
},
|
|
}
|
|
return TestPluginLoaderWithOptions(t, "", nil, driverConfigs)
|
|
}
|
|
|
|
// TestPluginLoaderWithOptions allows configuring the plugin loader fully.
|
|
func TestPluginLoaderWithOptions(t testing.T,
|
|
pluginDir string,
|
|
options map[string]string,
|
|
configs []*config.PluginConfig) loader.PluginCatalog {
|
|
|
|
// Get a logger
|
|
logger := testlog.HCLogger(t)
|
|
|
|
// Get the registered plugins
|
|
catalog := Catalog()
|
|
|
|
// Create our map of plugins
|
|
internal := make(map[loader.PluginID]*loader.InternalPluginConfig, len(catalog))
|
|
|
|
for id, reg := range catalog {
|
|
if reg.Config == nil {
|
|
logger.Warn("skipping loading internal plugin because it is missing its configuration", "plugin", id)
|
|
continue
|
|
}
|
|
|
|
pluginConfig := reg.Config.Config
|
|
if reg.ConfigLoader != nil {
|
|
pc, err := reg.ConfigLoader(options)
|
|
if err != nil {
|
|
t.Fatalf("failed to retrieve config for internal plugin %v: %v", id, err)
|
|
}
|
|
|
|
pluginConfig = pc
|
|
}
|
|
|
|
internal[id] = &loader.InternalPluginConfig{
|
|
Factory: reg.Config.Factory,
|
|
Config: pluginConfig,
|
|
}
|
|
}
|
|
|
|
// Build the plugin loader
|
|
config := &loader.PluginLoaderConfig{
|
|
Logger: logger,
|
|
PluginDir: "",
|
|
Configs: configs,
|
|
InternalPlugins: internal,
|
|
SupportedVersions: loader.AgentSupportedApiVersions,
|
|
}
|
|
l, err := loader.NewPluginLoader(config)
|
|
if err != nil {
|
|
t.Fatalf("failed to create plugin loader: %v", err)
|
|
}
|
|
|
|
return l
|
|
}
|