images: Support updating distribution and description. (digitalocean#413

)
dmartzol · Nov 13, 2020 · 45fcf9e · 45fcf9e
1 parent f70bf7f
commit 45fcf9e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/images.go b/images.go
@@ -52,7 +52,9 @@ type Image struct {
 
 // ImageUpdateRequest represents a request to update an image.
 type ImageUpdateRequest struct {
-	Name string `json:"name"`
+	Name         string `json:"name,omitempty"`
+	Distribution string `json:"distribution,omitempty"`
+	Description  string `json:"description,omitempty"`
 }
 
 // CustomImageCreateRequest represents a request to create a custom image.

diff --git a/images_test.go b/images_test.go
@@ -353,12 +353,16 @@ func TestImages_Update(t *testing.T) {
 	defer teardown()
 
 	updateRequest := &ImageUpdateRequest{
-		Name: "name",
+		Name:         "name",
+		Distribution: "Fedora",
+		Description:  "Just testing...",
 	}
 
 	mux.HandleFunc("/v2/images/12345", func(w http.ResponseWriter, r *http.Request) {
 		expected := map[string]interface{}{
-			"name": "name",
+			"name":         "name",
+			"distribution": "Fedora",
+			"description":  "Just testing...",
 		}
 
 		var v map[string]interface{}