123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133 |
- package client
- import (
- "bytes"
- "fmt"
- "io"
- "io/ioutil"
- "log"
- "net/http"
- "os"
- "strings"
- "testing"
- "time"
- "github.com/docker/docker/api/types"
- "golang.org/x/net/context"
- )
- func TestServiceLogsError(t *testing.T) {
- client := &Client{
- client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")),
- }
- _, err := client.ServiceLogs(context.Background(), "service_id", types.ContainerLogsOptions{})
- if err == nil || err.Error() != "Error response from daemon: Server error" {
- t.Fatalf("expected a Server Error, got %v", err)
- }
- _, err = client.ServiceLogs(context.Background(), "service_id", types.ContainerLogsOptions{
- Since: "2006-01-02TZ",
- })
- if err == nil || !strings.Contains(err.Error(), `parsing time "2006-01-02TZ"`) {
- t.Fatalf("expected a 'parsing time' error, got %v", err)
- }
- }
- func TestServiceLogs(t *testing.T) {
- expectedURL := "/services/service_id/logs"
- cases := []struct {
- options types.ContainerLogsOptions
- expectedQueryParams map[string]string
- }{
- {
- expectedQueryParams: map[string]string{
- "tail": "",
- },
- },
- {
- options: types.ContainerLogsOptions{
- Tail: "any",
- },
- expectedQueryParams: map[string]string{
- "tail": "any",
- },
- },
- {
- options: types.ContainerLogsOptions{
- ShowStdout: true,
- ShowStderr: true,
- Timestamps: true,
- Details: true,
- Follow: true,
- },
- expectedQueryParams: map[string]string{
- "tail": "",
- "stdout": "1",
- "stderr": "1",
- "timestamps": "1",
- "details": "1",
- "follow": "1",
- },
- },
- {
- options: types.ContainerLogsOptions{
- // An complete invalid date, timestamp or go duration will be
- // passed as is
- Since: "invalid but valid",
- },
- expectedQueryParams: map[string]string{
- "tail": "",
- "since": "invalid but valid",
- },
- },
- }
- for _, logCase := range cases {
- client := &Client{
- client: newMockClient(func(r *http.Request) (*http.Response, error) {
- if !strings.HasPrefix(r.URL.Path, expectedURL) {
- return nil, fmt.Errorf("Expected URL '%s', got '%s'", expectedURL, r.URL)
- }
- // Check query parameters
- query := r.URL.Query()
- for key, expected := range logCase.expectedQueryParams {
- actual := query.Get(key)
- if actual != expected {
- return nil, fmt.Errorf("%s not set in URL query properly. Expected '%s', got %s", key, expected, actual)
- }
- }
- return &http.Response{
- StatusCode: http.StatusOK,
- Body: ioutil.NopCloser(bytes.NewReader([]byte("response"))),
- }, nil
- }),
- }
- body, err := client.ServiceLogs(context.Background(), "service_id", logCase.options)
- if err != nil {
- t.Fatal(err)
- }
- defer body.Close()
- content, err := ioutil.ReadAll(body)
- if err != nil {
- t.Fatal(err)
- }
- if string(content) != "response" {
- t.Fatalf("expected response to contain 'response', got %s", string(content))
- }
- }
- }
- func ExampleClient_ServiceLogs_withTimeout() {
- ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
- defer cancel()
- client, _ := NewEnvClient()
- reader, err := client.ServiceLogs(ctx, "service_id", types.ContainerLogsOptions{})
- if err != nil {
- log.Fatal(err)
- }
- _, err = io.Copy(os.Stdout, reader)
- if err != nil && err != io.EOF {
- log.Fatal(err)
- }
- }
|