moteva/server/internal/application/image_generator_target_test.go

package application

import (
	"testing"
	"time"

	"img_infinite_canvas/internal/domain/design"
)

func TestMergeGeneratedNodesIntoImageGeneratorTargetUsesGeneratedSizeAndKeepsCenter(t *testing.T) {
	nodes := []design.Node{
		{
			ID:        "generator-1",
			Type:      design.NodeTypeImage,
			Title:     "图像生成器",
			X:         100,
			Y:         80,
			Width:     360,
			Height:    240,
			Status:    "generating",
			LayerRole: "image-generator",
		},
	}
	generated := []design.Node{
		{
			ID:      "generated-1",
			Type:    design.NodeTypeImage,
			Title:   "商品主图",
			Content: "https://example.com/generated.png",
			Tone:    "visual",
			Status:  "success",
			Width:   1536,
			Height:  1024,
		},
	}

	merged := mergeGeneratedNodesIntoTarget(nodes, generated, "generator-1")
	if len(merged) != 1 {
		t.Fatalf("expected one merged node, got %d", len(merged))
	}
	node := merged[0]
	if node.LayerRole != "" || node.Status != "success" {
		t.Fatalf("expected completed image node, got role=%q status=%q", node.LayerRole, node.Status)
	}
	if node.Width != 1536 || node.Height != 1024 {
		t.Fatalf("expected generated size, got %.0fx%.0f", node.Width, node.Height)
	}
	if node.X+node.Width/2 != 280 || node.Y+node.Height/2 != 200 {
		t.Fatalf("expected replacement to keep placeholder center, got center %.0f,%.0f", node.X+node.Width/2, node.Y+node.Height/2)
	}
}

func TestImageGeneratorTargetNodeIDParsesDirectiveCaseInsensitively(t *testing.T) {
	targetID := imageGeneratorTargetNodeID([]design.AgentChatMessage{
		{
			Role: "user",
			Contents: []design.AgentContent{
				{Type: "text", Text: "cat\nimage generator target node: generator-1"},
			},
		},
	})
	if targetID != "generator-1" {
		t.Fatalf("expected target id, got %q", targetID)
	}
}

func TestRestoreFinishedGeneratedImageNodesRepairsStaleCanvasSnapshot(t *testing.T) {
	now := time.Now()
	project := design.Project{
		ID:        "project-stale-snapshot",
		Status:    design.StatusReady,
		UpdatedAt: now,
		Nodes: []design.Node{
			{
				ID:        "artifact-node",
				Type:      design.NodeTypeImage,
				Title:     "去背景中",
				Content:   "http://localhost:19000/canvas-assets/old.webp",
				Status:    "generating",
				LayerRole: "background-removed",
				Width:     512,
				Height:    512,
			},
			{
				ID:      "stale-agent-placeholder",
				Type:    design.NodeTypeImage,
				Title:   "猫主题视觉图",
				Content: "cat",
				Status:  "generating",
				Width:   1024,
				Height:  1024,
			},
		},
		Messages: []design.Message{
			toolArtifactsMessage("thread-image", "cat", []design.Node{
				{
					ID:      "artifact-node",
					Type:    design.NodeTypeImage,
					Title:   "猫主题视觉图",
					Content: "http://localhost:19000/canvas-assets/cat.png",
					Status:  "success",
					Width:   1024,
					Height:  1024,
				},
			}, now),
		},
	}

	restored, changed := (*DesignService)(nil).restoreFinishedGeneratedImageNodes(project, now.Add(time.Second))
	if !changed {
		t.Fatal("expected stale snapshot to be repaired")
	}
	if len(restored.Nodes) != 1 {
		t.Fatalf("expected stale prompt placeholder removed, got %#v", restored.Nodes)
	}
	node := restored.Nodes[0]
	if node.ID != "artifact-node" || node.Status != "success" || node.LayerRole != "" || node.Content != "http://localhost:19000/canvas-assets/cat.png" {
		t.Fatalf("expected artifact node restored, got %#v", node)
	}
}

func TestMergeSavedCanvasWorkStateDoesNotReapplyStaleGeneratingNodes(t *testing.T) {
	existing := []design.Node{
		{ID: "generated", Type: design.NodeTypeImage, Title: "Generated", Content: "http://localhost:19000/canvas-assets/generated.png", Status: "success"},
	}
	incoming := []design.Node{
		{ID: "generated", Type: design.NodeTypeImage, Title: "Generated", Content: "cat", Status: "generating"},
		{ID: "stale-placeholder", Type: design.NodeTypeImage, Title: "猫主题视觉图", Content: "cat", Status: "generating"},
	}

	merged := mergeSavedCanvasWorkState(existing, incoming)
	if len(merged) != 1 {
		t.Fatalf("expected stale placeholder dropped, got %#v", merged)
	}
	if merged[0].Status != "success" || merged[0].Content != existing[0].Content {
		t.Fatalf("expected server-completed node preserved, got %#v", merged[0])
	}
}