cddfe04f6a
This enables docker cp and ADD/COPY docker build support for LCOW. Originally, the graphdriver.Get() interface returned a local path to the container root filesystem. This does not work for LCOW, so the Get() method now returns an interface that LCOW implements to support copying to and from the container. Signed-off-by: Akash Gupta <akagup@microsoft.com> Upstream-commit: 7a7357dae1bcccb17e9b2d4c7c8f5c025fce56ca Component: engine
132 lines
3.3 KiB
Go
132 lines
3.3 KiB
Go
package fscache
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/moby/buildkit/session/filesync"
|
|
"github.com/stretchr/testify/assert"
|
|
"golang.org/x/net/context"
|
|
)
|
|
|
|
func TestFSCache(t *testing.T) {
|
|
tmpDir, err := ioutil.TempDir("", "fscache")
|
|
assert.Nil(t, err)
|
|
defer os.RemoveAll(tmpDir)
|
|
|
|
backend := NewNaiveCacheBackend(filepath.Join(tmpDir, "backend"))
|
|
|
|
opt := Opt{
|
|
Root: tmpDir,
|
|
Backend: backend,
|
|
GCPolicy: GCPolicy{MaxSize: 15, MaxKeepDuration: time.Hour},
|
|
}
|
|
|
|
fscache, err := NewFSCache(opt)
|
|
assert.Nil(t, err)
|
|
|
|
defer fscache.Close()
|
|
|
|
err = fscache.RegisterTransport("test", &testTransport{})
|
|
assert.Nil(t, err)
|
|
|
|
src1, err := fscache.SyncFrom(context.TODO(), &testIdentifier{"foo", "data", "bar"})
|
|
assert.Nil(t, err)
|
|
|
|
dt, err := ioutil.ReadFile(filepath.Join(src1.Root().Path(), "foo"))
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, string(dt), "data")
|
|
|
|
// same id doesn't recalculate anything
|
|
src2, err := fscache.SyncFrom(context.TODO(), &testIdentifier{"foo", "data2", "bar"})
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, src1.Root().Path(), src2.Root().Path())
|
|
|
|
dt, err = ioutil.ReadFile(filepath.Join(src1.Root().Path(), "foo"))
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, string(dt), "data")
|
|
assert.Nil(t, src2.Close())
|
|
|
|
src3, err := fscache.SyncFrom(context.TODO(), &testIdentifier{"foo2", "data2", "bar"})
|
|
assert.Nil(t, err)
|
|
assert.NotEqual(t, src1.Root().Path(), src3.Root().Path())
|
|
|
|
dt, err = ioutil.ReadFile(filepath.Join(src3.Root().Path(), "foo2"))
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, string(dt), "data2")
|
|
|
|
s, err := fscache.DiskUsage()
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, s, int64(0))
|
|
|
|
assert.Nil(t, src3.Close())
|
|
|
|
s, err = fscache.DiskUsage()
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, s, int64(5))
|
|
|
|
// new upload with the same shared key shoutl overwrite
|
|
src4, err := fscache.SyncFrom(context.TODO(), &testIdentifier{"foo3", "data3", "bar"})
|
|
assert.Nil(t, err)
|
|
assert.NotEqual(t, src1.Root().Path(), src3.Root().Path())
|
|
|
|
dt, err = ioutil.ReadFile(filepath.Join(src3.Root().Path(), "foo3"))
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, string(dt), "data3")
|
|
assert.Equal(t, src4.Root().Path(), src3.Root().Path())
|
|
assert.Nil(t, src4.Close())
|
|
|
|
s, err = fscache.DiskUsage()
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, s, int64(10))
|
|
|
|
// this one goes over the GC limit
|
|
src5, err := fscache.SyncFrom(context.TODO(), &testIdentifier{"foo4", "datadata", "baz"})
|
|
assert.Nil(t, err)
|
|
assert.Nil(t, src5.Close())
|
|
|
|
// GC happens async
|
|
time.Sleep(100 * time.Millisecond)
|
|
|
|
// only last insertion after GC
|
|
s, err = fscache.DiskUsage()
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, s, int64(8))
|
|
|
|
// prune deletes everything
|
|
released, err := fscache.Prune(context.TODO())
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, released, uint64(8))
|
|
|
|
s, err = fscache.DiskUsage()
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, s, int64(0))
|
|
}
|
|
|
|
type testTransport struct {
|
|
}
|
|
|
|
func (t *testTransport) Copy(ctx context.Context, id RemoteIdentifier, dest string, cs filesync.CacheUpdater) error {
|
|
testid := id.(*testIdentifier)
|
|
return ioutil.WriteFile(filepath.Join(dest, testid.filename), []byte(testid.data), 0600)
|
|
}
|
|
|
|
type testIdentifier struct {
|
|
filename string
|
|
data string
|
|
sharedKey string
|
|
}
|
|
|
|
func (t *testIdentifier) Key() string {
|
|
return t.filename
|
|
}
|
|
func (t *testIdentifier) SharedKey() string {
|
|
return t.sharedKey
|
|
}
|
|
func (t *testIdentifier) Transport() string {
|
|
return "test"
|
|
}
|