package main
 
 import (
 	"os/exec"
 	"strings"

 	"time"

 
 	"github.com/go-check/check"

 )
 

 func (s *DockerSuite) TestRestartStoppedContainer(c *check.C) {

 	runCmd := exec.Command(dockerBinary, "run", "-d", "busybox", "echo", "foobar")
 	out, _, err := runCommandWithOutput(runCmd)

 	if err != nil {

 		c.Fatal(out, err)

 	}

 	cleanedContainerID := strings.TrimSpace(out)

 
 	runCmd = exec.Command(dockerBinary, "wait", cleanedContainerID)

 	if out, _, err = runCommandWithOutput(runCmd); err != nil {

 		c.Fatal(out, err)

 	}

 
 	runCmd = exec.Command(dockerBinary, "logs", cleanedContainerID)
 	out, _, err = runCommandWithOutput(runCmd)

 	if err != nil {

 		c.Fatal(out, err)

 	}

 
 	if out != "foobar\n" {

 		c.Errorf("container should've printed 'foobar'")

 	}
 
 	runCmd = exec.Command(dockerBinary, "restart", cleanedContainerID)

 	if out, _, err = runCommandWithOutput(runCmd); err != nil {

 		c.Fatal(out, err)

 	}

 
 	runCmd = exec.Command(dockerBinary, "logs", cleanedContainerID)
 	out, _, err = runCommandWithOutput(runCmd)

 	if err != nil {

 		c.Fatal(out, err)

 	}

 
 	if out != "foobar\nfoobar\n" {

 		c.Errorf("container should've printed 'foobar' twice")

 	}
 
 }
 

 func (s *DockerSuite) TestRestartRunningContainer(c *check.C) {

 	runCmd := exec.Command(dockerBinary, "run", "-d", "busybox", "sh", "-c", "echo foobar && sleep 30 && echo 'should not print this'")
 	out, _, err := runCommandWithOutput(runCmd)

 	if err != nil {

 		c.Fatal(out, err)

 	}

 	cleanedContainerID := strings.TrimSpace(out)

 	time.Sleep(1 * time.Second)
 

 	runCmd = exec.Command(dockerBinary, "logs", cleanedContainerID)
 	out, _, err = runCommandWithOutput(runCmd)

 	if err != nil {

 		c.Fatal(out, err)

 	}

 
 	if out != "foobar\n" {

 		c.Errorf("container should've printed 'foobar'")

 	}
 
 	runCmd = exec.Command(dockerBinary, "restart", "-t", "1", cleanedContainerID)

 	if out, _, err = runCommandWithOutput(runCmd); err != nil {

 		c.Fatal(out, err)

 	}

 
 	runCmd = exec.Command(dockerBinary, "logs", cleanedContainerID)
 	out, _, err = runCommandWithOutput(runCmd)

 	if err != nil {

 		c.Fatal(out, err)

 	}

 	time.Sleep(1 * time.Second)
 

 	if out != "foobar\nfoobar\n" {

 		c.Errorf("container should've printed 'foobar' twice")

 	}
 
 }
 
 // Test that restarting a container with a volume does not create a new volume on restart. Regression test for #819.

 func (s *DockerSuite) TestRestartWithVolumes(c *check.C) {

 	runCmd := exec.Command(dockerBinary, "run", "-d", "-v", "/test", "busybox", "top")
 	out, _, err := runCommandWithOutput(runCmd)

 	if err != nil {

 		c.Fatal(out, err)

 	}

 	cleanedContainerID := strings.TrimSpace(out)

 
 	runCmd = exec.Command(dockerBinary, "inspect", "--format", "{{ len .Volumes }}", cleanedContainerID)
 	out, _, err = runCommandWithOutput(runCmd)

 	if err != nil {

 		c.Fatal(out, err)

 	}

 
 	if out = strings.Trim(out, " \n\r"); out != "1" {

 		c.Errorf("expect 1 volume received %s", out)

 	}
 

 	volumes, err := inspectField(cleanedContainerID, ".Volumes")
 	c.Assert(err, check.IsNil)

 
 	runCmd = exec.Command(dockerBinary, "restart", cleanedContainerID)

 	if out, _, err = runCommandWithOutput(runCmd); err != nil {

 		c.Fatal(out, err)

 	}

 
 	runCmd = exec.Command(dockerBinary, "inspect", "--format", "{{ len .Volumes }}", cleanedContainerID)
 	out, _, err = runCommandWithOutput(runCmd)

 	if err != nil {

 		c.Fatal(out, err)

 	}

 
 	if out = strings.Trim(out, " \n\r"); out != "1" {

 		c.Errorf("expect 1 volume after restart received %s", out)

 	}
 

 	volumesAfterRestart, err := inspectField(cleanedContainerID, ".Volumes")
 	c.Assert(err, check.IsNil)

 
 	if volumes != volumesAfterRestart {

 		c.Errorf("expected volume path: %s Actual path: %s", volumes, volumesAfterRestart)

 	}
 
 }

 func (s *DockerSuite) TestRestartPolicyNO(c *check.C) {

 
 	cmd := exec.Command(dockerBinary, "run", "-d", "--restart=no", "busybox", "false")
 	out, _, err := runCommandWithOutput(cmd)
 	if err != nil {

 		c.Fatal(err, out)

 	}
 
 	id := strings.TrimSpace(string(out))
 	name, err := inspectField(id, "HostConfig.RestartPolicy.Name")

 	c.Assert(err, check.IsNil)

 	if name != "no" {

 		c.Fatalf("Container restart policy name is %s, expected %s", name, "no")

 	}
 
 }
 

 func (s *DockerSuite) TestRestartPolicyAlways(c *check.C) {

 
 	cmd := exec.Command(dockerBinary, "run", "-d", "--restart=always", "busybox", "false")
 	out, _, err := runCommandWithOutput(cmd)
 	if err != nil {

 		c.Fatal(err, out)

 	}
 
 	id := strings.TrimSpace(string(out))
 	name, err := inspectField(id, "HostConfig.RestartPolicy.Name")

 	c.Assert(err, check.IsNil)

 	if name != "always" {

 		c.Fatalf("Container restart policy name is %s, expected %s", name, "always")

 	}
 

 	MaximumRetryCount, err := inspectField(id, "HostConfig.RestartPolicy.MaximumRetryCount")

 	c.Assert(err, check.IsNil)

 
 	// MaximumRetryCount=0 if the restart policy is always
 	if MaximumRetryCount != "0" {

 		c.Fatalf("Container Maximum Retry Count is %s, expected %s", MaximumRetryCount, "0")

 	}
 

 }
 

 func (s *DockerSuite) TestRestartPolicyOnFailure(c *check.C) {

 
 	cmd := exec.Command(dockerBinary, "run", "-d", "--restart=on-failure:1", "busybox", "false")
 	out, _, err := runCommandWithOutput(cmd)
 	if err != nil {

 		c.Fatal(err, out)

 	}
 
 	id := strings.TrimSpace(string(out))
 	name, err := inspectField(id, "HostConfig.RestartPolicy.Name")

 	c.Assert(err, check.IsNil)

 	if name != "on-failure" {

 		c.Fatalf("Container restart policy name is %s, expected %s", name, "on-failure")

 	}
 
 }

 
 // a good container with --restart=on-failure:3
 // MaximumRetryCount!=0; RestartCount=0

 func (s *DockerSuite) TestContainerRestartwithGoodContainer(c *check.C) {

 	out, err := exec.Command(dockerBinary, "run", "-d", "--restart=on-failure:3", "busybox", "true").CombinedOutput()
 	if err != nil {

 		c.Fatal(string(out), err)

 	}
 	id := strings.TrimSpace(string(out))
 	if err := waitInspect(id, "{{ .State.Restarting }} {{ .State.Running }}", "false false", 5); err != nil {

 		c.Fatal(err)

 	}
 	count, err := inspectField(id, "RestartCount")

 	c.Assert(err, check.IsNil)

 	if count != "0" {

 		c.Fatalf("Container was restarted %s times, expected %d", count, 0)

 	}
 	MaximumRetryCount, err := inspectField(id, "HostConfig.RestartPolicy.MaximumRetryCount")

 	c.Assert(err, check.IsNil)

 	if MaximumRetryCount != "3" {

 		c.Fatalf("Container Maximum Retry Count is %s, expected %s", MaximumRetryCount, "3")

 	}
 
 }