github.com/jwhonce/docker@v0.6.7-0.20190327063223-da823cf3a5a3/integration-cli/docker_api_logs_test.go (about) 1 package main 2 3 import ( 4 "bufio" 5 "bytes" 6 "context" 7 "fmt" 8 "io" 9 "io/ioutil" 10 "net/http" 11 "strconv" 12 "strings" 13 "time" 14 15 "github.com/docker/docker/api/types" 16 "github.com/docker/docker/client" 17 "github.com/docker/docker/integration-cli/checker" 18 "github.com/docker/docker/internal/test/request" 19 "github.com/docker/docker/pkg/stdcopy" 20 "github.com/go-check/check" 21 ) 22 23 func (s *DockerSuite) TestLogsAPIWithStdout(c *check.C) { 24 out, _ := dockerCmd(c, "run", "-d", "-t", "busybox", "/bin/sh", "-c", "while true; do echo hello; sleep 1; done") 25 id := strings.TrimSpace(out) 26 c.Assert(waitRun(id), checker.IsNil) 27 28 type logOut struct { 29 out string 30 err error 31 } 32 33 chLog := make(chan logOut) 34 res, body, err := request.Get(fmt.Sprintf("/containers/%s/logs?follow=1&stdout=1×tamps=1", id)) 35 c.Assert(err, checker.IsNil) 36 c.Assert(res.StatusCode, checker.Equals, http.StatusOK) 37 38 go func() { 39 defer body.Close() 40 out, err := bufio.NewReader(body).ReadString('\n') 41 if err != nil { 42 chLog <- logOut{"", err} 43 return 44 } 45 chLog <- logOut{strings.TrimSpace(out), err} 46 }() 47 48 select { 49 case l := <-chLog: 50 c.Assert(l.err, checker.IsNil) 51 if !strings.HasSuffix(l.out, "hello") { 52 c.Fatalf("expected log output to container 'hello', but it does not") 53 } 54 case <-time.After(30 * time.Second): 55 c.Fatal("timeout waiting for logs to exit") 56 } 57 } 58 59 func (s *DockerSuite) TestLogsAPINoStdoutNorStderr(c *check.C) { 60 name := "logs_test" 61 dockerCmd(c, "run", "-d", "-t", "--name", name, "busybox", "/bin/sh") 62 cli, err := client.NewClientWithOpts(client.FromEnv) 63 c.Assert(err, checker.IsNil) 64 defer cli.Close() 65 66 _, err = cli.ContainerLogs(context.Background(), name, types.ContainerLogsOptions{}) 67 expected := "Bad parameters: you must choose at least one stream" 68 c.Assert(err.Error(), checker.Contains, expected) 69 } 70 71 // Regression test for #12704 72 func (s *DockerSuite) TestLogsAPIFollowEmptyOutput(c *check.C) { 73 name := "logs_test" 74 t0 := time.Now() 75 dockerCmd(c, "run", "-d", "-t", "--name", name, "busybox", "sleep", "10") 76 77 _, body, err := request.Get(fmt.Sprintf("/containers/%s/logs?follow=1&stdout=1&stderr=1&tail=all", name)) 78 t1 := time.Now() 79 c.Assert(err, checker.IsNil) 80 body.Close() 81 elapsed := t1.Sub(t0).Seconds() 82 if elapsed > 20.0 { 83 c.Fatalf("HTTP response was not immediate (elapsed %.1fs)", elapsed) 84 } 85 } 86 87 func (s *DockerSuite) TestLogsAPIContainerNotFound(c *check.C) { 88 name := "nonExistentContainer" 89 resp, _, err := request.Get(fmt.Sprintf("/containers/%s/logs?follow=1&stdout=1&stderr=1&tail=all", name)) 90 c.Assert(err, checker.IsNil) 91 c.Assert(resp.StatusCode, checker.Equals, http.StatusNotFound) 92 } 93 94 func (s *DockerSuite) TestLogsAPIUntilFutureFollow(c *check.C) { 95 testRequires(c, DaemonIsLinux) 96 name := "logsuntilfuturefollow" 97 dockerCmd(c, "run", "-d", "--name", name, "busybox", "/bin/sh", "-c", "while true; do date +%s; sleep 1; done") 98 c.Assert(waitRun(name), checker.IsNil) 99 100 untilSecs := 5 101 untilDur, err := time.ParseDuration(fmt.Sprintf("%ds", untilSecs)) 102 c.Assert(err, checker.IsNil) 103 until := daemonTime(c).Add(untilDur) 104 105 client, err := client.NewClientWithOpts(client.FromEnv) 106 if err != nil { 107 c.Fatal(err) 108 } 109 110 cfg := types.ContainerLogsOptions{Until: until.Format(time.RFC3339Nano), Follow: true, ShowStdout: true, Timestamps: true} 111 reader, err := client.ContainerLogs(context.Background(), name, cfg) 112 c.Assert(err, checker.IsNil) 113 114 type logOut struct { 115 out string 116 err error 117 } 118 119 chLog := make(chan logOut) 120 121 go func() { 122 bufReader := bufio.NewReader(reader) 123 defer reader.Close() 124 for i := 0; i < untilSecs; i++ { 125 out, _, err := bufReader.ReadLine() 126 if err != nil { 127 if err == io.EOF { 128 return 129 } 130 chLog <- logOut{"", err} 131 return 132 } 133 134 chLog <- logOut{strings.TrimSpace(string(out)), err} 135 } 136 }() 137 138 for i := 0; i < untilSecs; i++ { 139 select { 140 case l := <-chLog: 141 c.Assert(l.err, checker.IsNil) 142 i, err := strconv.ParseInt(strings.Split(l.out, " ")[1], 10, 64) 143 c.Assert(err, checker.IsNil) 144 c.Assert(time.Unix(i, 0).UnixNano(), checker.LessOrEqualThan, until.UnixNano()) 145 case <-time.After(20 * time.Second): 146 c.Fatal("timeout waiting for logs to exit") 147 } 148 } 149 } 150 151 func (s *DockerSuite) TestLogsAPIUntil(c *check.C) { 152 testRequires(c, MinimumAPIVersion("1.34")) 153 name := "logsuntil" 154 dockerCmd(c, "run", "--name", name, "busybox", "/bin/sh", "-c", "for i in $(seq 1 3); do echo log$i; sleep 1; done") 155 156 client, err := client.NewClientWithOpts(client.FromEnv) 157 if err != nil { 158 c.Fatal(err) 159 } 160 161 extractBody := func(c *check.C, cfg types.ContainerLogsOptions) []string { 162 reader, err := client.ContainerLogs(context.Background(), name, cfg) 163 c.Assert(err, checker.IsNil) 164 165 actualStdout := new(bytes.Buffer) 166 actualStderr := ioutil.Discard 167 _, err = stdcopy.StdCopy(actualStdout, actualStderr, reader) 168 c.Assert(err, checker.IsNil) 169 170 return strings.Split(actualStdout.String(), "\n") 171 } 172 173 // Get timestamp of second log line 174 allLogs := extractBody(c, types.ContainerLogsOptions{Timestamps: true, ShowStdout: true}) 175 c.Assert(len(allLogs), checker.GreaterOrEqualThan, 3) 176 177 t, err := time.Parse(time.RFC3339Nano, strings.Split(allLogs[1], " ")[0]) 178 c.Assert(err, checker.IsNil) 179 until := t.Format(time.RFC3339Nano) 180 181 // Get logs until the timestamp of second line, i.e. first two lines 182 logs := extractBody(c, types.ContainerLogsOptions{Timestamps: true, ShowStdout: true, Until: until}) 183 184 // Ensure log lines after cut-off are excluded 185 logsString := strings.Join(logs, "\n") 186 c.Assert(logsString, checker.Not(checker.Contains), "log3", check.Commentf("unexpected log message returned, until=%v", until)) 187 } 188 189 func (s *DockerSuite) TestLogsAPIUntilDefaultValue(c *check.C) { 190 name := "logsuntildefaultval" 191 dockerCmd(c, "run", "--name", name, "busybox", "/bin/sh", "-c", "for i in $(seq 1 3); do echo log$i; done") 192 193 client, err := client.NewClientWithOpts(client.FromEnv) 194 if err != nil { 195 c.Fatal(err) 196 } 197 198 extractBody := func(c *check.C, cfg types.ContainerLogsOptions) []string { 199 reader, err := client.ContainerLogs(context.Background(), name, cfg) 200 c.Assert(err, checker.IsNil) 201 202 actualStdout := new(bytes.Buffer) 203 actualStderr := ioutil.Discard 204 _, err = stdcopy.StdCopy(actualStdout, actualStderr, reader) 205 c.Assert(err, checker.IsNil) 206 207 return strings.Split(actualStdout.String(), "\n") 208 } 209 210 // Get timestamp of second log line 211 allLogs := extractBody(c, types.ContainerLogsOptions{Timestamps: true, ShowStdout: true}) 212 213 // Test with default value specified and parameter omitted 214 defaultLogs := extractBody(c, types.ContainerLogsOptions{Timestamps: true, ShowStdout: true, Until: "0"}) 215 c.Assert(defaultLogs, checker.DeepEquals, allLogs) 216 }