mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-21 14:20:23 -04:00
130 lines
3.4 KiB
Go
130 lines
3.4 KiB
Go
package build
|
|
|
|
import (
|
|
"context"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
|
|
dclient "github.com/docker/docker/client"
|
|
"github.com/docker/docker/internal/test/fakecontext"
|
|
"github.com/docker/docker/internal/test/request"
|
|
"github.com/moby/buildkit/session"
|
|
"github.com/moby/buildkit/session/filesync"
|
|
"golang.org/x/sync/errgroup"
|
|
"gotest.tools/assert"
|
|
is "gotest.tools/assert/cmp"
|
|
"gotest.tools/skip"
|
|
)
|
|
|
|
func TestBuildWithSession(t *testing.T) {
|
|
skip.If(t, !testEnv.DaemonInfo.ExperimentalBuild)
|
|
|
|
client := testEnv.APIClient()
|
|
|
|
dockerfile := `
|
|
FROM busybox
|
|
COPY file /
|
|
RUN cat /file
|
|
`
|
|
|
|
fctx := fakecontext.New(t, "",
|
|
fakecontext.WithFile("file", "some content"),
|
|
)
|
|
defer fctx.Close()
|
|
|
|
out := testBuildWithSession(t, client, client.DaemonHost(), fctx.Dir, dockerfile)
|
|
assert.Check(t, is.Contains(out, "some content"))
|
|
|
|
fctx.Add("second", "contentcontent")
|
|
|
|
dockerfile += `
|
|
COPY second /
|
|
RUN cat /second
|
|
`
|
|
|
|
out = testBuildWithSession(t, client, client.DaemonHost(), fctx.Dir, dockerfile)
|
|
assert.Check(t, is.Equal(strings.Count(out, "Using cache"), 2))
|
|
assert.Check(t, is.Contains(out, "contentcontent"))
|
|
|
|
du, err := client.DiskUsage(context.TODO())
|
|
assert.Check(t, err)
|
|
assert.Check(t, du.BuilderSize > 10)
|
|
|
|
out = testBuildWithSession(t, client, client.DaemonHost(), fctx.Dir, dockerfile)
|
|
assert.Check(t, is.Equal(strings.Count(out, "Using cache"), 4))
|
|
|
|
du2, err := client.DiskUsage(context.TODO())
|
|
assert.Check(t, err)
|
|
assert.Check(t, is.Equal(du.BuilderSize, du2.BuilderSize))
|
|
|
|
// rebuild with regular tar, confirm cache still applies
|
|
fctx.Add("Dockerfile", dockerfile)
|
|
// FIXME(vdemeester) use sock here
|
|
res, body, err := request.Do(
|
|
"/build",
|
|
request.Host(client.DaemonHost()),
|
|
request.Method(http.MethodPost),
|
|
request.RawContent(fctx.AsTarReader(t)),
|
|
request.ContentType("application/x-tar"))
|
|
assert.NilError(t, err)
|
|
assert.Check(t, is.DeepEqual(http.StatusOK, res.StatusCode))
|
|
|
|
outBytes, err := request.ReadBody(body)
|
|
assert.NilError(t, err)
|
|
assert.Check(t, is.Contains(string(outBytes), "Successfully built"))
|
|
assert.Check(t, is.Equal(strings.Count(string(outBytes), "Using cache"), 4))
|
|
|
|
_, err = client.BuildCachePrune(context.TODO())
|
|
assert.Check(t, err)
|
|
|
|
du, err = client.DiskUsage(context.TODO())
|
|
assert.Check(t, err)
|
|
assert.Check(t, is.Equal(du.BuilderSize, int64(0)))
|
|
}
|
|
|
|
func testBuildWithSession(t *testing.T, client dclient.APIClient, daemonHost string, dir, dockerfile string) (outStr string) {
|
|
ctx := context.Background()
|
|
sess, err := session.NewSession(ctx, "foo1", "foo")
|
|
assert.Check(t, err)
|
|
|
|
fsProvider := filesync.NewFSSyncProvider([]filesync.SyncedDir{
|
|
{Dir: dir},
|
|
})
|
|
sess.Allow(fsProvider)
|
|
|
|
g, ctx := errgroup.WithContext(ctx)
|
|
|
|
g.Go(func() error {
|
|
return sess.Run(ctx, client.DialSession)
|
|
})
|
|
|
|
g.Go(func() error {
|
|
// FIXME use sock here
|
|
res, body, err := request.Do(
|
|
"/build?remote=client-session&session="+sess.ID(),
|
|
request.Host(daemonHost),
|
|
request.Method(http.MethodPost),
|
|
request.With(func(req *http.Request) error {
|
|
req.Body = ioutil.NopCloser(strings.NewReader(dockerfile))
|
|
return nil
|
|
}),
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
assert.Check(t, is.DeepEqual(res.StatusCode, http.StatusOK))
|
|
out, err := request.ReadBody(body)
|
|
assert.NilError(t, err)
|
|
assert.Check(t, is.Contains(string(out), "Successfully built"))
|
|
sess.Close()
|
|
outStr = string(out)
|
|
return nil
|
|
})
|
|
|
|
err = g.Wait()
|
|
assert.Check(t, err)
|
|
return
|
|
}
|