Function TestVulnerabilities

checks/vulnerabilities_test.go:28–81

(t *testing.T)

)

func TestVulnerabilities(t *testing.T) {
	t.Parallel()
	tests := []struct {
		err      error
		name     string
		expected clients.VulnerabilitiesResponse
		isError  bool
	}{
		{
			name:     "Valid response",
			isError:  false,
			expected: clients.VulnerabilitiesResponse{},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			ctrl := gomock.NewController(t)
			mockRepo := mockrepo.NewMockRepoClient(ctrl)

			mockRepo.EXPECT().ListCommits().DoAndReturn(func() ([]clients.Commit, error) {
				if tt.err != nil {
					return nil, tt.err
				}
				return []clients.Commit{{SHA: "test"}}, nil
			}).MinTimes(1)

			mockRepo.EXPECT().LocalPath().DoAndReturn(func() (string, error) {
				return "test_path", nil
			}).AnyTimes()

			mockVulnClient := mockrepo.NewMockVulnerabilitiesClient(ctrl)
			mockVulnClient.EXPECT().ListUnfixedVulnerabilities(t.Context(), gomock.Any(), gomock.Any()).DoAndReturn(
				func(ctx context.Context, commit string, localPath string) (clients.VulnerabilitiesResponse, error) {
					return tt.expected, tt.err
				}).MinTimes(1)

			req := checker.CheckRequest{
				RepoClient:            mockRepo,
				Ctx:                   t.Context(),
				VulnerabilitiesClient: mockVulnClient,
			}
			res := Vulnerabilities(&req)
			if !tt.isError && res.Error != nil {
				t.Fail()
			} else if tt.isError && res.Error == nil {
				t.Fail()
			}
			ctrl.Finish()
		})
	}
}