package client import ( "bytes" "encoding/base64" "encoding/json" "fmt" "io/ioutil" "net/http" "strings" "testing" "golang.org/x/net/context" "github.com/docker/docker/api/types" ) func TestContainerStatPathError(t *testing.T) { client := &Client{ client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")), } _, err := client.ContainerStatPath(context.Background(), "container_id", "path") if err == nil || err.Error() != "Error response from daemon: Server error" { t.Fatalf("expected a Server error, got %v", err) } } func TestContainerStatPathNoHeaderError(t *testing.T) { client := &Client{ client: newMockClient(func(req *http.Request) (*http.Response, error) { return &http.Response{ StatusCode: http.StatusOK, Body: ioutil.NopCloser(bytes.NewReader([]byte(""))), }, nil }), } _, err := client.ContainerStatPath(context.Background(), "container_id", "path/to/file") if err == nil { t.Fatalf("expected an error, got nothing") } } func TestContainerStatPath(t *testing.T) { expectedURL := "/containers/container_id/archive" expectedPath := "path/to/file" client := &Client{ client: newMockClient(func(req *http.Request) (*http.Response, error) { if !strings.HasPrefix(req.URL.Path, expectedURL) { return nil, fmt.Errorf("Expected URL '%s', got '%s'", expectedURL, req.URL) } if req.Method != "HEAD" { return nil, fmt.Errorf("expected HEAD method, got %s", req.Method) } query := req.URL.Query() path := query.Get("path") if path != expectedPath { return nil, fmt.Errorf("path not set in URL query properly") } content, err := json.Marshal(types.ContainerPathStat{ Name: "name", Mode: 0700, }) if err != nil { return nil, err } base64PathStat := base64.StdEncoding.EncodeToString(content) return &http.Response{ StatusCode: http.StatusOK, Body: ioutil.NopCloser(bytes.NewReader([]byte(""))), Header: http.Header{ "X-Docker-Container-Path-Stat": []string{base64PathStat}, }, }, nil }), } stat, err := client.ContainerStatPath(context.Background(), "container_id", expectedPath) if err != nil { t.Fatal(err) } if stat.Name != "name" { t.Fatalf("expected container path stat name to be 'name', got '%s'", stat.Name) } if stat.Mode != 0700 { t.Fatalf("expected container path stat mode to be 0700, got '%v'", stat.Mode) } } func TestCopyToContainerError(t *testing.T) { client := &Client{ client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")), } err := client.CopyToContainer(context.Background(), "container_id", "path/to/file", bytes.NewReader([]byte("")), types.CopyToContainerOptions{}) if err == nil || err.Error() != "Error response from daemon: Server error" { t.Fatalf("expected a Server error, got %v", err) } } func TestCopyToContainerNotStatusOKError(t *testing.T) { client := &Client{ client: newMockClient(errorMock(http.StatusNoContent, "No content")), } err := client.CopyToContainer(context.Background(), "container_id", "path/to/file", bytes.NewReader([]byte("")), types.CopyToContainerOptions{}) if err == nil || err.Error() != "unexpected status code from daemon: 204" { t.Fatalf("expected an unexpected status code error, got %v", err) } } func TestCopyToContainer(t *testing.T) { expectedURL := "/containers/container_id/archive" expectedPath := "path/to/file" client := &Client{ client: newMockClient(func(req *http.Request) (*http.Response, error) { if !strings.HasPrefix(req.URL.Path, expectedURL) { return nil, fmt.Errorf("Expected URL '%s', got '%s'", expectedURL, req.URL) } if req.Method != "PUT" { return nil, fmt.Errorf("expected PUT method, got %s", req.Method) } query := req.URL.Query() path := query.Get("path") if path != expectedPath { return nil, fmt.Errorf("path not set in URL query properly, expected '%s', got %s", expectedPath, path) } noOverwriteDirNonDir := query.Get("noOverwriteDirNonDir") if noOverwriteDirNonDir != "true" { return nil, fmt.Errorf("noOverwriteDirNonDir not set in URL query properly, expected true, got %s", noOverwriteDirNonDir) } content, err := ioutil.ReadAll(req.Body) if err != nil { return nil, err } if err := req.Body.Close(); err != nil { return nil, err } if string(content) != "content" { return nil, fmt.Errorf("expected content to be 'content', got %s", string(content)) } return &http.Response{ StatusCode: http.StatusOK, Body: ioutil.NopCloser(bytes.NewReader([]byte(""))), }, nil }), } err := client.CopyToContainer(context.Background(), "container_id", expectedPath, bytes.NewReader([]byte("content")), types.CopyToContainerOptions{ AllowOverwriteDirWithFile: false, }) if err != nil { t.Fatal(err) } } func TestCopyFromContainerError(t *testing.T) { client := &Client{ client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")), } _, _, err := client.CopyFromContainer(context.Background(), "container_id", "path/to/file") if err == nil || err.Error() != "Error response from daemon: Server error" { t.Fatalf("expected a Server error, got %v", err) } } func TestCopyFromContainerNotStatusOKError(t *testing.T) { client := &Client{ client: newMockClient(errorMock(http.StatusNoContent, "No content")), } _, _, err := client.CopyFromContainer(context.Background(), "container_id", "path/to/file") if err == nil || err.Error() != "unexpected status code from daemon: 204" { t.Fatalf("expected an unexpected status code error, got %v", err) } } func TestCopyFromContainerNoHeaderError(t *testing.T) { client := &Client{ client: newMockClient(func(req *http.Request) (*http.Response, error) { return &http.Response{ StatusCode: http.StatusOK, Body: ioutil.NopCloser(bytes.NewReader([]byte(""))), }, nil }), } _, _, err := client.CopyFromContainer(context.Background(), "container_id", "path/to/file") if err == nil { t.Fatalf("expected an error, got nothing") } } func TestCopyFromContainer(t *testing.T) { expectedURL := "/containers/container_id/archive" expectedPath := "path/to/file" client := &Client{ client: newMockClient(func(req *http.Request) (*http.Response, error) { if !strings.HasPrefix(req.URL.Path, expectedURL) { return nil, fmt.Errorf("Expected URL '%s', got '%s'", expectedURL, req.URL) } if req.Method != "GET" { return nil, fmt.Errorf("expected PUT method, got %s", req.Method) } query := req.URL.Query() path := query.Get("path") if path != expectedPath { return nil, fmt.Errorf("path not set in URL query properly, expected '%s', got %s", expectedPath, path) } headercontent, err := json.Marshal(types.ContainerPathStat{ Name: "name", Mode: 0700, }) if err != nil { return nil, err } base64PathStat := base64.StdEncoding.EncodeToString(headercontent) return &http.Response{ StatusCode: http.StatusOK, Body: ioutil.NopCloser(bytes.NewReader([]byte("content"))), Header: http.Header{ "X-Docker-Container-Path-Stat": []string{base64PathStat}, }, }, nil }), } r, stat, err := client.CopyFromContainer(context.Background(), "container_id", expectedPath) if err != nil { t.Fatal(err) } if stat.Name != "name" { t.Fatalf("expected container path stat name to be 'name', got '%s'", stat.Name) } if stat.Mode != 0700 { t.Fatalf("expected container path stat mode to be 0700, got '%v'", stat.Mode) } content, err := ioutil.ReadAll(r) if err != nil { t.Fatal(err) } if err := r.Close(); err != nil { t.Fatal(err) } if string(content) != "content" { t.Fatalf("expected content to be 'content', got %s", string(content)) } }