1
0
mirror of https://github.com/docker/cli.git synced 2026-01-06 05:41:44 +03:00

fix: flaky prompt termination on reader close test

Signed-off-by: Alano Terblanche <18033717+Benehiko@users.noreply.github.com>
This commit is contained in:
Alano Terblanche
2024-03-21 10:53:30 +01:00
parent 2ae903e86c
commit d2ea5adfe4

View File

@@ -197,13 +197,11 @@ func TestPromptForConfirmation(t *testing.T) {
promptReader, promptWriter = io.Pipe()
wroteHook := make(chan struct{}, 1)
defer close(wroteHook)
promptOut := test.NewWriterWithHook(bufioWriter, func(p []byte) {
wroteHook <- struct{}{}
})
result := make(chan promptResult, 1)
defer close(result)
go func() {
r, err := command.PromptForConfirmation(ctx, promptReader, promptOut, "")
result <- promptResult{r, err}
@@ -216,7 +214,7 @@ func TestPromptForConfirmation(t *testing.T) {
assert.NilError(t, bufioWriter.Flush())
assert.Equal(t, strings.TrimSpace(buf.String()), "Are you sure you want to proceed? [y/N]")
resultCtx, resultCancel := context.WithTimeout(ctx, 100*time.Millisecond)
resultCtx, resultCancel := context.WithTimeout(ctx, 500*time.Millisecond)
defer resultCancel()
tc.f(t, resultCtx, result)