mirror of
https://git.vectorsigma.ru/public/atlantis.git
synced 2026-07-29 11:58:36 +00:00
3954955e13
* fix(deps): update module github.com/petergtz/pegomock/v3 to v4 in go.mod * remove pegomock generate m option, which is not support after v4 * make regen-mocks * replace pegomock v4 primitive eq/matchers * convert pegomock v4 Eq/Any matchers * remove custom models.Repo matcher * pegomock v4 cannot use result method args ref https://github.com/petergtz/pegomock/issues/123 --------- Co-authored-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com>
89 lines
2.8 KiB
Go
89 lines
2.8 KiB
Go
package events_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
. "github.com/petergtz/pegomock/v4"
|
|
"github.com/runatlantis/atlantis/server/events"
|
|
"github.com/runatlantis/atlantis/server/events/command"
|
|
"github.com/runatlantis/atlantis/server/events/models"
|
|
"github.com/runatlantis/atlantis/server/events/models/testdata"
|
|
"github.com/runatlantis/atlantis/server/logging"
|
|
"github.com/runatlantis/atlantis/server/metrics"
|
|
. "github.com/runatlantis/atlantis/testing"
|
|
)
|
|
|
|
func TestImportCommandRunner_Run(t *testing.T) {
|
|
logger := logging.NewNoopLogger(t)
|
|
RegisterMockTestingT(t)
|
|
|
|
tests := []struct {
|
|
name string
|
|
silenced bool
|
|
pullReqStatus models.PullReqStatus
|
|
projectCmds []command.ProjectContext
|
|
expComment string
|
|
expNoComment bool
|
|
}{
|
|
{
|
|
name: "success with zero projects",
|
|
pullReqStatus: models.PullReqStatus{
|
|
ApprovalStatus: models.ApprovalStatus{IsApproved: true},
|
|
Mergeable: true,
|
|
},
|
|
projectCmds: []command.ProjectContext{},
|
|
expComment: "Ran Import for 0 projects:",
|
|
},
|
|
{
|
|
name: "failure with multiple projects",
|
|
pullReqStatus: models.PullReqStatus{
|
|
ApprovalStatus: models.ApprovalStatus{IsApproved: true},
|
|
Mergeable: true,
|
|
},
|
|
projectCmds: []command.ProjectContext{{}, {}},
|
|
expComment: "**Import Failed**: import cannot run on multiple projects. please specify one project.",
|
|
},
|
|
{
|
|
name: "no comment with zero projects and silencing",
|
|
pullReqStatus: models.PullReqStatus{
|
|
ApprovalStatus: models.ApprovalStatus{IsApproved: true},
|
|
Mergeable: true,
|
|
},
|
|
projectCmds: []command.ProjectContext{},
|
|
silenced: true,
|
|
expNoComment: true,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
vcsClient := setup(t, func(tc *TestConfig) {
|
|
tc.SilenceNoProjects = tt.silenced
|
|
})
|
|
|
|
scopeNull, _, _ := metrics.NewLoggingScope(logger, "atlantis")
|
|
modelPull := models.PullRequest{BaseRepo: testdata.GithubRepo, State: models.OpenPullState, Num: testdata.Pull.Num}
|
|
ctx := &command.Context{
|
|
User: testdata.User,
|
|
Log: logging.NewNoopLogger(t),
|
|
Scope: scopeNull,
|
|
Pull: modelPull,
|
|
HeadRepo: testdata.GithubRepo,
|
|
Trigger: command.CommentTrigger,
|
|
}
|
|
cmd := &events.CommentCommand{Name: command.Import}
|
|
|
|
When(pullReqStatusFetcher.FetchPullStatus(modelPull)).ThenReturn(tt.pullReqStatus, nil)
|
|
When(projectCommandBuilder.BuildImportCommands(ctx, cmd)).ThenReturn(tt.projectCmds, nil)
|
|
|
|
importCommandRunner.Run(ctx, cmd)
|
|
|
|
Assert(t, ctx.PullRequestStatus.Mergeable == true, "PullRequestStatus must be set for import_requirements")
|
|
if tt.expNoComment {
|
|
vcsClient.VerifyWasCalled(Never()).CreateComment(Any[models.Repo](), Any[int](), Any[string](), Any[string]())
|
|
} else {
|
|
vcsClient.VerifyWasCalledOnce().CreateComment(testdata.GithubRepo, modelPull.Num, tt.expComment, "import")
|
|
}
|
|
})
|
|
}
|
|
}
|