Included tests for internal/collector/source

internal/collector/depsdev/source_test.go

@@ -0,0 +1,180 @@
+package depsdev
+
+import (
+	"context"
+	"net/url"
+	"reflect"
+	"testing"
+	"time"
+
+	"github.com/golang/mock/gomock"
+	"go.uber.org/zap"
+
+	"github.com/ossf/criticality_score/internal/collector/signal"
+)
+
+type mockRepo string
+
+func (r mockRepo) URL() *url.URL {
+	u := &url.URL{
+		Host: string(r),
+	}
+	return u
+}
+
+func Test_parseRepoURL(t *testing.T) {
+	tests := []struct {
+		name            string
+		u               *url.URL
+		wantProjectName string
+		wantProjectType string
+	}{
+		{
+			name: "github.com",
+			u: &url.URL{
+				Host: "github.com",
+			},
+			wantProjectName: "",
+			wantProjectType: "GITHUB",
+		},
+		{
+			name: "random",
+			u: &url.URL{
+				Host: "random",
+			},
+			wantProjectName: "",
+			wantProjectType: "",
+		},
+	}
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			gotProjectName, gotProjectType := parseRepoURL(test.u)
+			if gotProjectName != test.wantProjectName {
+				t.Errorf("parseRepoURL() gotProjectName = %v, want %v", gotProjectName, test.wantProjectName)
+			}
+			if gotProjectType != test.wantProjectType {
+				t.Errorf("parseRepoURL() gotProjectType = %v, want %v", gotProjectType, test.wantProjectType)
+			}
+		})
+	}
+}
+
+func Test_depsDevSource_Get(t *testing.T) {
+	type args struct {
+		ctx       context.Context
+		rHostName string
+		jobID     string
+	}
+	test := struct { //nolint:govet
+		name       string
+		logger     *zap.Logger
+		dependents *dependents
+		args       args
+		want       signal.Set
+		wantErr    bool
+	}{
+		name:       "invalid url",
+		logger:     zap.NewNop(),
+		dependents: &dependents{},
+		args: args{
+			ctx:       context.Background(),
+			rHostName: "random",
+		},
+		want: &depsDevSet{},
+	}
+
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+
+	c := &depsDevSource{
+		logger:     test.logger,
+		dependents: test.dependents,
+	}
+	got, err := c.Get(test.args.ctx, mockRepo(test.args.rHostName), test.args.jobID)
+	if (err != nil) != test.wantErr {
+		t.Errorf("Get() error = %v, wantErr %v", err, test.wantErr)
+		return
+	}
+	if !reflect.DeepEqual(got, test.want) {
+		t.Errorf("Get() got = %v, want %v", got, test.want)
+	}
+}
+
+func TestNewSource(t *testing.T) {
+	type args struct {
+		ctx         context.Context
+		logger      *zap.Logger
+		projectID   string
+		datasetName string
+		datasetTTL  time.Duration
+	}
+	test := struct { //nolint:govet
+		name    string
+		args    args
+		want    signal.Source
+		wantErr bool
+	}{
+		name: "new client error",
+		args: args{
+			ctx:         context.Background(),
+			logger:      zap.NewNop(),
+			projectID:   "",
+			datasetName: "dataset",
+			datasetTTL:  time.Hour,
+		},
+		wantErr: true,
+	}
+
+	got, err := NewSource(test.args.ctx, test.args.logger, test.args.projectID, test.args.datasetName, test.args.datasetTTL)
+	if (err != nil) != test.wantErr {
+		t.Errorf("NewSource() error = %v, wantErr %v", err, test.wantErr)
+		return
+	}
+	if !reflect.DeepEqual(got, test.want) {
+		t.Errorf("NewSource() got = %v, want %v", got, test.want)
+	}
+}
+
+func Test_depsDevSource_IsSupported(t *testing.T) {
+	type fields struct {
+		logger     *zap.Logger
+		dependents *dependents
+	}
+	tests := []struct {
+		name      string
+		fields    fields
+		rHostName string
+		want      bool
+	}{
+		{
+			name: "github",
+			fields: fields{
+				logger: zap.NewNop(),
+			},
+			rHostName: "github.com",
+			want:      true,
+		},
+		{
+			name: "random",
+			fields: fields{
+				logger: zap.NewNop(),
+			},
+			rHostName: "random",
+			want:      false,
+		},
+	}
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			c := &depsDevSource{
+				logger:     test.fields.logger,
+				dependents: test.fields.dependents,
+			}
+			ctrl := gomock.NewController(t)
+			defer ctrl.Finish()
+
+			if got := c.IsSupported(mockRepo(test.rHostName)); got != test.want {
+				t.Errorf("IsSupported() = %v, want %v", got, test.want)
+			}
+		})
+	}
+}