package service

import "testing"

func TestSanitizeCustomSnippet(t *testing.T) {
	got, err := sanitizeCustomSnippet("  body{color:red}  ", "</style", MaxCustomCSSRunes)
	if err != nil || got != "body{color:red}" {
		t.Fatalf("trim got %q err=%v", got, err)
	}
	if _, err = sanitizeCustomSnippet("a</style>b", "</style", MaxCustomCSSRunes); err == nil {
		t.Fatal("css closer should fail")
	}
	if _, err = sanitizeCustomSnippet("a</SCRIPT>b", "</script", MaxCustomJSRunes); err == nil {
		t.Fatal("js closer should fail")
	}
	if _, err = sanitizeCustomSnippet("ok", "</script", MaxCustomJSRunes); err != nil {
		t.Fatalf("valid js: %v", err)
	}
}