trivy/pkg/fanal/image/registry/google/google_test.go

package google

import (
	"errors"
	"reflect"
	"testing"

	"github.com/GoogleCloudPlatform/docker-credential-gcr/v2/store"

	"github.com/aquasecurity/trivy/pkg/fanal/types"
)

func TestCheckOptions(t *testing.T) {
	var tests = map[string]struct {
		domain  string
		opt     types.RegistryOptions
		grc     *GoogleRegistryClient
		wantErr error
	}{
		"InvalidURL": {
			domain:  "alpine:3.9",
			wantErr: types.InvalidURLPattern,
		},
		"InvalidDomain": {
			domain:  "not-gcr.io",
			wantErr: types.InvalidURLPattern,
		},
		"NoOption": {
			domain: "gcr.io",
			grc:    &GoogleRegistryClient{domain: "gcr.io"},
		},
		"CredOption": {
			domain: "gcr.io",
			opt:    types.RegistryOptions{GCPCredPath: "/path/to/file.json"},
			grc: &GoogleRegistryClient{
				domain: "gcr.io",
				Store:  store.NewGCRCredStore("/path/to/file.json"),
			},
		},
	}

	for testname, v := range tests {
		g := &Registry{}
		grc, err := g.CheckOptions(v.domain, v.opt)
		if v.wantErr != nil {
			if err == nil {
				t.Errorf("%s : expected error but no error", testname)
				continue
			}
			if !errors.Is(err, v.wantErr) {
				t.Errorf("[%s]\nexpected error based on %v\nactual : %v", testname, v.wantErr, err)
			}
			continue
		}
		if !reflect.DeepEqual(v.grc, grc) {
			t.Errorf("[%s]\nexpected : %v\nactual : %v", testname, v.grc, grc)
		}
	}
}