1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98
|
package shim
import (
"bytes"
"encoding/json"
"net/http"
"testing"
"github.com/docker/docker/pkg/containerfs"
"github.com/docker/docker/daemon/graphdriver"
"github.com/docker/docker/pkg/idtools"
"github.com/docker/go-connections/sockets"
graphPlugin "github.com/docker/go-plugins-helpers/graphdriver"
)
type testGraphDriver struct{}
// ProtoDriver
var _ graphdriver.ProtoDriver = &testGraphDriver{}
func (t *testGraphDriver) String() string {
return ""
}
// FIXME(samoht): this doesn't seem to be called by the plugins
func (t *testGraphDriver) CreateReadWrite(id, parent string, opts *graphdriver.CreateOpts) error {
return nil
}
func (t *testGraphDriver) Create(id, parent string, opts *graphdriver.CreateOpts) error {
return nil
}
func (t *testGraphDriver) Remove(id string) error {
return nil
}
func (t *testGraphDriver) Get(id, mountLabel string) (dir containerfs.ContainerFS, err error) {
return containerfs.NewLocalContainerFS(""), nil
}
func (t *testGraphDriver) Put(id string) error {
return nil
}
func (t *testGraphDriver) Exists(id string) bool {
return false
}
func (t *testGraphDriver) Status() [][2]string {
return nil
}
func (t *testGraphDriver) GetMetadata(id string) (map[string]string, error) {
return nil, nil
}
func (t *testGraphDriver) Cleanup() error {
return nil
}
func (t *testGraphDriver) Capabilities() graphdriver.Capabilities {
return graphdriver.Capabilities{}
}
func Init(root string, options []string, uidMaps, gidMaps []idtools.IDMap) (graphdriver.Driver, error) {
d := graphdriver.NewNaiveDiffDriver(&testGraphDriver{}, uidMaps, gidMaps)
return d, nil
}
func TestGraphDriver(t *testing.T) {
h := NewHandlerFromGraphDriver(Init)
l := sockets.NewInmemSocket("test", 0)
go h.Serve(l)
defer l.Close()
client := &http.Client{Transport: &http.Transport{
Dial: l.Dial,
}}
resp, err := pluginRequest(client, "/GraphDriver.Init", &graphPlugin.InitRequest{Home: "foo"})
if err != nil {
t.Fatal(err)
}
if resp.Err != "" {
t.Fatalf("error while creating GraphDriver: %v", err)
}
}
func pluginRequest(client *http.Client, method string, req *graphPlugin.InitRequest) (*graphPlugin.ErrorResponse, error) {
b, err := json.Marshal(req)
if err != nil {
return nil, err
}
resp, err := client.Post("http://localhost"+method, "application/json", bytes.NewReader(b))
if err != nil {
return nil, err
}
var gResp graphPlugin.ErrorResponse
err = json.NewDecoder(resp.Body).Decode(&gResp)
if err != nil {
return nil, err
}
return &gResp, nil
}
|