github.com/SophiaGitHub/hello@v1.7.1-rc3/integration-cli/docker_api_logs_test.go (about)

     1  package main
     2  
     3  import (
     4  	"bufio"
     5  	"bytes"
     6  	"fmt"
     7  	"net/http"
     8  	"os/exec"
     9  	"strings"
    10  	"time"
    11  
    12  	"github.com/go-check/check"
    13  )
    14  
    15  func (s *DockerSuite) TestLogsApiWithStdout(c *check.C) {
    16  	out, _ := dockerCmd(c, "run", "-d", "-t", "busybox", "/bin/sh", "-c", "while true; do echo hello; sleep 1; done")
    17  	id := strings.TrimSpace(out)
    18  	if err := waitRun(id); err != nil {
    19  		c.Fatal(err)
    20  	}
    21  
    22  	type logOut struct {
    23  		out string
    24  		res *http.Response
    25  		err error
    26  	}
    27  	chLog := make(chan logOut)
    28  
    29  	go func() {
    30  		res, body, err := sockRequestRaw("GET", fmt.Sprintf("/containers/%s/logs?follow=1&stdout=1&timestamps=1", id), nil, "")
    31  		out, _ := bufio.NewReader(body).ReadString('\n')
    32  		chLog <- logOut{strings.TrimSpace(out), res, err}
    33  	}()
    34  
    35  	select {
    36  	case l := <-chLog:
    37  		c.Assert(l.err, check.IsNil)
    38  		c.Assert(l.res.StatusCode, check.Equals, http.StatusOK)
    39  		if !strings.HasSuffix(l.out, "hello") {
    40  			c.Fatalf("expected log output to container 'hello', but it does not")
    41  		}
    42  	case <-time.After(2 * time.Second):
    43  		c.Fatal("timeout waiting for logs to exit")
    44  	}
    45  }
    46  
    47  func (s *DockerSuite) TestLogsApiNoStdoutNorStderr(c *check.C) {
    48  	name := "logs_test"
    49  	runCmd := exec.Command(dockerBinary, "run", "-d", "-t", "--name", name, "busybox", "/bin/sh")
    50  	if out, _, err := runCommandWithOutput(runCmd); err != nil {
    51  		c.Fatal(out, err)
    52  	}
    53  
    54  	status, body, err := sockRequest("GET", fmt.Sprintf("/containers/%s/logs", name), nil)
    55  	c.Assert(status, check.Equals, http.StatusBadRequest)
    56  	c.Assert(err, check.IsNil)
    57  
    58  	expected := "Bad parameters: you must choose at least one stream"
    59  	if !bytes.Contains(body, []byte(expected)) {
    60  		c.Fatalf("Expected %s, got %s", expected, string(body[:]))
    61  	}
    62  }
    63  
    64  // Regression test for #12704
    65  func (s *DockerSuite) TestLogsApiFollowEmptyOutput(c *check.C) {
    66  	name := "logs_test"
    67  	t0 := time.Now()
    68  	runCmd := exec.Command(dockerBinary, "run", "-d", "-t", "--name", name, "busybox", "sleep", "10")
    69  	if out, _, err := runCommandWithOutput(runCmd); err != nil {
    70  		c.Fatal(out, err)
    71  	}
    72  
    73  	_, body, err := sockRequestRaw("GET", fmt.Sprintf("/containers/%s/logs?follow=1&stdout=1&stderr=1&tail=all", name), bytes.NewBuffer(nil), "")
    74  	t1 := time.Now()
    75  	body.Close()
    76  	if err != nil {
    77  		c.Fatal(err)
    78  	}
    79  	elapsed := t1.Sub(t0).Seconds()
    80  	if elapsed > 5.0 {
    81  		c.Fatalf("HTTP response was not immediate (elapsed %.1fs)", elapsed)
    82  	}
    83  }