mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-21 06:10:23 -04:00
80 lines
2 KiB
Go
80 lines
2 KiB
Go
![]() |
package plugin
|
||
|
|
||
|
import (
|
||
|
"io/ioutil"
|
||
|
"os"
|
||
|
"path/filepath"
|
||
|
"testing"
|
||
|
|
||
|
"github.com/docker/docker/api/types"
|
||
|
"github.com/docker/docker/pkg/mount"
|
||
|
"github.com/docker/docker/pkg/system"
|
||
|
"github.com/docker/docker/plugin/v2"
|
||
|
)
|
||
|
|
||
|
func TestManagerWithPluginMounts(t *testing.T) {
|
||
|
root, err := ioutil.TempDir("", "test-store-with-plugin-mounts")
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
defer system.EnsureRemoveAll(root)
|
||
|
|
||
|
s := NewStore()
|
||
|
managerRoot := filepath.Join(root, "manager")
|
||
|
p1 := newTestPlugin(t, "test1", "testcap", managerRoot)
|
||
|
|
||
|
p2 := newTestPlugin(t, "test2", "testcap", managerRoot)
|
||
|
p2.PluginObj.Enabled = true
|
||
|
|
||
|
m, err := NewManager(
|
||
|
ManagerConfig{
|
||
|
Store: s,
|
||
|
Root: managerRoot,
|
||
|
ExecRoot: filepath.Join(root, "exec"),
|
||
|
CreateExecutor: func(*Manager) (Executor, error) { return nil, nil },
|
||
|
LogPluginEvent: func(_, _, _ string) {},
|
||
|
})
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
|
||
|
if err := s.Add(p1); err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
if err := s.Add(p2); err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
|
||
|
// Create a mount to simulate a plugin that has created it's own mounts
|
||
|
p2Mount := filepath.Join(p2.Rootfs, "testmount")
|
||
|
if err := os.MkdirAll(p2Mount, 0755); err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
if err := mount.Mount("tmpfs", p2Mount, "tmpfs", ""); err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
|
||
|
if err := m.Remove(p1.Name(), &types.PluginRmConfig{ForceRemove: true}); err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
if mounted, err := mount.Mounted(p2Mount); !mounted || err != nil {
|
||
|
t.Fatalf("expected %s to be mounted, err: %v", p2Mount, err)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func newTestPlugin(t *testing.T, name, cap, root string) *v2.Plugin {
|
||
|
rootfs := filepath.Join(root, name)
|
||
|
if err := os.MkdirAll(rootfs, 0755); err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
|
||
|
p := v2.Plugin{PluginObj: types.Plugin{Name: name}}
|
||
|
p.Rootfs = rootfs
|
||
|
iType := types.PluginInterfaceType{Capability: cap, Prefix: "docker", Version: "1.0"}
|
||
|
i := types.PluginConfigInterface{Socket: "plugins.sock", Types: []types.PluginInterfaceType{iType}}
|
||
|
p.PluginObj.Config.Interface = i
|
||
|
p.PluginObj.ID = name
|
||
|
|
||
|
return &p
|
||
|
}
|