mirror of
https://github.com/moby/moby.git
synced 2026-08-03 14:41:03 +00:00
20d594fb79
The stdcopy package is used to produce and read multiplexed streams for "attach" and "logs". It is used both by the API server (to produce), and the client (to read / de-multiplex). Move it to the api package, so that it can be included in the api module. Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
97 lines
2.6 KiB
Go
97 lines
2.6 KiB
Go
package logging
|
|
|
|
import (
|
|
"bytes"
|
|
"runtime"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
testContainer "github.com/docker/docker/integration/internal/container"
|
|
"github.com/docker/docker/testutil"
|
|
"github.com/docker/docker/testutil/daemon"
|
|
"github.com/moby/moby/api/stdcopy"
|
|
"github.com/moby/moby/api/types"
|
|
"github.com/moby/moby/api/types/container"
|
|
"gotest.tools/v3/assert"
|
|
"gotest.tools/v3/poll"
|
|
)
|
|
|
|
// TestReadPluginNoRead tests that reads are supported even if the plugin isn't capable.
|
|
func TestReadPluginNoRead(t *testing.T) {
|
|
if runtime.GOOS == "windows" {
|
|
t.Skip("no unix domain sockets on Windows")
|
|
}
|
|
t.Parallel()
|
|
|
|
ctx := testutil.StartSpan(baseContext, t)
|
|
|
|
d := daemon.New(t)
|
|
d.StartWithBusybox(ctx, t, "--iptables=false", "--ip6tables=false")
|
|
defer d.Stop(t)
|
|
|
|
client, err := d.NewClient()
|
|
assert.Assert(t, err)
|
|
createPlugin(ctx, t, client, "test", "discard", asLogDriver)
|
|
|
|
err = client.PluginEnable(ctx, "test", types.PluginEnableOptions{Timeout: 30})
|
|
assert.Check(t, err)
|
|
d.Stop(t)
|
|
|
|
cfg := &container.Config{
|
|
Image: "busybox",
|
|
Cmd: []string{"/bin/echo", "hello world"},
|
|
}
|
|
for desc, test := range map[string]struct {
|
|
dOpts []string
|
|
logsSupported bool
|
|
}{
|
|
"default": {logsSupported: true},
|
|
"disabled caching": {[]string{"--log-opt=cache-disabled=true"}, false},
|
|
"explicitly enabled caching": {[]string{"--log-opt=cache-disabled=false"}, true},
|
|
} {
|
|
t.Run(desc, func(t *testing.T) {
|
|
ctx := testutil.StartSpan(ctx, t)
|
|
d.Start(t, append([]string{"--iptables=false", "--ip6tables=false"}, test.dOpts...)...)
|
|
defer d.Stop(t)
|
|
c, err := client.ContainerCreate(ctx,
|
|
cfg,
|
|
&container.HostConfig{LogConfig: container.LogConfig{Type: "test"}},
|
|
nil,
|
|
nil,
|
|
"",
|
|
)
|
|
assert.Assert(t, err)
|
|
defer client.ContainerRemove(ctx, c.ID, container.RemoveOptions{Force: true})
|
|
|
|
err = client.ContainerStart(ctx, c.ID, container.StartOptions{})
|
|
assert.Assert(t, err)
|
|
|
|
poll.WaitOn(t, testContainer.IsStopped(ctx, client, c.ID))
|
|
logs, err := client.ContainerLogs(ctx, c.ID, container.LogsOptions{ShowStdout: true})
|
|
if !test.logsSupported {
|
|
assert.Assert(t, err != nil)
|
|
return
|
|
}
|
|
assert.Assert(t, err)
|
|
defer logs.Close()
|
|
|
|
buf := bytes.NewBuffer(nil)
|
|
|
|
errCh := make(chan error, 1)
|
|
go func() {
|
|
_, err := stdcopy.StdCopy(buf, buf, logs)
|
|
errCh <- err
|
|
}()
|
|
|
|
select {
|
|
case <-time.After(60 * time.Second):
|
|
t.Fatal("timeout waiting for IO to complete")
|
|
case err := <-errCh:
|
|
assert.Assert(t, err)
|
|
}
|
|
assert.Assert(t, strings.TrimSpace(buf.String()) == "hello world", buf.Bytes())
|
|
})
|
|
}
|
|
}
|