package main
 
 import (

 	"bufio"

 	"os/exec"

 	"regexp"

 	"strings"
 	"time"
 

 	"github.com/docker/docker/pkg/integration/checker"

 	"github.com/go-check/check"
 )
 

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

 	testRequires(c, DaemonIsLinux)

 	out, _ := dockerCmd(c, "run", "-d", "busybox", "top")

 	id := strings.TrimSpace(out)

 	c.Assert(waitRun(id), checker.IsNil)

 
 	statsCmd := exec.Command(dockerBinary, "stats", "--no-stream", id)

 	type output struct {
 		out []byte
 		err error
 	}
 
 	ch := make(chan output)

 	go func() {

 		out, err := statsCmd.Output()
 		ch <- output{out, err}

 	}()
 
 	select {

 	case outerr := <-ch:

 		c.Assert(outerr.err, checker.IsNil, check.Commentf("Error running stats: %v", outerr.err))
 		c.Assert(string(outerr.out), checker.Contains, id) //running container wasn't present in output

 	case <-time.After(3 * time.Second):

 		statsCmd.Process.Kill()
 		c.Fatalf("stats did not return immediately when not streaming")
 	}

 }
 
 func (s *DockerSuite) TestStatsContainerNotFound(c *check.C) {
 	testRequires(c, DaemonIsLinux)

 	out, _, err := dockerCmdWithError("stats", "notfound")

 	c.Assert(err, checker.NotNil)

 	c.Assert(out, checker.Contains, "No such container: notfound", check.Commentf("Expected to fail on not found container stats, got %q instead", out))

 
 	out, _, err = dockerCmdWithError("stats", "--no-stream", "notfound")

 	c.Assert(err, checker.NotNil)

 	c.Assert(out, checker.Contains, "No such container: notfound", check.Commentf("Expected to fail on not found container stats with --no-stream, got %q instead", out))

 }

 
 func (s *DockerSuite) TestStatsAllRunningNoStream(c *check.C) {
 	testRequires(c, DaemonIsLinux)
 
 	out, _ := dockerCmd(c, "run", "-d", "busybox", "top")
 	id1 := strings.TrimSpace(out)[:12]
 	c.Assert(waitRun(id1), check.IsNil)
 	out, _ = dockerCmd(c, "run", "-d", "busybox", "top")
 	id2 := strings.TrimSpace(out)[:12]
 	c.Assert(waitRun(id2), check.IsNil)
 	out, _ = dockerCmd(c, "run", "-d", "busybox", "top")
 	id3 := strings.TrimSpace(out)[:12]
 	c.Assert(waitRun(id3), check.IsNil)
 	dockerCmd(c, "stop", id3)
 
 	out, _ = dockerCmd(c, "stats", "--no-stream")
 	if !strings.Contains(out, id1) || !strings.Contains(out, id2) {
 		c.Fatalf("Expected stats output to contain both %s and %s, got %s", id1, id2, out)
 	}
 	if strings.Contains(out, id3) {
 		c.Fatalf("Did not expect %s in stats, got %s", id3, out)
 	}
 }
 
 func (s *DockerSuite) TestStatsAllNoStream(c *check.C) {
 	testRequires(c, DaemonIsLinux)
 
 	out, _ := dockerCmd(c, "run", "-d", "busybox", "top")
 	id1 := strings.TrimSpace(out)[:12]
 	c.Assert(waitRun(id1), check.IsNil)
 	dockerCmd(c, "stop", id1)
 	out, _ = dockerCmd(c, "run", "-d", "busybox", "top")
 	id2 := strings.TrimSpace(out)[:12]
 	c.Assert(waitRun(id2), check.IsNil)
 
 	out, _ = dockerCmd(c, "stats", "--all", "--no-stream")
 	if !strings.Contains(out, id1) || !strings.Contains(out, id2) {
 		c.Fatalf("Expected stats output to contain both %s and %s, got %s", id1, id2, out)
 	}
 }
 
 func (s *DockerSuite) TestStatsAllNewContainersAdded(c *check.C) {
 	testRequires(c, DaemonIsLinux)
 
 	id := make(chan string)
 	addedChan := make(chan struct{})
 
 	dockerCmd(c, "run", "-d", "busybox", "top")
 	statsCmd := exec.Command(dockerBinary, "stats")
 	stdout, err := statsCmd.StdoutPipe()
 	c.Assert(err, check.IsNil)
 	c.Assert(statsCmd.Start(), check.IsNil)
 	defer statsCmd.Process.Kill()
 
 	go func() {
 		containerID := <-id
 		matchID := regexp.MustCompile(containerID)
 
 		scanner := bufio.NewScanner(stdout)
 		for scanner.Scan() {
 			switch {
 			case matchID.MatchString(scanner.Text()):
 				close(addedChan)
 			}
 		}
 	}()
 
 	out, _ := dockerCmd(c, "run", "-d", "busybox", "top")

 	c.Assert(waitRun(strings.TrimSpace(out)), check.IsNil)

 	id <- strings.TrimSpace(out)[:12]
 
 	select {
 	case <-time.After(5 * time.Second):
 		c.Fatal("failed to observe new container created added to stats")
 	case <-addedChan:
 		// ignore, done
 	}
 }