148 lines
3.2 KiB
Go
148 lines
3.2 KiB
Go
package circleci
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"google.golang.org/protobuf/types/known/anypb"
|
|
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/common"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/context"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/pb/sourcespb"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/sources"
|
|
)
|
|
|
|
func TestSource_Scan(t *testing.T) {
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
|
|
defer cancel()
|
|
|
|
secret, err := common.GetSecret(ctx, "trufflehog-testing", "test")
|
|
if err != nil {
|
|
t.Fatal(fmt.Errorf("failed to access secret: %v", err))
|
|
}
|
|
// Fix the chunks test, which might involve updating this token.
|
|
token := secret.MustGetField("CIRCLECI_TOKEN")
|
|
|
|
type init struct {
|
|
name string
|
|
verify bool
|
|
connection *sourcespb.CircleCI
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
init init
|
|
wantErr bool
|
|
totalScannedProjects int32
|
|
}{
|
|
{
|
|
name: "scan all projects",
|
|
init: init{
|
|
name: "trufflehog-test",
|
|
connection: &sourcespb.CircleCI{
|
|
Credential: &sourcespb.CircleCI_Token{
|
|
Token: token,
|
|
},
|
|
},
|
|
verify: true,
|
|
},
|
|
wantErr: false,
|
|
totalScannedProjects: 3,
|
|
},
|
|
{
|
|
name: "invalid token",
|
|
init: init{
|
|
name: "invalid token test",
|
|
connection: &sourcespb.CircleCI{
|
|
Credential: &sourcespb.CircleCI_Token{
|
|
Token: "invalid-token",
|
|
},
|
|
},
|
|
verify: true,
|
|
},
|
|
wantErr: true,
|
|
totalScannedProjects: 0,
|
|
},
|
|
{
|
|
name: "empty token",
|
|
init: init{
|
|
name: "empty token test",
|
|
connection: &sourcespb.CircleCI{
|
|
Credential: &sourcespb.CircleCI_Token{
|
|
Token: "",
|
|
},
|
|
},
|
|
verify: true,
|
|
},
|
|
wantErr: true,
|
|
totalScannedProjects: 0,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
s := Source{}
|
|
|
|
conn, err := anypb.New(tt.init.connection)
|
|
assert.NoError(t, err)
|
|
|
|
err = s.Init(ctx, tt.init.name, 0, 0, tt.init.verify, conn, 5)
|
|
assert.NoError(t, err)
|
|
|
|
chunksCh := make(chan *sources.Chunk, 1000)
|
|
chunkErr := s.Chunks(ctx, chunksCh)
|
|
close(chunksCh)
|
|
|
|
// check error
|
|
if tt.wantErr {
|
|
assert.Error(t, chunkErr)
|
|
} else {
|
|
assert.NoError(t, chunkErr)
|
|
}
|
|
|
|
// check total count of projects scanned
|
|
progress := s.GetProgress()
|
|
assert.Equal(t, tt.totalScannedProjects, progress.SectionsCompleted)
|
|
})
|
|
}
|
|
}
|
|
|
|
// additional test for edge cases
|
|
func TestSource_EdgeCases(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
test func(t *testing.T)
|
|
}{
|
|
{
|
|
name: "nil connection",
|
|
test: func(t *testing.T) {
|
|
s := Source{}
|
|
err := s.Init(context.Background(), "test", 0, 0, false, nil, 5)
|
|
assert.Error(t, err)
|
|
},
|
|
},
|
|
{
|
|
name: "cancelled context",
|
|
test: func(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
cancel() // cancel immediately
|
|
|
|
s := Source{}
|
|
conn, _ := anypb.New(&sourcespb.CircleCI{
|
|
Credential: &sourcespb.CircleCI_Token{
|
|
Token: "test-token",
|
|
},
|
|
})
|
|
|
|
err := s.Init(ctx, "test", 0, 0, false, conn, 5)
|
|
assert.NoError(t, err)
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, tt.test)
|
|
}
|
|
}
|