Add canonical chunk plan source model

internal/framework/checkpoint/recorder_test.go

@@ -5,6 +5,7 @@ import (
 	"encoding/json"
 	"os"
 	"path/filepath"
+	"reflect"
 	"strings"
 	"testing"
 
@@ -33,6 +34,12 @@ func TestWorkspaceRecorderWritesSuccessfulCheckpointFiles(t *testing.T) {
 			Content:   []byte("chunk content"),
 			MediaType: "text/plain",
 			Units:     doc.Units,
+			Annotations: source.ChunkAnnotations{
+				"same": json.RawMessage(`{"range":1}`),
+			},
+			PlanAnnotations: source.ChunkAnnotations{
+				"same": json.RawMessage(`{"plan":2}`),
+			},
 		},
 	}
 
@@ -57,7 +64,9 @@ func TestWorkspaceRecorderWritesSuccessfulCheckpointFiles(t *testing.T) {
 	assertManifestStatus(t, filepath.Join(root, "chunk", "manifest.json"), coreworkspace.StatusSucceeded)
 	var chunkPayload struct {
 		Chunks []struct {
-			Content struct {
+			Annotations     source.ChunkAnnotations `json:"annotations"`
+			PlanAnnotations source.ChunkAnnotations `json:"plan_annotations"`
+			Content         struct {
 				ContentBase64 string `json:"content_base64"`
 				ContentDigest string `json:"content_digest"`
 			} `json:"content"`
@@ -77,6 +86,28 @@ func TestWorkspaceRecorderWritesSuccessfulCheckpointFiles(t *testing.T) {
 	if got, want := chunkPayload.Chunks[0].Content.ContentDigest, contentDigest([]byte("chunk content")); got != want {
 		t.Fatalf("content digest = %q, want %q", got, want)
 	}
+	if !jsonEqual(t, chunkPayload.Chunks[0].Annotations["same"], json.RawMessage(`{"range":1}`)) || !jsonEqual(t, chunkPayload.Chunks[0].PlanAnnotations["same"], json.RawMessage(`{"plan":2}`)) {
+		t.Fatalf("checkpoint annotations = %s / %s", chunkPayload.Chunks[0].Annotations["same"], chunkPayload.Chunks[0].PlanAnnotations["same"])
+	}
+	loaded, decision := (&WorkspaceLoader{root: root}).Chunk("generic", doc.Digest)
+	if !decision.Reused || len(loaded.Chunks) != 1 {
+		t.Fatalf("loaded chunk checkpoint = %#v, decision = %#v", loaded, decision)
+	}
+	if string(loaded.Chunks[0].Annotations["same"]) != `{"range":1}` || string(loaded.Chunks[0].PlanAnnotations["same"]) != `{"plan":2}` {
+		t.Fatalf("loaded canonical annotations = %s / %s", loaded.Chunks[0].Annotations["same"], loaded.Chunks[0].PlanAnnotations["same"])
+	}
+}
+
+func jsonEqual(t *testing.T, left, right []byte) bool {
+	t.Helper()
+	var leftValue, rightValue any
+	if err := json.Unmarshal(left, &leftValue); err != nil {
+		t.Fatalf("decode left JSON: %v", err)
+	}
+	if err := json.Unmarshal(right, &rightValue); err != nil {
+		t.Fatalf("decode right JSON: %v", err)
+	}
+	return reflect.DeepEqual(leftValue, rightValue)
 }
 
 func TestWorkspaceArtifactCheckpointsRoundTripCodecIdentityAndBytes(t *testing.T) {