github.com/zhouyu0/docker-note@v0.0.0-20190722021225-b8d3825084db/client/image_create_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 "github.com/docker/docker/api/types" 13 ) 14 15 func TestImageCreateError(t *testing.T) { 16 client := &Client{ 17 client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")), 18 } 19 _, err := client.ImageCreate(context.Background(), "reference", types.ImageCreateOptions{}) 20 if err == nil || err.Error() != "Error response from daemon: Server error" { 21 t.Fatalf("expected a Server error, got %v", err) 22 } 23 } 24 25 func TestImageCreate(t *testing.T) { 26 expectedURL := "/images/create" 27 expectedImage := "test:5000/my_image" 28 expectedTag := "sha256:ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff" 29 expectedReference := fmt.Sprintf("%s@%s", expectedImage, expectedTag) 30 expectedRegistryAuth := "eyJodHRwczovL2luZGV4LmRvY2tlci5pby92MS8iOnsiYXV0aCI6ImRHOTBid289IiwiZW1haWwiOiJqb2huQGRvZS5jb20ifX0=" 31 client := &Client{ 32 client: newMockClient(func(r *http.Request) (*http.Response, error) { 33 if !strings.HasPrefix(r.URL.Path, expectedURL) { 34 return nil, fmt.Errorf("Expected URL '%s', got '%s'", expectedURL, r.URL) 35 } 36 registryAuth := r.Header.Get("X-Registry-Auth") 37 if registryAuth != expectedRegistryAuth { 38 return nil, fmt.Errorf("X-Registry-Auth header not properly set in the request. Expected '%s', got %s", expectedRegistryAuth, registryAuth) 39 } 40 41 query := r.URL.Query() 42 fromImage := query.Get("fromImage") 43 if fromImage != expectedImage { 44 return nil, fmt.Errorf("fromImage not set in URL query properly. Expected '%s', got %s", expectedImage, fromImage) 45 } 46 47 tag := query.Get("tag") 48 if tag != expectedTag { 49 return nil, fmt.Errorf("tag not set in URL query properly. Expected '%s', got %s", expectedTag, tag) 50 } 51 52 return &http.Response{ 53 StatusCode: http.StatusOK, 54 Body: ioutil.NopCloser(bytes.NewReader([]byte("body"))), 55 }, nil 56 }), 57 } 58 59 createResponse, err := client.ImageCreate(context.Background(), expectedReference, types.ImageCreateOptions{ 60 RegistryAuth: expectedRegistryAuth, 61 }) 62 if err != nil { 63 t.Fatal(err) 64 } 65 response, err := ioutil.ReadAll(createResponse) 66 if err != nil { 67 t.Fatal(err) 68 } 69 if err = createResponse.Close(); err != nil { 70 t.Fatal(err) 71 } 72 if string(response) != "body" { 73 t.Fatalf("expected Body to contain 'body' string, got %s", response) 74 } 75 }