scorecard/e2e/e2e_suite_test.go
Naveen 4d7fb5d748
🌱 Fix the go.mod with v2 upgrade (#716)
The go.mod and the related files weren't t updated with the v2 upgrade.

https://github.com/ossf/scorecard/issues/711

This fix will address the issue.
2021-07-26 13:01:25 -05:00

77 lines
2.0 KiB
Go

// Copyright 2021 Security Scorecard Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package e2e
import (
"context"
"net/http"
"os"
"testing"
"github.com/google/go-github/v32/github"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"github.com/shurcooL/githubv4"
"go.uber.org/zap"
"github.com/ossf/scorecard/v2/roundtripper"
)
var (
ghClient *github.Client
graphClient *githubv4.Client
httpClient *http.Client
)
func TestE2e(t *testing.T) {
t.Parallel()
RegisterFailHandler(Fail)
RunSpecs(t, "E2e Suite")
}
var _ = BeforeSuite(func() {
// making sure the GITHUB_AUTH_TOKEN is set prior to running e2e tests
token, contains := os.LookupEnv("GITHUB_AUTH_TOKEN")
Expect(contains).ShouldNot(BeFalse(),
"GITHUB_AUTH_TOKEN env variable is not set.The GITHUB_AUTH_TOKEN env variable has to be set to run e2e test.")
Expect(len(token)).ShouldNot(BeZero(), "Length of the GITHUB_AUTH_TOKEN env variable is zero.")
ctx := context.TODO()
logLevel := zap.LevelFlag("verbosity", zap.InfoLevel, "override the default log level")
cfg := zap.NewProductionConfig()
cfg.Level.SetLevel(*logLevel)
logger, err := cfg.Build()
Expect(err).Should(BeNil())
sugar := logger.Sugar()
// nolint
defer logger.Sync() // flushes buffer, if any
Expect(sugar).ShouldNot(BeNil())
rt := roundtripper.NewTransport(ctx, sugar)
httpClient = &http.Client{
Transport: rt,
}
ghClient = github.NewClient(httpClient)
graphClient = githubv4.NewClient(httpClient)
})
var _ = AfterSuite(func() {
})