|
|
|
@ -6,8 +6,7 @@ import (
|
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
func TestSplitTestAtChuk(t *testing.T) {
|
|
|
|
|
|
|
|
|
|
func TestSplitTestAtChunkNoFooter(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
|
|
|
|
@ -17,7 +16,7 @@ sunt in culpa qui officia deserunt mollit anim id est laborum.`
|
|
|
|
|
|
|
|
|
|
assert.Equal(
|
|
|
|
|
t,
|
|
|
|
|
SplitTextAtChunk(text, 300),
|
|
|
|
|
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
|
|
|
|
@ -29,6 +28,32 @@ sunt in culpa qui officia deserunt mollit anim id est laborum.`})
|
|
|
|
|
|
|
|
|
|
assert.Equal(
|
|
|
|
|
t,
|
|
|
|
|
SplitTextAtChunk(text, 500),
|
|
|
|
|
SplitTextAtChunk(text, 500, ""),
|
|
|
|
|
[]string{text})
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
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"})
|
|
|
|
|
}
|
|
|
|
|