Sebastiaan van Stijn
d3e45f8743
This package was originally internal, but was moved out when BuildKit used it for its integration tests. That's no longer the case, so we can make it internal again. Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
65 lines
1.6 KiB
Go
65 lines
1.6 KiB
Go
package logging
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/moby/locker"
|
|
plugintypes "github.com/moby/moby/api/types/plugin"
|
|
"github.com/moby/moby/v2/internal/testutil/fixtures/plugin"
|
|
"github.com/pkg/errors"
|
|
"gotest.tools/v3/assert"
|
|
)
|
|
|
|
var pluginBuildLock = locker.New()
|
|
|
|
func ensurePlugin(t *testing.T, name string) string {
|
|
pluginBuildLock.Lock(name)
|
|
defer pluginBuildLock.Unlock(name)
|
|
|
|
installPath := filepath.Join(os.Getenv("GOPATH"), "bin", name)
|
|
if _, err := os.Stat(installPath); err == nil {
|
|
return installPath
|
|
}
|
|
|
|
goBin, err := exec.LookPath("go")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cmd := exec.Command(goBin, "build", "-o", installPath, "./"+filepath.Join("cmd", name))
|
|
cmd.Env = append(os.Environ(), "CGO_ENABLED=0", "GO111MODULE=off")
|
|
if out, err := cmd.CombinedOutput(); err != nil {
|
|
t.Fatal(errors.Wrapf(err, "error building basic plugin bin: %s", string(out)))
|
|
}
|
|
|
|
return installPath
|
|
}
|
|
|
|
func withSockPath(name string) func(*plugin.Config) {
|
|
return func(cfg *plugin.Config) {
|
|
cfg.Interface.Socket = name
|
|
}
|
|
}
|
|
|
|
func createPlugin(ctx context.Context, t *testing.T, client plugin.CreateClient, alias, bin string, opts ...plugin.CreateOpt) {
|
|
t.Helper()
|
|
|
|
pluginBin := ensurePlugin(t, bin)
|
|
|
|
opts = append(opts, withSockPath("plugin.sock"))
|
|
opts = append(opts, plugin.WithBinary(pluginBin))
|
|
|
|
err := plugin.Create(ctx, client, alias, opts...)
|
|
assert.NilError(t, err)
|
|
}
|
|
|
|
func asLogDriver(cfg *plugin.Config) {
|
|
cfg.Interface.Types = []plugintypes.CapabilityID{
|
|
{Capability: "logdriver", Prefix: "docker", Version: "1.0"},
|
|
}
|
|
}
|