test: fix flaky TestRunAttachTermination

Signed-off-by: Alano Terblanche <18033717+Benehiko@users.noreply.github.com>

Author: Benehiko

cli/command/container/run_test.go

@@ -147,7 +147,6 @@ func TestRunAttachTermination(t *testing.T) {
 		_ = p.Close()
 	}()
 
-	var conn net.Conn
 	killCh := make(chan struct{})
 	attachCh := make(chan struct{})
 	fakeCLI := test.NewFakeCli(&fakeClient{
@@ -156,23 +155,28 @@ func TestRunAttachTermination(t *testing.T) {
 				ID: "id",
 			}, nil
 		},
-		containerKillFunc: func(ctx context.Context, containerID, signal string) error {
-			killCh <- struct{}{}
+		containerKillFunc: func(ctx context.Context, containerID, sig string) error {
+			t.Logf("KillFunc called")
+			if sig == "TERM" {
+				close(killCh)
+			}
 			return nil
 		},
 		containerAttachFunc: func(ctx context.Context, containerID string, options container.AttachOptions) (types.HijackedResponse, error) {
+			t.Logf("AttachFunc called")
 			server, client := net.Pipe()
-			conn = server
 			t.Cleanup(func() {
 				_ = server.Close()
 			})
 			attachCh <- struct{}{}
 			return types.NewHijackedResponse(client, types.MediaTypeRawStream), nil
 		},
 		waitFunc: func(_ string) (<-chan container.WaitResponse, <-chan error) {
+			t.Logf("WaitFunc called")
 			responseChan := make(chan container.WaitResponse, 1)
 			errChan := make(chan error)
-
+			<-killCh
+			t.Logf("WaitFunc got closed")
 			responseChan <- container.WaitResponse{
 				StatusCode: 130,
 			}
@@ -201,9 +205,7 @@ func TestRunAttachTermination(t *testing.T) {
 	case <-attachCh:
 	}
 
-	assert.NilError(t, syscall.Kill(syscall.Getpid(), syscall.SIGINT))
-	// end stream from "container" so that we'll detach
-	conn.Close()
+	assert.NilError(t, syscall.Kill(syscall.Getpid(), syscall.SIGTERM))
 
 	select {
 	case <-killCh: