Skip to content

Commit bb3464e

Browse files
Added tests for new .msg and .doc file types to skipArchiverMimeTypes (#4546)
* Added Test for new .msg and .doc file types to skipArchiverMimeTypes * changed tests to align with code review
1 parent f1ccb39 commit bb3464e

File tree

3 files changed

+25
-0
lines changed

3 files changed

+25
-0
lines changed

pkg/handlers/handlers_test.go

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -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+
297322
func BenchmarkHandleAR(b *testing.B) {
298323
file, err := os.Open("testdata/test.deb")
299324
assert.Nil(b, err)

pkg/handlers/testdata/test.doc

29 KB
Binary file not shown.

pkg/handlers/testdata/test.msg

42.5 KB
Binary file not shown.

0 commit comments

Comments
 (0)