Function TestTruncate

coderd/util/strings/strings_test.go:20–82

(t *testing.T)

}

func TestTruncate(t *testing.T) {
	t.Parallel()

	for _, tt := range []struct {
		s        string
		n        int
		expected string
		options  []strings.TruncateOption
	}{
		{"foo", 4, "foo", nil},
		{"foo", 3, "foo", nil},
		{"foo", 2, "fo", nil},
		{"foo", 1, "f", nil},
		{"foo", 0, "", nil},
		{"foo", -1, "", nil},
		{"foo bar", 7, "foo bar", []strings.TruncateOption{strings.TruncateWithEllipsis}},
		{"foo bar", 6, "foo b…", []strings.TruncateOption{strings.TruncateWithEllipsis}},
		{"foo bar", 5, "foo …", []strings.TruncateOption{strings.TruncateWithEllipsis}},
		{"foo bar", 4, "foo…", []strings.TruncateOption{strings.TruncateWithEllipsis}},
		{"foo bar", 3, "fo…", []strings.TruncateOption{strings.TruncateWithEllipsis}},
		{"foo bar", 2, "f…", []strings.TruncateOption{strings.TruncateWithEllipsis}},
		{"foo bar", 1, "…", []strings.TruncateOption{strings.TruncateWithEllipsis}},
		{"foo bar", 0, "", []strings.TruncateOption{strings.TruncateWithEllipsis}},
		{"foo bar", 7, "foo bar", []strings.TruncateOption{strings.TruncateWithFullWords}},
		{"foo bar", 6, "foo", []strings.TruncateOption{strings.TruncateWithFullWords}},
		{"foo bar", 5, "foo", []strings.TruncateOption{strings.TruncateWithFullWords}},
		{"foo bar", 4, "foo", []strings.TruncateOption{strings.TruncateWithFullWords}},
		{"foo bar", 3, "foo", []strings.TruncateOption{strings.TruncateWithFullWords}},
		{"foo bar", 2, "fo", []strings.TruncateOption{strings.TruncateWithFullWords}},
		{"foo bar", 1, "f", []strings.TruncateOption{strings.TruncateWithFullWords}},
		{"foo bar", 0, "", []strings.TruncateOption{strings.TruncateWithFullWords}},
		{"foo bar", 7, "foo bar", []strings.TruncateOption{strings.TruncateWithFullWords, strings.TruncateWithEllipsis}},
		{"foo bar", 6, "foo…", []strings.TruncateOption{strings.TruncateWithFullWords, strings.TruncateWithEllipsis}},
		{"foo bar", 5, "foo…", []strings.TruncateOption{strings.TruncateWithFullWords, strings.TruncateWithEllipsis}},
		{"foo bar", 4, "foo…", []strings.TruncateOption{strings.TruncateWithFullWords, strings.TruncateWithEllipsis}},
		{"foo bar", 3, "fo…", []strings.TruncateOption{strings.TruncateWithFullWords, strings.TruncateWithEllipsis}},
		{"foo bar", 2, "f…", []strings.TruncateOption{strings.TruncateWithFullWords, strings.TruncateWithEllipsis}},
		{"foo bar", 1, "…", []strings.TruncateOption{strings.TruncateWithFullWords, strings.TruncateWithEllipsis}},
		{"foo bar", 0, "", []strings.TruncateOption{strings.TruncateWithFullWords, strings.TruncateWithEllipsis}},
		{"This is a very long task prompt that should be truncated to 160 characters. Lorem ipsum dolor sit amet, consectetur adipiscing elit. Sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.", 160, "This is a very long task prompt that should be truncated to 160 characters. Lorem ipsum dolor sit amet, consectetur adipiscing elit. Sed do eiusmod tempor…", []strings.TruncateOption{strings.TruncateWithFullWords, strings.TruncateWithEllipsis}},
		// Multi-byte rune handling.
		{"日本語テスト", 3, "日本語", nil},
		{"日本語テスト", 4, "日本語テ", nil},
		{"日本語テスト", 6, "日本語テスト", nil},
		{"日本語テスト", 4, "日本語…", []strings.TruncateOption{strings.TruncateWithEllipsis}},
		{"🎉🎊🎈🎁", 2, "🎉🎊", nil},
		{"🎉🎊🎈🎁", 3, "🎉🎊…", []strings.TruncateOption{strings.TruncateWithEllipsis}},
		// Multi-byte with full-word truncation.
		{"hello 日本語", 7, "hello…", []strings.TruncateOption{strings.TruncateWithFullWords, strings.TruncateWithEllipsis}},
		{"hello 日本語", 8, "hello 日…", []strings.TruncateOption{strings.TruncateWithEllipsis}},
		{"日本語 テスト", 4, "日本語", []strings.TruncateOption{strings.TruncateWithFullWords}},
	} {
		tName := fmt.Sprintf("%s_%d", tt.s, tt.n)
		for _, opt := range tt.options {
			tName += fmt.Sprintf("_%v", opt)
		}
		t.Run(tName, func(t *testing.T) {
			t.Parallel()