add benchmark

This commit is contained in:
Ahrav Dutta
2024-04-20 11:29:53 -07:00
parent ace7500eeb
commit df29eb0a3a
+25
View File
@@ -65,6 +65,31 @@ func BenchmarkHandleFile(b *testing.B) {
} }
} }
func BenchmarkHandleZipFile(b *testing.B) {
file, err := os.Open("testdata/example.zip.gz")
assert.Nil(b, err)
defer file.Close()
b.ResetTimer()
for i := 0; i < b.N; i++ {
sourceChan := make(chan *sources.Chunk, 1)
b.StartTimer()
go func() {
defer close(sourceChan)
err := HandleFile(context.Background(), file, &sources.Chunk{}, sources.ChanReporter{Ch: sourceChan})
assert.NoError(b, err)
}()
for range sourceChan {
}
b.StopTimer()
_, err = file.Seek(0, io.SeekStart)
assert.NoError(b, err)
}
}
func TestSkipArchive(t *testing.T) { func TestSkipArchive(t *testing.T) {
file, err := os.Open("testdata/test.tgz") file, err := os.Open("testdata/test.tgz")
assert.Nil(t, err) assert.Nil(t, err)