Fixed patchObject ignoring ContentType and ContentEncoding

fsouza · Dec 15, 2023 · be30dba · be30dba
1 parent 445e8cf
commit be30dba
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/fakestorage/object.go b/fakestorage/object.go
@@ -1010,9 +1010,11 @@ func (s *Server) patchObject(r *http.Request) jsonResponse {
 	}
 
 	var payload struct {
-		Metadata   map[string]string `json:"metadata"`
-		CustomTime string
-		Acl        []acls
+		ContentType     string
+		ContentEncoding string
+		Metadata        map[string]string `json:"metadata"`
+		CustomTime      string
+		Acl             []acls
 	}
 	err := json.NewDecoder(r.Body).Decode(&payload)
 	if err != nil {
@@ -1024,6 +1026,8 @@ func (s *Server) patchObject(r *http.Request) jsonResponse {
 
 	var attrsToUpdate backend.ObjectAttrs
 
+	attrsToUpdate.ContentType = payload.ContentType
+	attrsToUpdate.ContentEncoding = payload.ContentEncoding
 	attrsToUpdate.Metadata = payload.Metadata
 	attrsToUpdate.CustomTime = payload.CustomTime
 

diff --git a/internal/grpc/server.go b/internal/grpc/server.go
@@ -44,7 +44,7 @@ func (g *Server) GetBucket(ctx context.Context, req *pb.GetBucketRequest) (*pb.B
 		TimeCreated:           timestamppb.New(bucket.TimeCreated),
 	}
 	return grpc_bucket, nil
-	///return GetBucketFromBackend(g.backend, req.Bucket)
+	// /return GetBucketFromBackend(g.backend, req.Bucket)
 }
 
 func (g *Server) DeleteBucket(ctx context.Context, req *pb.DeleteBucketRequest) (*pb.Empty, error) {
@@ -148,7 +148,9 @@ func (g *Server) UpdateObject(ctx context.Context, req *pb.UpdateObjectRequest)
 
 func (g *Server) PatchObject(ctx context.Context, req *pb.PatchObjectRequest) (*pb.Empty, error) {
 	attrs := backend.ObjectAttrs{
-		Metadata: req.Metadata.Metadata,
+		Metadata:        req.Metadata.Metadata,
+		ContentType:     req.Metadata.ContentType,
+		ContentEncoding: req.Metadata.ContentEncoding,
 	}
 	_, err := g.backend.PatchObject(req.Bucket, req.Object, attrs)
 	return &pb.Empty{}, err