Change chunker test source (#959)

* Change chunker test source

* Emit chunk if the size isn't 0
This commit is contained in:
Bill Rich
2022-12-06 12:45:08 -08:00
committed by GitHub
parent 9f99ee470d
commit 1a1c2e275e
2 changed files with 5 additions and 8 deletions
+3 -1
View File
@@ -37,7 +37,9 @@ func Chunker(originalChunk *Chunk) chan *Chunk {
}
peekData, _ := reader.Peek(PeekSize)
chunk.Data = append(chunkBytes[:n], peekData...)
chunkChan <- &chunk
if n > 0 {
chunkChan <- &chunk
}
if errors.Is(err, io.EOF) {
break
}
+2 -7
View File
@@ -5,19 +5,14 @@ import (
"bytes"
"errors"
"io"
"net/http"
"testing"
diskbufferreader "github.com/bill-rich/disk-buffer-reader"
)
func TestChunker(t *testing.T) {
resp, err := http.Get("https://raw.githubusercontent.com/bill-rich/bad-secrets/master/FifteenMB.gz")
if err != nil {
t.Fatal(err)
}
defer resp.Body.Close()
reReader, err := diskbufferreader.New(resp.Body)
byteBuffer := bytes.NewBuffer(make([]byte, ChunkSize*9))
reReader, err := diskbufferreader.New(byteBuffer)
if err != nil {
t.Fatal(err)
}