feat(t211): return SKU-matched image search candidates

2026-07-27 18:35:59 +08:00
parent 667a64f6c2
commit edfee2c35e
26 changed files with 1838 additions and 153 deletions
@@ -0,0 +1,109 @@
+package usecase
+
+import (
+	"strings"
+	"testing"
+)
+
+func TestValidateCandidateCommandAcceptsEmptyMatchedBatch(t *testing.T) {
+	command := validAIExecutionCandidateCommand()
+	command.Candidates = nil
+	command.Recommendation = nil
+
+	if err := validateCandidateCommand(command); err != nil {
+		t.Fatalf("validate empty matched batch: %v", err)
+	}
+}
+
+func TestValidateCandidateCommandAcceptsMatchedColorAndSize(t *testing.T) {
+	command := validAIExecutionCandidateCommand()
+
+	if err := validateCandidateCommand(command); err != nil {
+		t.Fatalf("validate matched candidate: %v", err)
+	}
+}
+
+func TestValidateCandidateCommandRejectsUnknownHardConstraint(t *testing.T) {
+	command := validAIExecutionCandidateCommand()
+	command.Candidates[0].Evaluation.HardConstraints[1].Status = "UNKNOWN"
+
+	if err := validateCandidateCommand(command); err == nil {
+		t.Fatal("expected unknown hard constraint to be rejected")
+	}
+}
+
+func TestValidateCandidateCommandRejectsV2CandidateWithoutHardConstraints(t *testing.T) {
+	command := validAIExecutionCandidateCommand()
+	command.Candidates[0].Evaluation.HardConstraints = nil
+
+	if err := validateCandidateCommand(command); err == nil {
+		t.Fatal("expected missing v2 hard constraints to be rejected")
+	}
+}
+
+func TestValidateCandidateCommandRejectsWeakV2Candidate(t *testing.T) {
+	command := validAIExecutionCandidateCommand()
+	command.Candidates[0].Evaluation.Score = 0.74
+
+	if err := validateCandidateCommand(command); err == nil {
+		t.Fatal("expected weak v2 candidate to be rejected")
+	}
+}
+
+func TestValidateCandidateCommandRejectsV2RecommendationAfterFirstCandidate(t *testing.T) {
+	command := validAIExecutionCandidateCommand()
+	second := command.Candidates[0]
+	second.Ordinal = 2
+	second.Title = "拼多多图片候选 2"
+	command.Candidates = append(command.Candidates, second)
+	command.Recommendation.CandidateOrdinal = 2
+
+	if err := validateCandidateCommand(command); err == nil {
+		t.Fatal("expected v2 recommendation after first candidate to be rejected")
+	}
+}
+
+func validAIExecutionCandidateCommand() StoreExecutionCandidatesCommand {
+	return StoreExecutionCandidatesCommand{
+		TaskContentSHA256: strings.Repeat("a", 64),
+		ExecutionMode:     aiAssistedMode,
+		SearchQuery:       "PDD_IMAGE_SEARCH",
+		Provenance: &ExecutionProvenance{
+			ProviderID:    "openai-compatible",
+			Model:         "test-model",
+			PromptVersion: "candidate-evaluation-v2",
+			SchemaVersion: 2,
+		},
+		Candidates: []ExecutionCandidate{
+			{
+				Ordinal: 1,
+				Title:   "拼多多图片候选 1",
+				Evaluation: &CandidateEvaluation{
+					Decision:   "REVIEW",
+					Score:      0.9,
+					Matched:    []string{"图片相似"},
+					Confidence: 0.9,
+					HardConstraints: []CandidateHardConstraintEvaluation{
+						{
+							Kind:     "COLOR",
+							Expected: "BLACK",
+							Status:   "MATCH",
+							Evidence: "截图显示黑色",
+						},
+						{
+							Kind:     "SIZE",
+							Expected: "L",
+							Status:   "MATCH",
+							Evidence: "截图显示 L 码",
+						},
+					},
+				},
+			},
+		},
+		Recommendation: &CandidateRecommendation{
+			CandidateOrdinal: 1,
+			PolicyVersion:    "sku-hard-constraints-v1",
+			Reasons:          []string{"SKU 颜色和尺码硬约束均匹配"},
+		},
+	}
+}