2022-03-12 23:28:04 +01:00
|
|
|
package utils
|
|
|
|
|
|
|
|
import (
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
)
|
|
|
|
|
2022-03-12 23:51:08 +01:00
|
|
|
func TestSplitTestAtChunkNoFooter(t *testing.T) {
|
2022-03-12 23:28:04 +01:00
|
|
|
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,
|
2022-03-12 23:51:08 +01:00
|
|
|
SplitTextAtChunk(text, 300, ""),
|
2022-03-12 23:28:04 +01:00
|
|
|
[]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,
|
2022-03-12 23:51:08 +01:00
|
|
|
SplitTextAtChunk(text, 500, ""),
|
2022-03-12 23:28:04 +01:00
|
|
|
[]string{text})
|
|
|
|
}
|
2022-03-12 23:51:08 +01:00
|
|
|
|
|
|
|
func TestSplitTestAtChunk(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, "FOOTER TEXT"),
|
|
|
|
[]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 velitFOOTER TEXT`,
|
|
|
|
`esse cillum
|
|
|
|
dolore eu fugiat nulla pariatur. Excepteur sint occaecat cupidatat non proident,
|
|
|
|
sunt in culpa qui officia deserunt mollit anim id est laborum.FOOTER TEXT`})
|
|
|
|
|
|
|
|
assert.Equal(
|
|
|
|
t,
|
|
|
|
SplitTextAtChunk(text, 500, "FOOTER TEXT"),
|
|
|
|
[]string{text + "FOOTER TEXT"})
|
|
|
|
}
|