74 lines
1.9 KiB
Go
74 lines
1.9 KiB
Go
|
package memory
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"errors"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"img_infinite_canvas/internal/domain/design"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestSaveCanvasPreservesMessages(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
repo := NewProjectRepository()
|
||
|
|
project := design.Project{
|
||
|
|
ID: "project-canvas-save",
|
||
|
|
UserID: design.DefaultUserID,
|
||
|
|
Title: "Canvas save",
|
||
|
|
Status: design.StatusReady,
|
||
|
|
Version: "v1",
|
||
|
|
UpdatedAt: time.Now(),
|
||
|
|
Messages: []design.Message{{ID: "message-1", Role: "assistant", Content: "keep me"}},
|
||
|
|
}
|
||
|
|
if err := repo.Save(ctx, project); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
canvasProject := design.Project{
|
||
|
|
ID: project.ID,
|
||
|
|
UserID: project.UserID,
|
||
|
|
Title: project.Title,
|
||
|
|
Status: design.StatusReady,
|
||
|
|
Version: "v2",
|
||
|
|
SnapshotID: "snapshot-2",
|
||
|
|
UpdatedAt: time.Now(),
|
||
|
|
Viewport: design.Viewport{K: 1},
|
||
|
|
Nodes: []design.Node{{ID: "node-1", Type: design.NodeTypeFrame}},
|
||
|
|
}
|
||
|
|
if err := repo.SaveCanvas(ctx, canvasProject, "v1"); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
saved, err := repo.Get(ctx, project.ID)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
if len(saved.Messages) != 1 || saved.Messages[0].ID != "message-1" {
|
||
|
|
t.Fatalf("expected existing messages to be preserved, got %#v", saved.Messages)
|
||
|
|
}
|
||
|
|
if len(saved.Nodes) != 1 || saved.Nodes[0].ID != "node-1" {
|
||
|
|
t.Fatalf("expected canvas nodes to update, got %#v", saved.Nodes)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestSaveCanvasRejectsStaleVersion(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
repo := NewProjectRepository()
|
||
|
|
project := design.Project{
|
||
|
|
ID: "project-stale-canvas-save",
|
||
|
|
UserID: design.DefaultUserID,
|
||
|
|
Title: "Canvas save",
|
||
|
|
Status: design.StatusReady,
|
||
|
|
Version: "v2",
|
||
|
|
}
|
||
|
|
if err := repo.Save(ctx, project); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
err := repo.SaveCanvas(ctx, design.Project{ID: project.ID, UserID: project.UserID, Version: "v3"}, "v1")
|
||
|
|
if !errors.Is(err, design.ErrConflict) {
|
||
|
|
t.Fatalf("expected version conflict, got %v", err)
|
||
|
|
}
|
||
|
|
}
|