Merge pull request #5296 from vvoland/5295-27.0

[27.1 backport] attach: don't return context cancelled error
This commit is contained in:
Sebastiaan van Stijn 2024-07-25 09:59:38 +02:00 committed by GitHub
commit fd3157bf35
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 39 additions and 1 deletions

View File

@ -146,7 +146,8 @@ func RunAttach(ctx context.Context, dockerCLI command.Cli, containerID string, o
detachKeys: options.DetachKeys, detachKeys: options.DetachKeys,
} }
if err := streamer.stream(ctx); err != nil { // if the context was canceled, this was likely intentional and we shouldn't return an error
if err := streamer.stream(ctx); err != nil && !errors.Is(err, context.Canceled) {
return err return err
} }
@ -163,6 +164,9 @@ func getExitStatus(errC <-chan error, resultC <-chan container.WaitResponse) err
return cli.StatusError{StatusCode: int(result.StatusCode)} return cli.StatusError{StatusCode: int(result.StatusCode)}
} }
case err := <-errC: case err := <-errC:
if errors.Is(err, context.Canceled) {
return nil
}
return err return err
} }

View File

@ -1,6 +1,7 @@
package container package container
import ( import (
"context"
"io" "io"
"testing" "testing"
@ -113,6 +114,10 @@ func TestGetExitStatus(t *testing.T) {
}, },
expectedError: cli.StatusError{StatusCode: 15}, expectedError: cli.StatusError{StatusCode: 15},
}, },
{
err: context.Canceled,
expectedError: nil,
},
} }
for _, testcase := range testcases { for _, testcase := range testcases {

View File

@ -1,11 +1,17 @@
package container package container
import ( import (
"bytes"
"fmt" "fmt"
"os"
"os/exec"
"strings" "strings"
"testing" "testing"
"time"
"github.com/creack/pty"
"github.com/docker/cli/e2e/internal/fixtures" "github.com/docker/cli/e2e/internal/fixtures"
"gotest.tools/v3/assert"
"gotest.tools/v3/icmd" "gotest.tools/v3/icmd"
) )
@ -23,3 +29,26 @@ func TestAttachExitCode(t *testing.T) {
func withStdinNewline(cmd *icmd.Cmd) { func withStdinNewline(cmd *icmd.Cmd) {
cmd.Stdin = strings.NewReader("\n") cmd.Stdin = strings.NewReader("\n")
} }
// Regression test for https://github.com/docker/cli/issues/5294
func TestAttachInterrupt(t *testing.T) {
result := icmd.RunCommand("docker", "run", "-d", fixtures.AlpineImage, "sh", "-c", "sleep 5")
result.Assert(t, icmd.Success)
containerID := strings.TrimSpace(result.Stdout())
// run it as such so we can signal it later
c := exec.Command("docker", "attach", containerID)
d := bytes.Buffer{}
c.Stdout = &d
c.Stderr = &d
_, err := pty.Start(c)
assert.NilError(t, err)
// have to wait a bit to give time for the command to execute/print
time.Sleep(500 * time.Millisecond)
c.Process.Signal(os.Interrupt)
_ = c.Wait()
assert.Equal(t, c.ProcessState.ExitCode(), 0)
assert.Equal(t, d.String(), "")
}