123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112 |
- package container
- import (
- "context"
- "fmt"
- "testing"
- "time"
- "github.com/docker/docker/api/types"
- "github.com/docker/docker/api/types/container"
- "github.com/docker/docker/integration-cli/daemon"
- )
- func TestDaemonRestartKillContainers(t *testing.T) {
- type testCase struct {
- desc string
- config *container.Config
- hostConfig *container.HostConfig
- xRunning bool
- xRunningLiveRestore bool
- }
- for _, c := range []testCase{
- {
- desc: "container without restart policy",
- config: &container.Config{Image: "busybox", Cmd: []string{"top"}},
- xRunningLiveRestore: true,
- },
- {
- desc: "container with restart=always",
- config: &container.Config{Image: "busybox", Cmd: []string{"top"}},
- hostConfig: &container.HostConfig{RestartPolicy: container.RestartPolicy{Name: "always"}},
- xRunning: true,
- xRunningLiveRestore: true,
- },
- } {
- for _, liveRestoreEnabled := range []bool{false, true} {
- for fnName, stopDaemon := range map[string]func(*testing.T, *daemon.Daemon){
- "kill-daemon": func(t *testing.T, d *daemon.Daemon) {
- if err := d.Kill(); err != nil {
- t.Fatal(err)
- }
- },
- "stop-daemon": func(t *testing.T, d *daemon.Daemon) {
- d.Stop(t)
- },
- } {
- t.Run(fmt.Sprintf("live-restore=%v/%s/%s", liveRestoreEnabled, c.desc, fnName), func(t *testing.T) {
- c := c
- liveRestoreEnabled := liveRestoreEnabled
- stopDaemon := stopDaemon
- t.Parallel()
- d := daemon.New(t, "", "dockerd", daemon.Config{})
- client, err := d.NewClient()
- if err != nil {
- t.Fatal(err)
- }
- args := []string{"--iptables=false"}
- if liveRestoreEnabled {
- args = append(args, "--live-restore")
- }
- d.StartWithBusybox(t, args...)
- defer d.Stop(t)
- ctx := context.Background()
- resp, err := client.ContainerCreate(ctx, c.config, c.hostConfig, nil, "")
- if err != nil {
- t.Fatal(err)
- }
- defer client.ContainerRemove(ctx, resp.ID, types.ContainerRemoveOptions{Force: true})
- if err := client.ContainerStart(ctx, resp.ID, types.ContainerStartOptions{}); err != nil {
- t.Fatal(err)
- }
- stopDaemon(t, d)
- d.Start(t, args...)
- expected := c.xRunning
- if liveRestoreEnabled {
- expected = c.xRunningLiveRestore
- }
- var running bool
- for i := 0; i < 30; i++ {
- inspect, err := client.ContainerInspect(ctx, resp.ID)
- if err != nil {
- t.Fatal(err)
- }
- running = inspect.State.Running
- if running == expected {
- break
- }
- time.Sleep(2 * time.Second)
- }
- if running != expected {
- t.Fatalf("got unexpected running state, expected %v, got: %v", expected, running)
- }
- // TODO(cpuguy83): test pause states... this seems to be rather undefined currently
- })
- }
- }
- }
- }
|