diff --git a/utils/text.go b/utils/text.go index 8711354..0ec83da 100644 --- a/utils/text.go +++ b/utils/text.go @@ -8,14 +8,18 @@ func SplitTextAtChunk(text string, size int) []string { chunks := []string{} var message string - for _, word := range words { + for i, word := range words { if len(message+" "+word) > size { chunks = append(chunks, message) message = word continue } - message += " " + word + if i == 0 { + message += word + } else { + message += " " + word + } } chunks = append(chunks, message) diff --git a/utils/text_test.go b/utils/text_test.go new file mode 100644 index 0000000..fc481c8 --- /dev/null +++ b/utils/text_test.go @@ -0,0 +1,34 @@ +package utils + +import ( + "testing" + + "github.com/stretchr/testify/assert" +) + +func TestSplitTestAtChuk(t *testing.T) { + + text := `Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do +eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim +veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo +consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum +dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, +sunt in culpa qui officia deserunt mollit anim id est laborum.` + + assert.Equal( + t, + SplitTextAtChunk(text, 300), + []string{ + `Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do +eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim +veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo +consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse`, + `cillum +dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident, +sunt in culpa qui officia deserunt mollit anim id est laborum.`}) + + assert.Equal( + t, + SplitTextAtChunk(text, 500), + []string{text}) +}