mirror of
https://github.com/trufflesecurity/trufflehog.git
synced 2024-09-20 14:42:03 +00:00
ebfbd21707
* inital work * fix and add tests * uncomment * fix seek end * use buffer pool * revert timeout * make linter happy * More linting :()
36 lines
677 B
Go
36 lines
677 B
Go
package handlers
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/context"
|
|
)
|
|
|
|
func TestHandleARFile(t *testing.T) {
|
|
file, err := os.Open("testdata/test.deb")
|
|
assert.Nil(t, err)
|
|
defer file.Close()
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
|
defer cancel()
|
|
|
|
rdr, err := newFileReader(file)
|
|
assert.NoError(t, err)
|
|
defer rdr.Close()
|
|
|
|
handler := newARHandler()
|
|
archiveChan, err := handler.HandleFile(context.AddLogger(ctx), rdr)
|
|
assert.NoError(t, err)
|
|
|
|
wantChunkCount := 102
|
|
count := 0
|
|
for range archiveChan {
|
|
count++
|
|
}
|
|
|
|
assert.Equal(t, wantChunkCount, count)
|
|
}
|