point-source
91 lines
3.5 KiB
Go
91 lines
3.5 KiB
Go
package previewfeedback
|
|
|
|
import (
|
|
"encoding/json"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/kandev/kandev/internal/task/models"
|
|
)
|
|
|
|
func TestAppendMarkdownIncludesRuntimeTextLocationAndEscapesFences(t *testing.T) {
|
|
item := &models.TaskPreviewFeedback{
|
|
Kind: models.TaskPreviewFeedbackText, Comment: "Generated at runtime",
|
|
SourceKind: models.TaskPreviewFeedbackBrowser, SourceLabel: "Local app",
|
|
PageRoute: "/products?tab=all", PageTitle: "Products",
|
|
SelectedText: "Choose `this` plan",
|
|
TextAnchor: json.RawMessage(`{
|
|
"start":{"selector":"#offer","node_path":[0,1],"offset":2},
|
|
"end":{"selector":"#offer","node_path":[0,1],"offset":20},
|
|
"rects":[{"x":20,"y":30,"width":160,"height":20}],
|
|
"union_rect":{"x":20,"y":30,"width":160,"height":20},
|
|
"scroll_x":0,"scroll_y":480,"viewport_width":1280,"viewport_height":720,
|
|
"device_pixel_ratio":2,
|
|
"containing_element":{"tag":"button","classes":["offer"],"outer_html":"<button>` + "```generated```" + `</button>"}
|
|
}`),
|
|
}
|
|
|
|
got, err := AppendMarkdown("Please fix this.", []*models.TaskPreviewFeedback{item})
|
|
if err != nil {
|
|
t.Fatalf("AppendMarkdown: %v", err)
|
|
}
|
|
for _, want := range []string{
|
|
"Please fix this.", "### Web Preview Feedback", "Text feedback",
|
|
`Source: "Local app"`, `Page route: "/products?tab=all"`,
|
|
"Choose `this` plan", "containing_element", "node_path", "union_rect",
|
|
"scroll_y", "viewport_width", "device_pixel_ratio", "````json",
|
|
} {
|
|
if !strings.Contains(got, want) {
|
|
t.Fatalf("formatted prompt missing %q:\n%s", want, got)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestAppendMarkdownFormatsElementAndScreenshot(t *testing.T) {
|
|
items := []*models.TaskPreviewFeedback{
|
|
{
|
|
Kind: models.TaskPreviewFeedbackElement, Comment: "Align with price",
|
|
SourceKind: models.TaskPreviewFeedbackHTMLFile, SourceLabel: "checkout.html",
|
|
SourcePath: "mockups/checkout.html", PageRoute: "/checkout", PageTitle: "Checkout",
|
|
ElementSnapshot: json.RawMessage(`{"tag":"button","selector":"button.checkout","outer_html":"<button class=\"checkout\">Pay</button>"}`),
|
|
CaptureRect: json.RawMessage(`{"x":12,"y":24,"width":80,"height":32}`),
|
|
},
|
|
{
|
|
Kind: models.TaskPreviewFeedbackScreenshot, Comment: "The form clips here",
|
|
SourceKind: models.TaskPreviewFeedbackBrowser, SourceLabel: "Local app",
|
|
PageRoute: "/account", PageTitle: "Account",
|
|
CaptureRect: json.RawMessage(`{"x":10,"y":20,"width":820,"height":360}`),
|
|
ScreenshotAttachmentID: "shot-1",
|
|
ScreenshotAttachment: &models.TaskMessageAttachment{
|
|
ID: "shot-1", Name: "account-form.png", MimeType: "image/png", SizeBytes: 1234,
|
|
},
|
|
},
|
|
}
|
|
|
|
got, err := AppendMarkdown("", items)
|
|
if err != nil {
|
|
t.Fatalf("AppendMarkdown: %v", err)
|
|
}
|
|
for _, want := range []string{
|
|
"Element feedback", `<button class=\"checkout\">Pay</button>`,
|
|
"Rendered position", "Screenshot feedback", `Image: "account-form.png"`,
|
|
"820", "360", "The form clips here",
|
|
} {
|
|
if !strings.Contains(got, want) {
|
|
t.Fatalf("formatted prompt missing %q:\n%s", want, got)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestAppendMarkdownRejectsOversizedPrompt(t *testing.T) {
|
|
_, err := AppendMarkdown(strings.Repeat("x", MaxPromptBytes), []*models.TaskPreviewFeedback{{
|
|
Kind: models.TaskPreviewFeedbackElement, Comment: "comment",
|
|
SourceKind: models.TaskPreviewFeedbackBrowser, SourceLabel: "app",
|
|
PageRoute: "/", ElementSnapshot: json.RawMessage(`{"outer_html":"<main></main>"}`),
|
|
CaptureRect: json.RawMessage(`{"x":0,"y":0,"width":1,"height":1}`),
|
|
}})
|
|
if err == nil {
|
|
t.Fatal("AppendMarkdown accepted an oversized prompt")
|
|
}
|
|
}
|