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) { |
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×tamps=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")
} |
76a40092 |
case <-time.After(20 * time.Second): |
9e0ffae8 |
c.Fatal("timeout waiting for logs to exit") |
91bfed60 |
}
}
|
dc944ea7 |
func (s *DockerSuite) TestLogsApiNoStdoutNorStderr(c *check.C) { |
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) {
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() |
00f65ae8 |
if elapsed > 20.0 { |
589de356 |
c.Fatalf("HTTP response was not immediate (elapsed %.1fs)", elapsed)
}
} |
1eecc1e7 |
|
00f65ae8 |
func (s *DockerSuite) TestLogsApiContainerNotFound(c *check.C) { |
1eecc1e7 |
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 |
} |