Carlos Florêncio
123 lines
3.8 KiB
Go
123 lines
3.8 KiB
Go
package main
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestReplaceOrAppendSection_append(t *testing.T) {
|
|
body := "Existing PR description."
|
|
section := "### Preview\nURL: https://example.com"
|
|
got := replaceOrAppendSection(body, section)
|
|
|
|
if !strings.Contains(got, sectionStart) {
|
|
t.Error("expected sectionStart marker")
|
|
}
|
|
if !strings.Contains(got, sectionEnd) {
|
|
t.Error("expected sectionEnd marker")
|
|
}
|
|
if !strings.HasPrefix(got, body) {
|
|
t.Errorf("expected original body to be preserved at start; got: %q", got)
|
|
}
|
|
}
|
|
|
|
func TestReplaceOrAppendSection_replace(t *testing.T) {
|
|
old := "### Preview\nURL: https://old.example.com"
|
|
body := "Existing body.\n\n" + sectionStart + "\n" + old + "\n" + sectionEnd
|
|
|
|
newSection := "### Preview\nURL: https://new.example.com"
|
|
got := replaceOrAppendSection(body, newSection)
|
|
|
|
if strings.Contains(got, "old.example.com") {
|
|
t.Error("expected old section to be replaced")
|
|
}
|
|
if !strings.Contains(got, "new.example.com") {
|
|
t.Error("expected new section to be present")
|
|
}
|
|
// Only one start marker.
|
|
if strings.Count(got, sectionStart) != 1 {
|
|
t.Errorf("expected exactly one start marker, got %d", strings.Count(got, sectionStart))
|
|
}
|
|
}
|
|
|
|
func TestReplaceOrAppendSection_emptyBody(t *testing.T) {
|
|
section := "### Preview"
|
|
got := replaceOrAppendSection("", section)
|
|
if !strings.HasPrefix(got, sectionStart) {
|
|
t.Errorf("expected section at start of empty body; got: %q", got)
|
|
}
|
|
}
|
|
|
|
func TestStripSection(t *testing.T) {
|
|
section := "### Preview\nURL: https://example.com"
|
|
body := "My PR description.\n\n" + sectionStart + "\n" + section + "\n" + sectionEnd
|
|
got := stripSection(body)
|
|
|
|
if strings.Contains(got, sectionStart) || strings.Contains(got, sectionEnd) {
|
|
t.Error("expected section markers to be removed")
|
|
}
|
|
if !strings.Contains(got, "My PR description.") {
|
|
t.Error("expected original body to be preserved")
|
|
}
|
|
if strings.Contains(got, "https://example.com") {
|
|
t.Error("expected section content to be removed")
|
|
}
|
|
}
|
|
|
|
func TestStripSection_noMarker(t *testing.T) {
|
|
body := "No preview section here."
|
|
got := stripSection(body)
|
|
if got != body {
|
|
t.Errorf("expected no-op; got %q", got)
|
|
}
|
|
}
|
|
|
|
func TestStripSection_otherBotAppended(t *testing.T) {
|
|
// Simulates another bot having appended its own section after ours.
|
|
section := "### Preview"
|
|
otherBot := "<!-- other-bot-start -->\nOther bot content\n<!-- other-bot-end -->"
|
|
body := "PR body.\n\n" + sectionStart + "\n" + section + "\n" + sectionEnd + "\n\n" + otherBot
|
|
got := stripSection(body)
|
|
|
|
if strings.Contains(got, sectionStart) {
|
|
t.Error("expected our section to be removed")
|
|
}
|
|
if !strings.Contains(got, "other-bot-start") {
|
|
t.Error("expected other bot section to be preserved")
|
|
}
|
|
}
|
|
|
|
func TestReplaceSection_orphanStart(t *testing.T) {
|
|
// Orphaned start marker (no end) should not produce a duplicate start marker.
|
|
body := "PR body.\n\n" + sectionStart + "\ntruncated content with no end"
|
|
newSection := "### Preview\nURL: https://new.example.com"
|
|
got := replaceOrAppendSection(body, newSection)
|
|
|
|
if strings.Count(got, sectionStart) != 1 {
|
|
t.Errorf("expected exactly one start marker, got %d: %q", strings.Count(got, sectionStart), got)
|
|
}
|
|
if !strings.Contains(got, "new.example.com") {
|
|
t.Error("expected new section content to be present")
|
|
}
|
|
if strings.Contains(got, "truncated content") {
|
|
t.Error("expected orphaned content to be removed")
|
|
}
|
|
}
|
|
|
|
func TestBuildDeploySection(t *testing.T) {
|
|
tests := []struct {
|
|
sha string
|
|
wantSha string
|
|
}{
|
|
{"abc1234", "abc1234"}, // exactly 7: no truncation
|
|
{"abc1234deadbeef", "abc1234"}, // > 7: truncated to 7
|
|
{"abc", "abc"}, // < 7: kept as-is
|
|
}
|
|
for _, tc := range tests {
|
|
got := buildDeploySection("https://example.com", tc.sha)
|
|
if !strings.Contains(got, "`"+tc.wantSha+"`") {
|
|
t.Errorf("sha=%q: expected %q in output, got: %s", tc.sha, tc.wantSha, got)
|
|
}
|
|
}
|
|
}
|