Fix e2e branch (#1835)

This commit is contained in:
laurentsimon 2022-04-13 09:16:38 -07:00 committed by GitHub
parent eedd16d5be
commit 91202855fd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 28 additions and 0 deletions

View File

@ -33,6 +33,8 @@ var _ = Describe("E2E TEST: githubrepo.branchesHandler", func() {
})
Context("E2E TEST: Validate query cost", func() {
skipIfTokenIsNot(githubWorkflowDefaultTokenType, "GITHUB_TOKEN only")
It("Should not have increased for HEAD query", func() {
repourl := &repoURL{
owner: "ossf",

View File

@ -16,6 +16,7 @@ package githubrepo
import (
"context"
"fmt"
"net/http"
"os"
"testing"
@ -39,6 +40,21 @@ func TestGithubrepo(t *testing.T) {
var graphClient *githubv4.Client
type tokenType int
const (
patTokenType tokenType = iota
githubWorkflowDefaultTokenType
)
var tokType tokenType
func skipIfTokenIsNot(t tokenType, msg string) {
if tokType != t {
Skip(msg)
}
}
var _ = BeforeSuite(func() {
ctx := context.Background()
logger := log.NewLogger(log.DebugLevel)
@ -47,4 +63,14 @@ var _ = BeforeSuite(func() {
Transport: rt,
}
graphClient = githubv4.NewClient(httpClient)
tt := os.Getenv("TOKEN_TYPE")
switch tt {
case "PAT":
tokType = patTokenType
case "GITHUB_TOKEN":
tokType = githubWorkflowDefaultTokenType
default:
panic(fmt.Sprintf("invald TOKEN_TYPE: %s", tt))
}
})