github.com/zhouyu0/docker-note@v0.0.0-20190722021225-b8d3825084db/client/container_export_test.go (about) 1 package client // import "github.com/docker/docker/client" 2 3 import ( 4 "bytes" 5 "context" 6 "fmt" 7 "io/ioutil" 8 "net/http" 9 "strings" 10 "testing" 11 ) 12 13 func TestContainerExportError(t *testing.T) { 14 client := &Client{ 15 client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")), 16 } 17 _, err := client.ContainerExport(context.Background(), "nothing") 18 if err == nil || err.Error() != "Error response from daemon: Server error" { 19 t.Fatalf("expected a Server Error, got %v", err) 20 } 21 } 22 23 func TestContainerExport(t *testing.T) { 24 expectedURL := "/containers/container_id/export" 25 client := &Client{ 26 client: newMockClient(func(r *http.Request) (*http.Response, error) { 27 if !strings.HasPrefix(r.URL.Path, expectedURL) { 28 return nil, fmt.Errorf("Expected URL '%s', got '%s'", expectedURL, r.URL) 29 } 30 31 return &http.Response{ 32 StatusCode: http.StatusOK, 33 Body: ioutil.NopCloser(bytes.NewReader([]byte("response"))), 34 }, nil 35 }), 36 } 37 body, err := client.ContainerExport(context.Background(), "container_id") 38 if err != nil { 39 t.Fatal(err) 40 } 41 defer body.Close() 42 content, err := ioutil.ReadAll(body) 43 if err != nil { 44 t.Fatal(err) 45 } 46 if string(content) != "response" { 47 t.Fatalf("expected response to contain 'response', got %s", string(content)) 48 } 49 }