mirror of
https://wget.la/https://github.com/leookun/cursor-byok
synced 2026-08-20 13:07:00 +08:00
25 lines
602 B
Go
25 lines
602 B
Go
package prompt
|
|
|
|
import "testing"
|
|
|
|
func TestRenderPromptTemplateReplacesModelPlaceholders(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
got := RenderPromptTemplate(
|
|
"id={{FAKE_MODEL_ID}} name={{FAKE_MODEL_NAME}}",
|
|
"Test Model",
|
|
)
|
|
if want := "id=Test Model name=Test Model"; got != want {
|
|
t.Fatalf("RenderPromptTemplate() = %q, want %q", got, want)
|
|
}
|
|
}
|
|
|
|
func TestRenderPromptTemplateUsesFallbackModelName(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
got := RenderPromptTemplate("{{FAKE_MODEL_NAME}}", " ")
|
|
if want := "当前请求模型"; got != want {
|
|
t.Fatalf("RenderPromptTemplate() = %q, want %q", got, want)
|
|
}
|
|
}
|