Files
trufflehog/pkg/detectors/multi_part_credential_provider_test.go
Cody Rose 15719c2c4f Fix integration tests (#2970)
The recent engine changes (#2887) broke some integration tests. This PR fixes them. I also fixed an unrelated linter issue.
2024-06-14 14:30:18 -04:00

37 lines
756 B
Go

package detectors
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestMultiPartCredentialProviders(t *testing.T) {
testCases := []struct {
name string
provider MultiPartCredentialProvider
expectedSpan int64
}{
{
name: "DefaultMultiPartCredentialProvider",
provider: DefaultMultiPartCredentialProvider{},
expectedSpan: defaultMaxCredentialSpan,
},
{
name: "CustomMultiPartCredentialProvider",
provider: NewCustomMultiPartCredentialProvider(2048),
expectedSpan: 2048,
},
}
for _, tc := range testCases {
tc := tc
t.Run(tc.name, func(t *testing.T) {
t.Parallel()
span := tc.provider.MaxCredentialSpan()
assert.Equal(t, tc.expectedSpan, span)
})
}
}