integration-cli/docker_cli_volume_driver_compat_unix_test.go
fb6eda91
 // +build !windows
 
 package main
 
 import (
 	"encoding/json"
 	"fmt"
 	"io"
 	"io/ioutil"
 	"net/http"
 	"net/http/httptest"
 	"os"
 	"path/filepath"
 	"strings"
 
 	"github.com/docker/docker/pkg/integration/checker"
 
 	"github.com/go-check/check"
 )
 
 func init() {
 	check.Suite(&DockerExternalVolumeSuiteCompatV1_1{
 		ds: &DockerSuite{},
 	})
 }
 
0bc43280
 type vol struct {
 	Name       string
 	Mountpoint string
 	Opts       map[string]string
 }
 
fb6eda91
 type DockerExternalVolumeSuiteCompatV1_1 struct {
0bc43280
 	server  *httptest.Server
 	ds      *DockerSuite
 	d       *Daemon
 	ec      *eventCounter
 	volList []vol
fb6eda91
 }
 
 func (s *DockerExternalVolumeSuiteCompatV1_1) SetUpTest(c *check.C) {
 	s.d = NewDaemon(c)
 	s.ec = &eventCounter{}
 }
 
 func (s *DockerExternalVolumeSuiteCompatV1_1) TearDownTest(c *check.C) {
 	s.d.Stop()
 	s.ds.TearDownTest(c)
 }
 
 func (s *DockerExternalVolumeSuiteCompatV1_1) SetUpSuite(c *check.C) {
 	mux := http.NewServeMux()
 	s.server = httptest.NewServer(mux)
 
 	type pluginRequest struct {
 		Name string
0bc43280
 		Opts map[string]string
fb6eda91
 	}
 
 	type pluginResp struct {
 		Mountpoint string `json:",omitempty"`
 		Err        string `json:",omitempty"`
 	}
 
 	read := func(b io.ReadCloser) (pluginRequest, error) {
 		defer b.Close()
 		var pr pluginRequest
 		if err := json.NewDecoder(b).Decode(&pr); err != nil {
 			return pr, err
 		}
 		return pr, nil
 	}
 
 	send := func(w http.ResponseWriter, data interface{}) {
 		switch t := data.(type) {
 		case error:
 			http.Error(w, t.Error(), 500)
 		case string:
 			w.Header().Set("Content-Type", "application/vnd.docker.plugins.v1+json")
 			fmt.Fprintln(w, t)
 		default:
 			w.Header().Set("Content-Type", "application/vnd.docker.plugins.v1+json")
 			json.NewEncoder(w).Encode(&data)
 		}
 	}
 
 	mux.HandleFunc("/Plugin.Activate", func(w http.ResponseWriter, r *http.Request) {
 		s.ec.activations++
 		send(w, `{"Implements": ["VolumeDriver"]}`)
 	})
 
 	mux.HandleFunc("/VolumeDriver.Create", func(w http.ResponseWriter, r *http.Request) {
 		s.ec.creations++
 		pr, err := read(r.Body)
 		if err != nil {
 			send(w, err)
 			return
 		}
0bc43280
 		s.volList = append(s.volList, vol{Name: pr.Name, Opts: pr.Opts})
fb6eda91
 		send(w, nil)
 	})
 
 	mux.HandleFunc("/VolumeDriver.Remove", func(w http.ResponseWriter, r *http.Request) {
 		s.ec.removals++
 		pr, err := read(r.Body)
 		if err != nil {
 			send(w, err)
 			return
 		}
 
 		if err := os.RemoveAll(hostVolumePath(pr.Name)); err != nil {
 			send(w, &pluginResp{Err: err.Error()})
 			return
 		}
 
0bc43280
 		for i, v := range s.volList {
fb6eda91
 			if v.Name == pr.Name {
 				if err := os.RemoveAll(hostVolumePath(v.Name)); err != nil {
 					send(w, fmt.Sprintf(`{"Err": "%v"}`, err))
 					return
 				}
0bc43280
 				s.volList = append(s.volList[:i], s.volList[i+1:]...)
fb6eda91
 				break
 			}
 		}
 		send(w, nil)
 	})
 
 	mux.HandleFunc("/VolumeDriver.Path", func(w http.ResponseWriter, r *http.Request) {
 		s.ec.paths++
 
 		pr, err := read(r.Body)
 		if err != nil {
 			send(w, err)
 			return
 		}
 		p := hostVolumePath(pr.Name)
 		send(w, &pluginResp{Mountpoint: p})
 	})
 
 	mux.HandleFunc("/VolumeDriver.Mount", func(w http.ResponseWriter, r *http.Request) {
 		s.ec.mounts++
 
 		pr, err := read(r.Body)
 		if err != nil {
 			send(w, err)
 			return
 		}
 
 		p := hostVolumePath(pr.Name)
 		if err := os.MkdirAll(p, 0755); err != nil {
 			send(w, &pluginResp{Err: err.Error()})
 			return
 		}
 
 		if err := ioutil.WriteFile(filepath.Join(p, "test"), []byte(s.server.URL), 0644); err != nil {
 			send(w, err)
 			return
 		}
 
 		send(w, &pluginResp{Mountpoint: p})
 	})
 
 	mux.HandleFunc("/VolumeDriver.Unmount", func(w http.ResponseWriter, r *http.Request) {
 		s.ec.unmounts++
 
 		_, err := read(r.Body)
 		if err != nil {
 			send(w, err)
 			return
 		}
 
 		send(w, nil)
 	})
 
 	err := os.MkdirAll("/etc/docker/plugins", 0755)
 	c.Assert(err, checker.IsNil)
 
 	err = ioutil.WriteFile("/etc/docker/plugins/test-external-volume-driver.spec", []byte(s.server.URL), 0644)
 	c.Assert(err, checker.IsNil)
 }
 
 func (s *DockerExternalVolumeSuiteCompatV1_1) TearDownSuite(c *check.C) {
 	s.server.Close()
 
 	err := os.RemoveAll("/etc/docker/plugins")
 	c.Assert(err, checker.IsNil)
 }
 
 func (s *DockerExternalVolumeSuiteCompatV1_1) TestExternalVolumeDriverCompatV1_1(c *check.C) {
 	err := s.d.StartWithBusybox()
 	c.Assert(err, checker.IsNil)
 
 	out, err := s.d.Cmd("run", "--name=test", "-v", "foo:/bar", "--volume-driver", "test-external-volume-driver", "busybox", "sh", "-c", "echo hello > /bar/hello")
 	c.Assert(err, checker.IsNil, check.Commentf(out))
 	out, err = s.d.Cmd("rm", "test")
 	c.Assert(err, checker.IsNil, check.Commentf(out))
 
 	out, err = s.d.Cmd("run", "--name=test2", "-v", "foo:/bar", "busybox", "cat", "/bar/hello")
 	c.Assert(err, checker.IsNil, check.Commentf(out))
 	c.Assert(strings.TrimSpace(out), checker.Equals, "hello")
 
 	err = s.d.Restart()
 	c.Assert(err, checker.IsNil)
 
 	out, err = s.d.Cmd("start", "-a", "test2")
 	c.Assert(strings.TrimSpace(out), checker.Equals, "hello")
 
 	out, err = s.d.Cmd("rm", "test2")
 	c.Assert(err, checker.IsNil, check.Commentf(out))
 
 	out, err = s.d.Cmd("volume", "inspect", "foo")
 	c.Assert(err, checker.IsNil, check.Commentf(out))
 
 	out, err = s.d.Cmd("volume", "rm", "foo")
 	c.Assert(err, checker.IsNil, check.Commentf(out))
 }
0bc43280
 
 func (s *DockerExternalVolumeSuiteCompatV1_1) TestExternalVolumeDriverCompatOptionsV1_1(c *check.C) {
 	err := s.d.StartWithBusybox()
 	c.Assert(err, checker.IsNil)
 
 	out, err := s.d.Cmd("volume", "create", "--name", "optvol", "--driver", "test-external-volume-driver", "--opt", "opt1=opt1val", "--opt", "opt2=opt2val")
 	c.Assert(err, checker.IsNil, check.Commentf(out))
 
 	out, err = s.d.Cmd("volume", "inspect", "optvol")
 	c.Assert(err, checker.IsNil, check.Commentf(out))
 
 	c.Assert(s.volList[0].Opts["opt1"], checker.Equals, "opt1val")
 	c.Assert(s.volList[0].Opts["opt2"], checker.Equals, "opt2val")
 
 	out, err = s.d.Cmd("volume", "rm", "optvol")
 	c.Assert(err, checker.IsNil, check.Commentf(out))
 }