github.com/lacework-dev/go-moby@v20.10.12+incompatible/integration/container/wait_test.go (about)

     1  package container // import "github.com/docker/docker/integration/container"
     2  
     3  import (
     4  	"context"
     5  	"testing"
     6  	"time"
     7  
     8  	"github.com/docker/docker/integration/internal/container"
     9  	"github.com/docker/docker/testutil/request"
    10  	"gotest.tools/v3/assert"
    11  	is "gotest.tools/v3/assert/cmp"
    12  	"gotest.tools/v3/poll"
    13  	"gotest.tools/v3/skip"
    14  )
    15  
    16  func TestWaitNonBlocked(t *testing.T) {
    17  	defer setupTest(t)()
    18  	cli := request.NewAPIClient(t)
    19  
    20  	testCases := []struct {
    21  		doc          string
    22  		cmd          string
    23  		expectedCode int64
    24  	}{
    25  		{
    26  			doc:          "wait-nonblocking-exit-0",
    27  			cmd:          "exit 0",
    28  			expectedCode: 0,
    29  		},
    30  		{
    31  			doc:          "wait-nonblocking-exit-random",
    32  			cmd:          "exit 99",
    33  			expectedCode: 99,
    34  		},
    35  	}
    36  
    37  	for _, tc := range testCases {
    38  		tc := tc
    39  		t.Run(tc.doc, func(t *testing.T) {
    40  			t.Parallel()
    41  			ctx := context.Background()
    42  			containerID := container.Run(ctx, t, cli, container.WithCmd("sh", "-c", tc.cmd))
    43  			poll.WaitOn(t, container.IsInState(ctx, cli, containerID, "exited"), poll.WithTimeout(30*time.Second), poll.WithDelay(100*time.Millisecond))
    44  
    45  			waitResC, errC := cli.ContainerWait(ctx, containerID, "")
    46  			select {
    47  			case err := <-errC:
    48  				assert.NilError(t, err)
    49  			case waitRes := <-waitResC:
    50  				assert.Check(t, is.Equal(tc.expectedCode, waitRes.StatusCode))
    51  			}
    52  		})
    53  	}
    54  }
    55  
    56  func TestWaitBlocked(t *testing.T) {
    57  	// Windows busybox does not support trap in this way, not sleep with sub-second
    58  	// granularity. It will always exit 0x40010004.
    59  	skip.If(t, testEnv.DaemonInfo.OSType != "linux")
    60  	defer setupTest(t)()
    61  	cli := request.NewAPIClient(t)
    62  
    63  	testCases := []struct {
    64  		doc          string
    65  		cmd          string
    66  		expectedCode int64
    67  	}{
    68  		{
    69  			doc:          "test-wait-blocked-exit-zero",
    70  			cmd:          "trap 'exit 0' TERM; while true; do usleep 10; done",
    71  			expectedCode: 0,
    72  		},
    73  		{
    74  			doc:          "test-wait-blocked-exit-random",
    75  			cmd:          "trap 'exit 99' TERM; while true; do usleep 10; done",
    76  			expectedCode: 99,
    77  		},
    78  	}
    79  	for _, tc := range testCases {
    80  		tc := tc
    81  		t.Run(tc.doc, func(t *testing.T) {
    82  			t.Parallel()
    83  			ctx := context.Background()
    84  			containerID := container.Run(ctx, t, cli, container.WithCmd("sh", "-c", tc.cmd))
    85  			poll.WaitOn(t, container.IsInState(ctx, cli, containerID, "running"), poll.WithTimeout(30*time.Second), poll.WithDelay(100*time.Millisecond))
    86  
    87  			waitResC, errC := cli.ContainerWait(ctx, containerID, "")
    88  
    89  			err := cli.ContainerStop(ctx, containerID, nil)
    90  			assert.NilError(t, err)
    91  
    92  			select {
    93  			case err := <-errC:
    94  				assert.NilError(t, err)
    95  			case waitRes := <-waitResC:
    96  				assert.Check(t, is.Equal(tc.expectedCode, waitRes.StatusCode))
    97  			case <-time.After(2 * time.Second):
    98  				t.Fatal("timeout waiting for `docker wait`")
    99  			}
   100  		})
   101  	}
   102  }