package plugins import ( "fmt" "io/ioutil" "net" "os" "path/filepath" "reflect" "testing" ) func TestUnknownLocalPath(t *testing.T) { tmpdir, err := ioutil.TempDir("", "docker-test") if err != nil { t.Fatal(err) } defer os.RemoveAll(tmpdir) l := newLocalRegistry(filepath.Join(tmpdir, "unknown")) _, err = l.Plugin("foo") if err == nil || err != ErrNotFound { t.Fatalf("Expected error for unknown directory") } } func TestLocalSocket(t *testing.T) { tmpdir, err := ioutil.TempDir("", "docker-test") if err != nil { t.Fatal(err) } defer os.RemoveAll(tmpdir) l, err := net.Listen("unix", filepath.Join(tmpdir, "echo.sock")) if err != nil { t.Fatal(err) } defer l.Close() r := newLocalRegistry(tmpdir) 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/echo.sock", tmpdir) if p.Addr != addr { t.Fatalf("Expected plugin addr `%s`, got %s\n", addr, p.Addr) } } func TestFileSpecPlugin(t *testing.T) { tmpdir, err := ioutil.TempDir("", "docker-test") if err != nil { t.Fatal(err) } defer os.RemoveAll(tmpdir) cases := []struct { path string name string addr string fail bool }{ {filepath.Join(tmpdir, "echo.spec"), "echo", "unix://var/lib/docker/plugins/echo.sock", false}, {filepath.Join(tmpdir, "foo.spec"), "foo", "tcp://localhost:8080", false}, {filepath.Join(tmpdir, "bar.spec"), "bar", "localhost:8080", true}, // unknown transport } for _, c := range cases { if err = ioutil.WriteFile(c.path, []byte(c.addr), 0644); err != nil { t.Fatal(err) } r := newLocalRegistry(tmpdir) p, err := r.Plugin(c.name) if c.fail && err == nil { continue } if err != nil { t.Fatal(err) } if p.Name != c.name { t.Fatalf("Expected plugin `%s`, got %s\n", c.name, p.Name) } if p.Addr != c.addr { t.Fatalf("Expected plugin addr `%s`, got %s\n", c.addr, p.Addr) } } }