mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-21 06:10:23 -04:00
65 lines
1.9 KiB
Go
65 lines
1.9 KiB
Go
package container // import "github.com/docker/docker/integration/container"
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
containertypes "github.com/docker/docker/api/types/container"
|
|
"github.com/docker/docker/integration/internal/container"
|
|
"github.com/docker/docker/internal/test/request"
|
|
"gotest.tools/assert"
|
|
is "gotest.tools/assert/cmp"
|
|
"gotest.tools/poll"
|
|
)
|
|
|
|
func TestUpdateRestartPolicy(t *testing.T) {
|
|
defer setupTest(t)()
|
|
client := request.NewAPIClient(t)
|
|
ctx := context.Background()
|
|
|
|
cID := container.Run(t, ctx, client, container.WithCmd("sh", "-c", "sleep 1 && false"), func(c *container.TestContainerConfig) {
|
|
c.HostConfig.RestartPolicy = containertypes.RestartPolicy{
|
|
Name: "on-failure",
|
|
MaximumRetryCount: 3,
|
|
}
|
|
})
|
|
|
|
_, err := client.ContainerUpdate(ctx, cID, containertypes.UpdateConfig{
|
|
RestartPolicy: containertypes.RestartPolicy{
|
|
Name: "on-failure",
|
|
MaximumRetryCount: 5,
|
|
},
|
|
})
|
|
assert.NilError(t, err)
|
|
|
|
timeout := 60 * time.Second
|
|
if testEnv.OSType == "windows" {
|
|
timeout = 180 * time.Second
|
|
}
|
|
|
|
poll.WaitOn(t, container.IsInState(ctx, client, cID, "exited"), poll.WithDelay(100*time.Millisecond), poll.WithTimeout(timeout))
|
|
|
|
inspect, err := client.ContainerInspect(ctx, cID)
|
|
assert.NilError(t, err)
|
|
assert.Check(t, is.Equal(inspect.RestartCount, 5))
|
|
assert.Check(t, is.Equal(inspect.HostConfig.RestartPolicy.MaximumRetryCount, 5))
|
|
}
|
|
|
|
func TestUpdateRestartWithAutoRemove(t *testing.T) {
|
|
defer setupTest(t)()
|
|
client := request.NewAPIClient(t)
|
|
ctx := context.Background()
|
|
|
|
cID := container.Run(t, ctx, client, func(c *container.TestContainerConfig) {
|
|
c.HostConfig.AutoRemove = true
|
|
})
|
|
|
|
_, err := client.ContainerUpdate(ctx, cID, containertypes.UpdateConfig{
|
|
RestartPolicy: containertypes.RestartPolicy{
|
|
Name: "always",
|
|
},
|
|
})
|
|
assert.Check(t, is.ErrorContains(err, "Restart policy cannot be updated because AutoRemove is enabled for the container"))
|
|
}
|