Merge pull request #17 from mildwonkey/mildwonkey/upgrade-state-response

Continue UpgradeResourceState_Response implementation

Author: Paddy

tfprotov5/internal/fromproto/resource.go

@@ -43,9 +43,13 @@ func UpgradeResourceStateResponse(in *tfplugin5.UpgradeResourceState_Response) (
 	if err != nil {
 		return nil, err
 	}
-	return &tfprotov5.UpgradeResourceStateResponse{
+	resp := &tfprotov5.UpgradeResourceStateResponse{
 		Diagnostics: diags,
-	}, nil
+	}
+	if in.UpgradedState != nil {
+		resp.UpgradedState = DynamicValue(in.UpgradedState)
+	}
+	return resp, nil
 }
 
 func ReadResourceRequest(in *tfplugin5.ReadResource_Request) (*tfprotov5.ReadResourceRequest, error) {

tfprotov5/internal/toproto/resource.go

@@ -41,9 +41,13 @@ func UpgradeResourceState_Response(in *tfprotov5.UpgradeResourceStateResponse) (
 	if err != nil {
 		return nil, err
 	}
-	return &tfplugin5.UpgradeResourceState_Response{
+	resp := &tfplugin5.UpgradeResourceState_Response{
 		Diagnostics: diags,
-	}, nil
+	}
+	if in.UpgradedState != nil {
+		resp.UpgradedState = DynamicValue(in.UpgradedState)
+	}
+	return resp, nil
 }
 
 func ReadResource_Request(in *tfprotov5.ReadResourceRequest) (*tfplugin5.ReadResource_Request, error) {