mirror of
https://git.vectorsigma.ru/public/atlantis.git
synced 2026-07-28 23:18:15 +00:00
4a921b3d65
* Rename arguments in mocked functions from result to res The code generated by pegomock is broken if any of the arguments to the mocked function are named "result", so they must be named something else. * Regenerate all mocks with make go-generate The mocks now work without further modification after running go-generate.
43 lines
1.5 KiB
Go
43 lines
1.5 KiB
Go
package jobs
|
|
|
|
import (
|
|
"github.com/runatlantis/atlantis/server/events/command"
|
|
"github.com/runatlantis/atlantis/server/events/models"
|
|
)
|
|
|
|
//go:generate pegomock generate --package mocks -o mocks/mock_project_job_url_generator.go ProjectJobURLGenerator
|
|
|
|
// ProjectJobURLGenerator generates urls to view project's progress.
|
|
type ProjectJobURLGenerator interface {
|
|
GenerateProjectJobURL(p command.ProjectContext) (string, error)
|
|
}
|
|
|
|
//go:generate pegomock generate --package mocks -o mocks/mock_project_status_updater.go ProjectStatusUpdater
|
|
|
|
type ProjectStatusUpdater interface {
|
|
// UpdateProject sets the commit status for the project represented by
|
|
// ctx.
|
|
UpdateProject(ctx command.ProjectContext, cmdName command.Name, status models.CommitStatus, url string, res *command.ProjectResult) error
|
|
}
|
|
|
|
type JobURLSetter struct {
|
|
projectJobURLGenerator ProjectJobURLGenerator
|
|
projectStatusUpdater ProjectStatusUpdater
|
|
}
|
|
|
|
func NewJobURLSetter(projectJobURLGenerator ProjectJobURLGenerator, projectStatusUpdater ProjectStatusUpdater) *JobURLSetter {
|
|
return &JobURLSetter{
|
|
projectJobURLGenerator: projectJobURLGenerator,
|
|
projectStatusUpdater: projectStatusUpdater,
|
|
}
|
|
}
|
|
|
|
func (j *JobURLSetter) SetJobURLWithStatus(ctx command.ProjectContext, cmdName command.Name, status models.CommitStatus, result *command.ProjectResult) error {
|
|
url, err := j.projectJobURLGenerator.GenerateProjectJobURL(ctx)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return j.projectStatusUpdater.UpdateProject(ctx, cmdName, status, url, result)
|
|
}
|