integration-cli/docker_api_logs_test.go
91bfed60
 package main
 
 import (
9e0ffae8
 	"bufio"
91bfed60
 	"bytes"
 	"fmt"
 	"net/http"
9e0ffae8
 	"strings"
 	"time"
dc944ea7
 
7aab43e8
 	"github.com/docker/docker/pkg/integration/checker"
dc944ea7
 	"github.com/go-check/check"
91bfed60
 )
 
dc944ea7
 func (s *DockerSuite) TestLogsApiWithStdout(c *check.C) {
f9a3558a
 	testRequires(c, DaemonIsLinux)
9e0ffae8
 	out, _ := dockerCmd(c, "run", "-d", "-t", "busybox", "/bin/sh", "-c", "while true; do echo hello; sleep 1; done")
 	id := strings.TrimSpace(out)
7aab43e8
 	c.Assert(waitRun(id), checker.IsNil)
91bfed60
 
9e0ffae8
 	type logOut struct {
bb1c576e
 		out string
 		res *http.Response
 		err error
9e0ffae8
 	}
 	chLog := make(chan logOut)
 
 	go func() {
bb1c576e
 		res, body, err := sockRequestRaw("GET", fmt.Sprintf("/containers/%s/logs?follow=1&stdout=1&timestamps=1", id), nil, "")
18faf6f9
 		if err != nil {
 			chLog <- logOut{"", nil, err}
 			return
 		}
 		defer body.Close()
 		out, err := bufio.NewReader(body).ReadString('\n')
 		if err != nil {
 			chLog <- logOut{"", nil, err}
 			return
 		}
bb1c576e
 		chLog <- logOut{strings.TrimSpace(out), res, err}
9e0ffae8
 	}()
 
 	select {
 	case l := <-chLog:
7aab43e8
 		c.Assert(l.err, checker.IsNil)
 		c.Assert(l.res.StatusCode, checker.Equals, http.StatusOK)
9e0ffae8
 		if !strings.HasSuffix(l.out, "hello") {
 			c.Fatalf("expected log output to container 'hello', but it does not")
 		}
 	case <-time.After(2 * time.Second):
 		c.Fatal("timeout waiting for logs to exit")
91bfed60
 	}
 }
 
dc944ea7
 func (s *DockerSuite) TestLogsApiNoStdoutNorStderr(c *check.C) {
f9a3558a
 	testRequires(c, DaemonIsLinux)
91bfed60
 	name := "logs_test"
5c295460
 	dockerCmd(c, "run", "-d", "-t", "--name", name, "busybox", "/bin/sh")
91bfed60
 
c7845e27
 	status, body, err := sockRequest("GET", fmt.Sprintf("/containers/%s/logs", name), nil)
7aab43e8
 	c.Assert(status, checker.Equals, http.StatusBadRequest)
 	c.Assert(err, checker.IsNil)
91bfed60
 
 	expected := "Bad parameters: you must choose at least one stream"
 	if !bytes.Contains(body, []byte(expected)) {
dc944ea7
 		c.Fatalf("Expected %s, got %s", expected, string(body[:]))
91bfed60
 	}
 }
589de356
 
 // Regression test for #12704
 func (s *DockerSuite) TestLogsApiFollowEmptyOutput(c *check.C) {
f9a3558a
 	testRequires(c, DaemonIsLinux)
589de356
 	name := "logs_test"
 	t0 := time.Now()
5c295460
 	dockerCmd(c, "run", "-d", "-t", "--name", name, "busybox", "sleep", "10")
589de356
 
 	_, body, err := sockRequestRaw("GET", fmt.Sprintf("/containers/%s/logs?follow=1&stdout=1&stderr=1&tail=all", name), bytes.NewBuffer(nil), "")
 	t1 := time.Now()
7aab43e8
 	c.Assert(err, checker.IsNil)
589de356
 	body.Close()
 	elapsed := t1.Sub(t0).Seconds()
 	if elapsed > 5.0 {
 		c.Fatalf("HTTP response was not immediate (elapsed %.1fs)", elapsed)
 	}
 }
1eecc1e7
 
 func (s *DockerSuite) TestLogsAPIContainerNotFound(c *check.C) {
 	name := "nonExistentContainer"
 	resp, _, err := sockRequestRaw("GET", fmt.Sprintf("/containers/%s/logs?follow=1&stdout=1&stderr=1&tail=all", name), bytes.NewBuffer(nil), "")
7aab43e8
 	c.Assert(err, checker.IsNil)
 	c.Assert(resp.StatusCode, checker.Equals, http.StatusNotFound)
1eecc1e7
 }