first commit

run/spool/artifact_queue_test.go

@@ -0,0 +1,75 @@
+package spool
+
+import (
+	"testing"
+
+	"browser.local/run/protocol"
+)
+
+func TestArtifactQueueRetainsPendingAndRemovesAcknowledgedChunk(t *testing.T) {
+	queue, err := NewArtifactQueue(t.TempDir())
+	if err != nil {
+		t.Fatalf("new artifact queue: %v", err)
+	}
+	first := validQueuedArtifactChunk(0)
+	second := validQueuedArtifactChunk(1)
+	if err := queue.Enqueue(first); err != nil {
+		t.Fatalf("enqueue first: %v", err)
+	}
+	if err := queue.Enqueue(second); err != nil {
+		t.Fatalf("enqueue second: %v", err)
+	}
+	pending, err := queue.Pending()
+	if err != nil {
+		t.Fatalf("pending before ack: %v", err)
+	}
+	if len(pending) != 2 {
+		t.Fatalf("expected two pending chunks, got %+v", pending)
+	}
+
+	if err := queue.Ack(protocol.ArtifactChunkUploadResponse{Accepted: true, TransferID: "transfer-1", ArtifactID: "artifact-1", ChunkIndex: 0}); err != nil {
+		t.Fatalf("ack first: %v", err)
+	}
+	pending, err = queue.Pending()
+	if err != nil {
+		t.Fatalf("pending after ack: %v", err)
+	}
+	if len(pending) != 1 || pending[0].ChunkIndex != 1 {
+		t.Fatalf("expected second chunk pending, got %+v", pending)
+	}
+}
+
+func TestArtifactQueueRetainsChunkWhenAckDoesNotMatch(t *testing.T) {
+	queue, err := NewArtifactQueue(t.TempDir())
+	if err != nil {
+		t.Fatalf("new artifact queue: %v", err)
+	}
+	if err := queue.Enqueue(validQueuedArtifactChunk(0)); err != nil {
+		t.Fatalf("enqueue: %v", err)
+	}
+	if err := queue.Ack(protocol.ArtifactChunkUploadResponse{Accepted: true, TransferID: "transfer-1", ArtifactID: "artifact-1", ChunkIndex: 1}); err != nil {
+		t.Fatalf("ack mismatch: %v", err)
+	}
+	pending, err := queue.Pending()
+	if err != nil {
+		t.Fatalf("pending: %v", err)
+	}
+	if len(pending) != 1 || pending[0].ChunkIndex != 0 {
+		t.Fatalf("expected original chunk pending, got %+v", pending)
+	}
+}
+
+func validQueuedArtifactChunk(index int) protocol.ArtifactChunkUploadRequest {
+	payload := []byte{byte(index), byte(index + 1)}
+	return protocol.ArtifactChunkUploadRequest{
+		RunEndpointID: "run-local",
+		SessionToken:  "session-token",
+		TransferID:    "transfer-1",
+		ArtifactID:    "artifact-1",
+		ChunkIndex:    index,
+		Offset:        int64(index * len(payload)),
+		SizeBytes:     len(payload),
+		Checksum:      "sha256:0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef",
+		Payload:       payload,
+	}
+}