trufflehog/pkg/detectors/detectors_test.go
Dustin Decker 52ed87edb7
Add an option to filter unverified results using shannon entropy (#1875)
* Add an option to filter unverified results using shannon entropy

* lint

* add test, update test, and optimize
2023-10-08 19:52:28 -07:00

39 lines
779 B
Go

//go:build detectors
// +build detectors
package detectors
import "testing"
func TestPrefixRegex(t *testing.T) {
tests := []struct {
keywords []string
expected string
}{
{
keywords: []string{"securitytrails"},
expected: `(?i)(?:securitytrails)(?:.|[\n\r]){0,40}`,
},
{
keywords: []string{"zipbooks"},
expected: `(?i)(?:zipbooks)(?:.|[\n\r]){0,40}`,
},
{
keywords: []string{"wrike"},
expected: `(?i)(?:wrike)(?:.|[\n\r]){0,40}`,
},
}
for _, tt := range tests {
got := PrefixRegex(tt.keywords)
if got != tt.expected {
t.Errorf("PrefixRegex(%v) got: %v want: %v", tt.keywords, got, tt.expected)
}
}
}
func BenchmarkPrefixRegex(b *testing.B) {
kws := []string{"securitytrails"}
for i := 0; i < b.N; i++ {
PrefixRegex(kws)
}
}