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