Wait for Split and Append in benchmark test

This commit is contained in:
Balint Gabor 2018-01-04 12:21:40 +01:00
parent 772153a4b4
commit 755cf50d6f

View file

@ -467,18 +467,18 @@ func benchmarkAppendPyramid(n, m int, t *testing.B) {
data1 := testDataReader(m) data1 := testDataReader(m)
chunkC := make(chan *Chunk, 1000) chunkC := make(chan *Chunk, 1000)
key, _, err := tester.Split(chunker, data, int64(n), chunkC, nil) key, wait, err := tester.Split(chunker, data, int64(n), chunkC, nil)
if err != nil { if err != nil {
tester.t.Fatalf(err.Error()) tester.t.Fatalf(err.Error())
} }
wait()
chunkC = make(chan *Chunk, 1000) chunkC = make(chan *Chunk, 1000)
_, _, err = tester.Append(chunker, key, data1, chunkC, nil) _, wait, err = tester.Append(chunker, key, data1, chunkC, nil)
if err != nil { if err != nil {
tester.t.Fatalf(err.Error()) tester.t.Fatalf(err.Error())
} }
wait()
close(chunkC) close(chunkC)
} }
} }