38 lines
790 B
Go
38 lines
790 B
Go
package handlers
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/context"
|
|
bufferwriter "github.com/trufflesecurity/trufflehog/v3/pkg/writers/buffer_writer"
|
|
)
|
|
|
|
func TestHandleRPMFile(t *testing.T) {
|
|
file, err := os.Open("testdata/test.rpm")
|
|
assert.Nil(t, err)
|
|
defer file.Close()
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
|
defer cancel()
|
|
|
|
rdr, err := bufferwriter.NewBufferReadSeekCloser(ctx, file)
|
|
assert.NoError(t, err)
|
|
defer rdr.Close()
|
|
|
|
handler := newRPMHandler()
|
|
archiveChan, err := handler.HandleFile(context.AddLogger(ctx), rdr)
|
|
assert.NoError(t, err)
|
|
|
|
wantChunkCount := 179
|
|
count := 0
|
|
for range archiveChan {
|
|
count++
|
|
}
|
|
|
|
assert.Equal(t, wantChunkCount, count)
|
|
}
|