@@ -17,6 +17,7 @@ import (
1717 "time"
1818
1919 "github.com/stretchr/testify/assert"
20+ "github.com/stretchr/testify/require"
2021 diskbufferreader "github.com/trufflesecurity/disk-buffer-reader"
2122
2223 "github.com/trufflesecurity/trufflehog/v3/pkg/context"
@@ -294,6 +295,30 @@ func TestHandleFileAR(t *testing.T) {
294295 assert .Equal (t , wantChunkCount , len (reporter .Ch ))
295296}
296297
298+ func TestHandleFileMSG (t * testing.T ) {
299+ wantChunkCount := 5
300+ reporter := sources.ChanReporter {Ch : make (chan * sources.Chunk , wantChunkCount )}
301+
302+ file , err := os .Open ("testdata/test.msg" )
303+ require .NoError (t , err )
304+
305+ assert .Empty (t , reporter .Ch )
306+ assert .NoError (t , HandleFile (context .Background (), file , & sources.Chunk {}, reporter ))
307+ assert .Equal (t , wantChunkCount , len (reporter .Ch ))
308+ }
309+
310+ func TestHandleFileDOC (t * testing.T ) {
311+ wantChunkCount := 3
312+ reporter := sources.ChanReporter {Ch : make (chan * sources.Chunk , wantChunkCount )}
313+
314+ file , err := os .Open ("testdata/test.doc" )
315+ require .NoError (t , err )
316+
317+ assert .Empty (t , reporter .Ch )
318+ assert .NoError (t , HandleFile (context .Background (), file , & sources.Chunk {}, reporter ))
319+ assert .Equal (t , wantChunkCount , len (reporter .Ch ))
320+ }
321+
297322func BenchmarkHandleAR (b * testing.B ) {
298323 file , err := os .Open ("testdata/test.deb" )
299324 assert .Nil (b , err )
0 commit comments