Add round-trip test of chunking
parent
b81f1fdbe0
commit
5fd9c9185f
@ -0,0 +1,62 @@
|
||||
package chunking
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"crypto/rand"
|
||||
"io/ioutil"
|
||||
"os"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func Test_ChunkingRoundTrip(t *testing.T) {
|
||||
data := make([]byte, 1024*1024)
|
||||
if _, err := rand.Read(data); err != nil {
|
||||
t.Fatalf("failed to generate random data: %v", err)
|
||||
}
|
||||
|
||||
chunker := NewChunker(bytes.NewReader(data), 1024)
|
||||
|
||||
dir, err := ioutil.TempDir("", "test-*")
|
||||
if err != nil {
|
||||
t.Fatalf("failed to create temp dir: %v", err)
|
||||
}
|
||||
defer os.Remove(dir)
|
||||
|
||||
dechunker, err := NewDechunker(dir)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to create Dechunker: %v", err)
|
||||
}
|
||||
|
||||
// Read chunks from the Chunker and write them to the Dechunker.
|
||||
for {
|
||||
chunk, err := chunker.Next()
|
||||
if err != nil {
|
||||
t.Fatalf("failed to read chunk: %v", err)
|
||||
}
|
||||
done, err := dechunker.WriteChunk(chunk)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to write chunk: %v", err)
|
||||
}
|
||||
if chunk.IsLast {
|
||||
if !done {
|
||||
t.Fatalf("WriteChunk did not return true after writing last chunk")
|
||||
}
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
outFilePath, err := dechunker.Close()
|
||||
if err != nil {
|
||||
t.Fatalf("failed to close Dechunker: %v", err)
|
||||
}
|
||||
defer os.Remove(outFilePath)
|
||||
|
||||
// The output data should be the same as the original data.
|
||||
outData, err := ioutil.ReadFile(outFilePath)
|
||||
if err != nil {
|
||||
t.Fatalf("failed to read output file data: %v", err)
|
||||
}
|
||||
if !bytes.Equal(outData, data) {
|
||||
t.Fatalf("output file data does not match original data")
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue