133 lines
2.7 KiB
Go
133 lines
2.7 KiB
Go
package model
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestSetRefAndSha(t *testing.T) {
|
|
log.SetLevel(log.DebugLevel)
|
|
|
|
oldFindGitRef := findGitRef
|
|
oldFindGitRevision := findGitRevision
|
|
defer func() { findGitRef = oldFindGitRef }()
|
|
defer func() { findGitRevision = oldFindGitRevision }()
|
|
|
|
findGitRef = func(file string) (string, error) {
|
|
return "refs/heads/master", nil
|
|
}
|
|
|
|
findGitRevision = func(file string) (string, string, error) {
|
|
return "", "1234fakesha", nil
|
|
}
|
|
|
|
tables := []struct {
|
|
eventName string
|
|
event map[string]interface{}
|
|
ref string
|
|
sha string
|
|
}{
|
|
{
|
|
eventName: "pull_request_target",
|
|
event: map[string]interface{}{
|
|
"pull_request": map[string]interface{}{
|
|
"base": map[string]interface{}{
|
|
"sha": "pr-base-sha",
|
|
},
|
|
},
|
|
},
|
|
ref: "master",
|
|
sha: "pr-base-sha",
|
|
},
|
|
{
|
|
eventName: "pull_request",
|
|
event: map[string]interface{}{
|
|
"number": "1234",
|
|
},
|
|
ref: "refs/pull/1234/merge",
|
|
sha: "1234fakesha",
|
|
},
|
|
{
|
|
eventName: "deployment",
|
|
event: map[string]interface{}{
|
|
"deployment": map[string]interface{}{
|
|
"ref": "refs/heads/somebranch",
|
|
"sha": "deployment-sha",
|
|
},
|
|
},
|
|
ref: "refs/heads/somebranch",
|
|
sha: "deployment-sha",
|
|
},
|
|
{
|
|
eventName: "release",
|
|
event: map[string]interface{}{
|
|
"release": map[string]interface{}{
|
|
"tag_name": "v1.0.0",
|
|
},
|
|
},
|
|
ref: "v1.0.0",
|
|
sha: "1234fakesha",
|
|
},
|
|
{
|
|
eventName: "push",
|
|
event: map[string]interface{}{
|
|
"ref": "refs/heads/somebranch",
|
|
"after": "push-sha",
|
|
"deleted": false,
|
|
},
|
|
ref: "refs/heads/somebranch",
|
|
sha: "push-sha",
|
|
},
|
|
{
|
|
eventName: "unknown",
|
|
event: map[string]interface{}{
|
|
"repository": map[string]interface{}{
|
|
"default_branch": "main",
|
|
},
|
|
},
|
|
ref: "main",
|
|
sha: "1234fakesha",
|
|
},
|
|
{
|
|
eventName: "no-event",
|
|
event: map[string]interface{}{},
|
|
ref: "refs/heads/master",
|
|
sha: "1234fakesha",
|
|
},
|
|
}
|
|
|
|
for _, table := range tables {
|
|
t.Run(table.eventName, func(t *testing.T) {
|
|
ghc := &GithubContext{
|
|
EventName: table.eventName,
|
|
BaseRef: "master",
|
|
Event: table.event,
|
|
}
|
|
|
|
ghc.SetRefAndSha("main", "/some/dir")
|
|
|
|
assert.Equal(t, table.ref, ghc.Ref)
|
|
assert.Equal(t, table.sha, ghc.Sha)
|
|
})
|
|
}
|
|
|
|
t.Run("no-default-branch", func(t *testing.T) {
|
|
findGitRef = func(file string) (string, error) {
|
|
return "", fmt.Errorf("no default branch")
|
|
}
|
|
|
|
ghc := &GithubContext{
|
|
EventName: "no-default-branch",
|
|
Event: map[string]interface{}{},
|
|
}
|
|
|
|
ghc.SetRefAndSha("", "/some/dir")
|
|
|
|
assert.Equal(t, "master", ghc.Ref)
|
|
assert.Equal(t, "1234fakesha", ghc.Sha)
|
|
})
|
|
}
|