mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
e4b6adc88e
This cleans up some of the package API's used for interacting with volumes, and simplifies management. Signed-off-by: Brian Goff <cpuguy83@gmail.com>
253 lines
7.9 KiB
Go
253 lines
7.9 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/docker/docker/api/types/filters"
|
|
"github.com/docker/docker/errdefs"
|
|
"github.com/docker/docker/volume"
|
|
volumedrivers "github.com/docker/docker/volume/drivers"
|
|
"github.com/docker/docker/volume/service/opts"
|
|
"github.com/docker/docker/volume/testutils"
|
|
"github.com/gotestyourself/gotestyourself/assert"
|
|
is "github.com/gotestyourself/gotestyourself/assert/cmp"
|
|
)
|
|
|
|
func TestServiceCreate(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ds := volumedrivers.NewStore(nil)
|
|
assert.Assert(t, ds.Register(testutils.NewFakeDriver("d1"), "d1"))
|
|
assert.Assert(t, ds.Register(testutils.NewFakeDriver("d2"), "d2"))
|
|
|
|
ctx := context.Background()
|
|
service, cleanup := newTestService(t, ds)
|
|
defer cleanup()
|
|
|
|
_, err := service.Create(ctx, "v1", "notexist")
|
|
assert.Assert(t, errdefs.IsNotFound(err), err)
|
|
|
|
v, err := service.Create(ctx, "v1", "d1")
|
|
assert.Assert(t, err)
|
|
|
|
vCopy, err := service.Create(ctx, "v1", "d1")
|
|
assert.Assert(t, err)
|
|
assert.Assert(t, is.DeepEqual(v, vCopy))
|
|
|
|
_, err = service.Create(ctx, "v1", "d2")
|
|
assert.Check(t, IsNameConflict(err), err)
|
|
assert.Check(t, errdefs.IsConflict(err), err)
|
|
|
|
assert.Assert(t, service.Remove(ctx, "v1"))
|
|
_, err = service.Create(ctx, "v1", "d2")
|
|
assert.Assert(t, err)
|
|
_, err = service.Create(ctx, "v1", "d2")
|
|
assert.Assert(t, err)
|
|
|
|
}
|
|
|
|
func TestServiceList(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ds := volumedrivers.NewStore(nil)
|
|
assert.Assert(t, ds.Register(testutils.NewFakeDriver("d1"), "d1"))
|
|
assert.Assert(t, ds.Register(testutils.NewFakeDriver("d2"), "d2"))
|
|
|
|
service, cleanup := newTestService(t, ds)
|
|
defer cleanup()
|
|
|
|
ctx := context.Background()
|
|
|
|
_, err := service.Create(ctx, "v1", "d1")
|
|
assert.Assert(t, err)
|
|
_, err = service.Create(ctx, "v2", "d1")
|
|
assert.Assert(t, err)
|
|
_, err = service.Create(ctx, "v3", "d2")
|
|
assert.Assert(t, err)
|
|
|
|
ls, _, err := service.List(ctx, filters.NewArgs(filters.Arg("driver", "d1")))
|
|
assert.Assert(t, err)
|
|
assert.Check(t, is.Len(ls, 2))
|
|
|
|
ls, _, err = service.List(ctx, filters.NewArgs(filters.Arg("driver", "d2")))
|
|
assert.Assert(t, err)
|
|
assert.Check(t, is.Len(ls, 1))
|
|
|
|
ls, _, err = service.List(ctx, filters.NewArgs(filters.Arg("driver", "notexist")))
|
|
assert.Assert(t, err)
|
|
assert.Check(t, is.Len(ls, 0))
|
|
|
|
ls, _, err = service.List(ctx, filters.NewArgs(filters.Arg("dangling", "true")))
|
|
assert.Assert(t, err)
|
|
assert.Check(t, is.Len(ls, 3))
|
|
ls, _, err = service.List(ctx, filters.NewArgs(filters.Arg("dangling", "false")))
|
|
assert.Assert(t, err)
|
|
assert.Check(t, is.Len(ls, 0))
|
|
|
|
_, err = service.Get(ctx, "v1", opts.WithGetReference("foo"))
|
|
assert.Assert(t, err)
|
|
ls, _, err = service.List(ctx, filters.NewArgs(filters.Arg("dangling", "true")))
|
|
assert.Assert(t, err)
|
|
assert.Check(t, is.Len(ls, 2))
|
|
ls, _, err = service.List(ctx, filters.NewArgs(filters.Arg("dangling", "false")))
|
|
assert.Assert(t, err)
|
|
assert.Check(t, is.Len(ls, 1))
|
|
|
|
ls, _, err = service.List(ctx, filters.NewArgs(filters.Arg("dangling", "false"), filters.Arg("driver", "d2")))
|
|
assert.Assert(t, err)
|
|
assert.Check(t, is.Len(ls, 0))
|
|
ls, _, err = service.List(ctx, filters.NewArgs(filters.Arg("dangling", "true"), filters.Arg("driver", "d2")))
|
|
assert.Assert(t, err)
|
|
assert.Check(t, is.Len(ls, 1))
|
|
}
|
|
|
|
func TestServiceRemove(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ds := volumedrivers.NewStore(nil)
|
|
assert.Assert(t, ds.Register(testutils.NewFakeDriver("d1"), "d1"))
|
|
|
|
service, cleanup := newTestService(t, ds)
|
|
defer cleanup()
|
|
ctx := context.Background()
|
|
|
|
_, err := service.Create(ctx, "test", "d1")
|
|
assert.Assert(t, err)
|
|
|
|
assert.Assert(t, service.Remove(ctx, "test"))
|
|
assert.Assert(t, service.Remove(ctx, "test", opts.WithPurgeOnError(true)))
|
|
}
|
|
|
|
func TestServiceGet(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ds := volumedrivers.NewStore(nil)
|
|
assert.Assert(t, ds.Register(testutils.NewFakeDriver("d1"), "d1"))
|
|
|
|
service, cleanup := newTestService(t, ds)
|
|
defer cleanup()
|
|
ctx := context.Background()
|
|
|
|
v, err := service.Get(ctx, "notexist")
|
|
assert.Assert(t, IsNotExist(err))
|
|
assert.Check(t, v == nil)
|
|
|
|
created, err := service.Create(ctx, "test", "d1")
|
|
assert.Assert(t, err)
|
|
assert.Assert(t, created != nil)
|
|
|
|
v, err = service.Get(ctx, "test")
|
|
assert.Assert(t, err)
|
|
assert.Assert(t, is.DeepEqual(created, v))
|
|
|
|
v, err = service.Get(ctx, "test", opts.WithGetResolveStatus)
|
|
assert.Assert(t, err)
|
|
assert.Assert(t, is.Len(v.Status, 1), v.Status)
|
|
|
|
v, err = service.Get(ctx, "test", opts.WithGetDriver("notarealdriver"))
|
|
assert.Assert(t, errdefs.IsConflict(err), err)
|
|
v, err = service.Get(ctx, "test", opts.WithGetDriver("d1"))
|
|
assert.Assert(t, err == nil)
|
|
assert.Assert(t, is.DeepEqual(created, v))
|
|
|
|
assert.Assert(t, ds.Register(testutils.NewFakeDriver("d2"), "d2"))
|
|
v, err = service.Get(ctx, "test", opts.WithGetDriver("d2"))
|
|
assert.Assert(t, errdefs.IsConflict(err), err)
|
|
}
|
|
|
|
func TestServicePrune(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ds := volumedrivers.NewStore(nil)
|
|
assert.Assert(t, ds.Register(testutils.NewFakeDriver(volume.DefaultDriverName), volume.DefaultDriverName))
|
|
assert.Assert(t, ds.Register(testutils.NewFakeDriver("other"), "other"))
|
|
|
|
service, cleanup := newTestService(t, ds)
|
|
defer cleanup()
|
|
ctx := context.Background()
|
|
|
|
_, err := service.Create(ctx, "test", volume.DefaultDriverName)
|
|
assert.Assert(t, err)
|
|
_, err = service.Create(ctx, "test2", "other")
|
|
assert.Assert(t, err)
|
|
|
|
pr, err := service.Prune(ctx, filters.NewArgs(filters.Arg("label", "banana")))
|
|
assert.Assert(t, err)
|
|
assert.Assert(t, is.Len(pr.VolumesDeleted, 0))
|
|
|
|
pr, err = service.Prune(ctx, filters.NewArgs())
|
|
assert.Assert(t, err)
|
|
assert.Assert(t, is.Len(pr.VolumesDeleted, 1))
|
|
assert.Assert(t, is.Equal(pr.VolumesDeleted[0], "test"))
|
|
|
|
_, err = service.Get(ctx, "test")
|
|
assert.Assert(t, IsNotExist(err), err)
|
|
|
|
v, err := service.Get(ctx, "test2")
|
|
assert.Assert(t, err)
|
|
assert.Assert(t, is.Equal(v.Driver, "other"))
|
|
|
|
_, err = service.Create(ctx, "test", volume.DefaultDriverName)
|
|
assert.Assert(t, err)
|
|
|
|
pr, err = service.Prune(ctx, filters.NewArgs(filters.Arg("label!", "banana")))
|
|
assert.Assert(t, err)
|
|
assert.Assert(t, is.Len(pr.VolumesDeleted, 1))
|
|
assert.Assert(t, is.Equal(pr.VolumesDeleted[0], "test"))
|
|
v, err = service.Get(ctx, "test2")
|
|
assert.Assert(t, err)
|
|
assert.Assert(t, is.Equal(v.Driver, "other"))
|
|
|
|
_, err = service.Create(ctx, "test", volume.DefaultDriverName, opts.WithCreateLabels(map[string]string{"banana": ""}))
|
|
assert.Assert(t, err)
|
|
pr, err = service.Prune(ctx, filters.NewArgs(filters.Arg("label!", "banana")))
|
|
assert.Assert(t, err)
|
|
assert.Assert(t, is.Len(pr.VolumesDeleted, 0))
|
|
|
|
_, err = service.Create(ctx, "test3", volume.DefaultDriverName, opts.WithCreateLabels(map[string]string{"banana": "split"}))
|
|
assert.Assert(t, err)
|
|
pr, err = service.Prune(ctx, filters.NewArgs(filters.Arg("label!", "banana=split")))
|
|
assert.Assert(t, err)
|
|
assert.Assert(t, is.Len(pr.VolumesDeleted, 1))
|
|
assert.Assert(t, is.Equal(pr.VolumesDeleted[0], "test"))
|
|
|
|
pr, err = service.Prune(ctx, filters.NewArgs(filters.Arg("label", "banana=split")))
|
|
assert.Assert(t, err)
|
|
assert.Assert(t, is.Len(pr.VolumesDeleted, 1))
|
|
assert.Assert(t, is.Equal(pr.VolumesDeleted[0], "test3"))
|
|
|
|
v, err = service.Create(ctx, "test", volume.DefaultDriverName, opts.WithCreateReference(t.Name()))
|
|
assert.Assert(t, err)
|
|
|
|
pr, err = service.Prune(ctx, filters.NewArgs())
|
|
assert.Assert(t, err)
|
|
assert.Assert(t, is.Len(pr.VolumesDeleted, 0))
|
|
assert.Assert(t, service.Release(ctx, v.Name, t.Name()))
|
|
|
|
pr, err = service.Prune(ctx, filters.NewArgs())
|
|
assert.Assert(t, err)
|
|
assert.Assert(t, is.Len(pr.VolumesDeleted, 1))
|
|
assert.Assert(t, is.Equal(pr.VolumesDeleted[0], "test"))
|
|
}
|
|
|
|
func newTestService(t *testing.T, ds *volumedrivers.Store) (*VolumesService, func()) {
|
|
t.Helper()
|
|
|
|
dir, err := ioutil.TempDir("", t.Name())
|
|
assert.Assert(t, err)
|
|
|
|
store, err := NewStore(dir, ds)
|
|
assert.Assert(t, err)
|
|
s := &VolumesService{vs: store, eventLogger: dummyEventLogger{}}
|
|
return s, func() {
|
|
assert.Check(t, s.Shutdown())
|
|
assert.Check(t, os.RemoveAll(dir))
|
|
}
|
|
}
|
|
|
|
type dummyEventLogger struct{}
|
|
|
|
func (dummyEventLogger) LogVolumeEvent(_, _ string, _ map[string]string) {}
|