248 lines
6.5 KiB
Go
248 lines
6.5 KiB
Go
// +build experimental
|
|
// +build !windows
|
|
|
|
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/go-check/check"
|
|
)
|
|
|
|
func init() {
|
|
check.Suite(&DockerExternalVolumeSuite{
|
|
ds: &DockerSuite{},
|
|
})
|
|
}
|
|
|
|
type eventCounter struct {
|
|
activations int
|
|
creations int
|
|
removals int
|
|
mounts int
|
|
unmounts int
|
|
paths int
|
|
}
|
|
|
|
type DockerExternalVolumeSuite struct {
|
|
server *httptest.Server
|
|
ds *DockerSuite
|
|
d *Daemon
|
|
ec *eventCounter
|
|
}
|
|
|
|
func (s *DockerExternalVolumeSuite) SetUpTest(c *check.C) {
|
|
s.d = NewDaemon(c)
|
|
s.ec = &eventCounter{}
|
|
}
|
|
|
|
func (s *DockerExternalVolumeSuite) TearDownTest(c *check.C) {
|
|
s.d.Stop()
|
|
s.ds.TearDownTest(c)
|
|
}
|
|
|
|
func (s *DockerExternalVolumeSuite) SetUpSuite(c *check.C) {
|
|
mux := http.NewServeMux()
|
|
s.server = httptest.NewServer(mux)
|
|
|
|
type pluginRequest struct {
|
|
name string
|
|
}
|
|
|
|
mux.HandleFunc("/Plugin.Activate", func(w http.ResponseWriter, r *http.Request) {
|
|
s.ec.activations++
|
|
|
|
w.Header().Set("Content-Type", "application/vnd.docker.plugins.v1+json")
|
|
fmt.Fprintln(w, `{"Implements": ["VolumeDriver"]}`)
|
|
})
|
|
|
|
mux.HandleFunc("/VolumeDriver.Create", func(w http.ResponseWriter, r *http.Request) {
|
|
s.ec.creations++
|
|
|
|
w.Header().Set("Content-Type", "application/vnd.docker.plugins.v1+json")
|
|
fmt.Fprintln(w, `{}`)
|
|
})
|
|
|
|
mux.HandleFunc("/VolumeDriver.Remove", func(w http.ResponseWriter, r *http.Request) {
|
|
s.ec.removals++
|
|
|
|
w.Header().Set("Content-Type", "application/vnd.docker.plugins.v1+json")
|
|
fmt.Fprintln(w, `{}`)
|
|
})
|
|
|
|
mux.HandleFunc("/VolumeDriver.Path", func(w http.ResponseWriter, r *http.Request) {
|
|
s.ec.paths++
|
|
|
|
var pr pluginRequest
|
|
if err := json.NewDecoder(r.Body).Decode(&pr); err != nil {
|
|
http.Error(w, err.Error(), 500)
|
|
}
|
|
|
|
p := hostVolumePath(pr.name)
|
|
|
|
w.Header().Set("Content-Type", "application/vnd.docker.plugins.v1+json")
|
|
fmt.Fprintln(w, fmt.Sprintf("{\"Mountpoint\": \"%s\"}", p))
|
|
})
|
|
|
|
mux.HandleFunc("/VolumeDriver.Mount", func(w http.ResponseWriter, r *http.Request) {
|
|
s.ec.mounts++
|
|
|
|
var pr pluginRequest
|
|
if err := json.NewDecoder(r.Body).Decode(&pr); err != nil {
|
|
http.Error(w, err.Error(), 500)
|
|
}
|
|
|
|
p := hostVolumePath(pr.name)
|
|
if err := os.MkdirAll(p, 0755); err != nil {
|
|
http.Error(w, err.Error(), 500)
|
|
}
|
|
|
|
if err := ioutil.WriteFile(filepath.Join(p, "test"), []byte(s.server.URL), 0644); err != nil {
|
|
http.Error(w, err.Error(), 500)
|
|
}
|
|
|
|
w.Header().Set("Content-Type", "application/vnd.docker.plugins.v1+json")
|
|
fmt.Fprintln(w, fmt.Sprintf("{\"Mountpoint\": \"%s\"}", p))
|
|
})
|
|
|
|
mux.HandleFunc("/VolumeDriver.Unmount", func(w http.ResponseWriter, r *http.Request) {
|
|
s.ec.unmounts++
|
|
|
|
var pr pluginRequest
|
|
if err := json.NewDecoder(r.Body).Decode(&pr); err != nil {
|
|
http.Error(w, err.Error(), 500)
|
|
}
|
|
|
|
p := hostVolumePath(pr.name)
|
|
if err := os.RemoveAll(p); err != nil {
|
|
http.Error(w, err.Error(), 500)
|
|
}
|
|
|
|
w.Header().Set("Content-Type", "application/vnd.docker.plugins.v1+json")
|
|
fmt.Fprintln(w, `{}`)
|
|
})
|
|
|
|
if err := os.MkdirAll("/usr/share/docker/plugins", 0755); err != nil {
|
|
c.Fatal(err)
|
|
}
|
|
|
|
if err := ioutil.WriteFile("/usr/share/docker/plugins/test-external-volume-driver.spec", []byte(s.server.URL), 0644); err != nil {
|
|
c.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func (s *DockerExternalVolumeSuite) TearDownSuite(c *check.C) {
|
|
s.server.Close()
|
|
|
|
if err := os.RemoveAll("/usr/share/docker/plugins"); err != nil {
|
|
c.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func (s *DockerExternalVolumeSuite) TestStartExternalNamedVolumeDriver(c *check.C) {
|
|
if err := s.d.StartWithBusybox(); err != nil {
|
|
c.Fatal(err)
|
|
}
|
|
|
|
out, err := s.d.Cmd("run", "--rm", "--name", "test-data", "-v", "external-volume-test:/tmp/external-volume-test", "--volume-driver", "test-external-volume-driver", "busybox:latest", "cat", "/tmp/external-volume-test/test")
|
|
if err != nil {
|
|
c.Fatal(err)
|
|
}
|
|
|
|
if !strings.Contains(out, s.server.URL) {
|
|
c.Fatalf("External volume mount failed. Output: %s\n", out)
|
|
}
|
|
|
|
p := hostVolumePath("external-volume-test")
|
|
_, err = os.Lstat(p)
|
|
if err == nil {
|
|
c.Fatalf("Expected error checking volume path in host: %s\n", p)
|
|
}
|
|
|
|
if !os.IsNotExist(err) {
|
|
c.Fatalf("Expected volume path in host to not exist: %s, %v\n", p, err)
|
|
}
|
|
|
|
c.Assert(s.ec.activations, check.Equals, 1)
|
|
c.Assert(s.ec.creations, check.Equals, 1)
|
|
c.Assert(s.ec.removals, check.Equals, 1)
|
|
c.Assert(s.ec.mounts, check.Equals, 1)
|
|
c.Assert(s.ec.unmounts, check.Equals, 1)
|
|
}
|
|
|
|
func (s *DockerExternalVolumeSuite) TestStartExternalVolumeUnnamedDriver(c *check.C) {
|
|
if err := s.d.StartWithBusybox(); err != nil {
|
|
c.Fatal(err)
|
|
}
|
|
|
|
out, err := s.d.Cmd("run", "--rm", "--name", "test-data", "-v", "/tmp/external-volume-test", "--volume-driver", "test-external-volume-driver", "busybox:latest", "cat", "/tmp/external-volume-test/test")
|
|
if err != nil {
|
|
c.Fatal(err)
|
|
}
|
|
|
|
if !strings.Contains(out, s.server.URL) {
|
|
c.Fatalf("External volume mount failed. Output: %s\n", out)
|
|
}
|
|
|
|
c.Assert(s.ec.activations, check.Equals, 1)
|
|
c.Assert(s.ec.creations, check.Equals, 1)
|
|
c.Assert(s.ec.removals, check.Equals, 1)
|
|
c.Assert(s.ec.mounts, check.Equals, 1)
|
|
c.Assert(s.ec.unmounts, check.Equals, 1)
|
|
}
|
|
|
|
func (s DockerExternalVolumeSuite) TestStartExternalVolumeDriverVolumesFrom(c *check.C) {
|
|
if err := s.d.StartWithBusybox(); err != nil {
|
|
c.Fatal(err)
|
|
}
|
|
|
|
if _, err := s.d.Cmd("run", "-d", "--name", "vol-test1", "-v", "/foo", "--volume-driver", "test-external-volume-driver", "busybox:latest"); err != nil {
|
|
c.Fatal(err)
|
|
}
|
|
|
|
if _, err := s.d.Cmd("run", "--rm", "--volumes-from", "vol-test1", "--name", "vol-test2", "busybox", "ls", "/tmp"); err != nil {
|
|
c.Fatal(err)
|
|
}
|
|
|
|
if _, err := s.d.Cmd("rm", "-f", "vol-test1"); err != nil {
|
|
c.Fatal(err)
|
|
}
|
|
|
|
c.Assert(s.ec.activations, check.Equals, 1)
|
|
c.Assert(s.ec.creations, check.Equals, 2)
|
|
c.Assert(s.ec.removals, check.Equals, 1)
|
|
c.Assert(s.ec.mounts, check.Equals, 2)
|
|
c.Assert(s.ec.unmounts, check.Equals, 2)
|
|
}
|
|
|
|
func (s DockerExternalVolumeSuite) TestStartExternalVolumeDriverDeleteContainer(c *check.C) {
|
|
if err := s.d.StartWithBusybox(); err != nil {
|
|
c.Fatal(err)
|
|
}
|
|
|
|
if _, err := s.d.Cmd("run", "-d", "--name", "vol-test1", "-v", "/foo", "--volume-driver", "test-external-volume-driver", "busybox:latest"); err != nil {
|
|
c.Fatal(err)
|
|
}
|
|
|
|
if _, err := s.d.Cmd("rm", "-fv", "vol-test1"); err != nil {
|
|
c.Fatal(err)
|
|
}
|
|
|
|
c.Assert(s.ec.activations, check.Equals, 1)
|
|
c.Assert(s.ec.creations, check.Equals, 1)
|
|
c.Assert(s.ec.removals, check.Equals, 1)
|
|
c.Assert(s.ec.mounts, check.Equals, 1)
|
|
c.Assert(s.ec.unmounts, check.Equals, 1)
|
|
}
|
|
|
|
func hostVolumePath(name string) string {
|
|
return fmt.Sprintf("/var/lib/docker/volumes/%s", name)
|
|
}
|