mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-21 14:20:23 -04:00
160 lines
3.3 KiB
Go
160 lines
3.3 KiB
Go
// +build !windows
|
|
|
|
package plugins // import "github.com/docker/docker/pkg/plugins"
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net"
|
|
"os"
|
|
"path/filepath"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"gotest.tools/assert"
|
|
)
|
|
|
|
func TestLocalSocket(t *testing.T) {
|
|
// TODO Windows: Enable a similar version for Windows named pipes
|
|
tmpdir, unregister := Setup(t)
|
|
defer unregister()
|
|
|
|
cases := []string{
|
|
filepath.Join(tmpdir, "echo.sock"),
|
|
filepath.Join(tmpdir, "echo", "echo.sock"),
|
|
}
|
|
|
|
for _, c := range cases {
|
|
if err := os.MkdirAll(filepath.Dir(c), 0755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
l, err := net.Listen("unix", c)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
r := newLocalRegistry()
|
|
p, err := r.Plugin("echo")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
pp, err := r.Plugin("echo")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !reflect.DeepEqual(p, pp) {
|
|
t.Fatalf("Expected %v, was %v\n", p, pp)
|
|
}
|
|
|
|
if p.name != "echo" {
|
|
t.Fatalf("Expected plugin `echo`, got %s\n", p.name)
|
|
}
|
|
|
|
addr := fmt.Sprintf("unix://%s", c)
|
|
if p.Addr != addr {
|
|
t.Fatalf("Expected plugin addr `%s`, got %s\n", addr, p.Addr)
|
|
}
|
|
if !p.TLSConfig.InsecureSkipVerify {
|
|
t.Fatalf("Expected TLS verification to be skipped")
|
|
}
|
|
l.Close()
|
|
}
|
|
}
|
|
|
|
func TestScan(t *testing.T) {
|
|
tmpdir, unregister := Setup(t)
|
|
defer unregister()
|
|
|
|
pluginNames, err := Scan()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if pluginNames != nil {
|
|
t.Fatal("Plugin names should be empty.")
|
|
}
|
|
|
|
path := filepath.Join(tmpdir, "echo.spec")
|
|
addr := "unix://var/lib/docker/plugins/echo.sock"
|
|
name := "echo"
|
|
|
|
err = os.MkdirAll(filepath.Dir(path), 0755)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = ioutil.WriteFile(path, []byte(addr), 0644)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
r := newLocalRegistry()
|
|
p, err := r.Plugin(name)
|
|
assert.NilError(t, err)
|
|
|
|
pluginNamesNotEmpty, err := Scan()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(pluginNamesNotEmpty) != 1 {
|
|
t.Fatalf("expected 1 plugin entry: %v", pluginNamesNotEmpty)
|
|
}
|
|
if p.Name() != pluginNamesNotEmpty[0] {
|
|
t.Fatalf("Unable to scan plugin with name %s", p.name)
|
|
}
|
|
}
|
|
|
|
func TestScanNotPlugins(t *testing.T) {
|
|
tmpdir, unregister := Setup(t)
|
|
defer unregister()
|
|
|
|
// not that `Setup()` above sets the sockets path and spec path dirs, which
|
|
// `Scan()` uses to find plugins to the returned `tmpdir`
|
|
|
|
notPlugin := filepath.Join(tmpdir, "not-a-plugin")
|
|
if err := os.MkdirAll(notPlugin, 0700); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// this is named differently than the dir it's in, so the scanner should ignore it
|
|
l, err := net.Listen("unix", filepath.Join(notPlugin, "foo.sock"))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer l.Close()
|
|
|
|
// same let's test a spec path
|
|
f, err := os.Create(filepath.Join(notPlugin, "foo.spec"))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer f.Close()
|
|
|
|
names, err := Scan()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(names) != 0 {
|
|
t.Fatalf("expected no plugins, got %v", names)
|
|
}
|
|
|
|
// Just as a sanity check, let's make an entry that the scanner should read
|
|
f, err = os.Create(filepath.Join(notPlugin, "not-a-plugin.spec"))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer f.Close()
|
|
|
|
names, err = Scan()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(names) != 1 {
|
|
t.Fatalf("expected 1 entry in result: %v", names)
|
|
}
|
|
if names[0] != "not-a-plugin" {
|
|
t.Fatalf("expected plugin named `not-a-plugin`, got: %s", names[0])
|
|
}
|
|
}
|