mirror of
https://git.vectorsigma.ru/public/atlantis.git
synced 2026-08-05 17:38:32 +00:00
af2a806870
* Adding policy_check support into yaml config * Added policy check model and runtime structs * Adding BuildPolicyCheckCommand to ProjectCommandBuilder * Return incorrectly deleted code * Remove BuildAutoPolicyPlanCommand from ProjectCommandBuilder * Split runAutoCommand into two functions runAutoPlanCommand - does what originally RunAutoplancommand was doing except now it returns CommandResult and []models.ProjectCommandContext runAutoPolicyCheckCommand - accepts CommandContext, CommandResult, and []models.ProjectCommandContext as arguments and runs PolicyCheckStep runner * Refactor RunCommentCommand * Remove BuildPolicyCheckCommand and rename StepCmdExec back to TerraformExec * Add policy step runner logic and conftest interfaces. * Add show step runner to policy check stage. * Adding models.PolicyCheckCommand to buildCtx This also means buildPlanAllCommands call buildCtx twice once with models.PlanCommand and once with models.PolicyCheckCommand * Adding new project_command_builder that supports policy_check * Refactoring PolicyCheck specific logic into a PolicyCheckProjectCommandBuilder * Moving events.CommandContext to models.CommandContext this will allow me to remove buildCtx method and move ProjectCommandContext creation into models package * Policy Owners might be different types, for that reason we are refactoring Owners into its own struct with specific keys defining different owner types Co-authored-by: Nish Krishnan <nishk@lyft.com> Co-authored-by: Nish Krishnan <nishkrishnan@users.noreply.github.com>
95 lines
2.5 KiB
Go
95 lines
2.5 KiB
Go
package runtime
|
|
|
|
import (
|
|
"errors"
|
|
"io/ioutil"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/go-version"
|
|
. "github.com/petergtz/pegomock"
|
|
"github.com/runatlantis/atlantis/server/events/models"
|
|
"github.com/runatlantis/atlantis/server/events/terraform/mocks"
|
|
"github.com/runatlantis/atlantis/server/logging"
|
|
. "github.com/runatlantis/atlantis/testing"
|
|
)
|
|
|
|
func TestShowStepRunnner(t *testing.T) {
|
|
logger := logging.NewNoopLogger()
|
|
path, _ := ioutil.TempDir("", "")
|
|
resultPath := filepath.Join(path, "test-default.json")
|
|
envs := map[string]string{"key": "val"}
|
|
tfVersion, _ := version.NewVersion("0.12")
|
|
context := models.ProjectCommandContext{
|
|
Workspace: "default",
|
|
ProjectName: "test",
|
|
Log: logger,
|
|
}
|
|
|
|
RegisterMockTestingT(t)
|
|
|
|
mockExecutor := mocks.NewMockClient()
|
|
|
|
subject := ShowStepRunner{
|
|
TerraformExecutor: mockExecutor,
|
|
DefaultTFVersion: tfVersion,
|
|
}
|
|
|
|
t.Run("success", func(t *testing.T) {
|
|
|
|
When(mockExecutor.RunCommandWithVersion(
|
|
logger, path, []string{"show", "-no-color", "-json", filepath.Join(path, "test-default.tfplan")}, envs, tfVersion, context.Workspace,
|
|
)).ThenReturn("success", nil)
|
|
|
|
r, err := subject.Run(context, []string{}, path, envs)
|
|
|
|
Ok(t, err)
|
|
|
|
actual, _ := ioutil.ReadFile(resultPath)
|
|
|
|
actualStr := string(actual)
|
|
Assert(t, actualStr == "success", "got expected result")
|
|
Assert(t, r == "success", "returned expected result")
|
|
|
|
})
|
|
|
|
t.Run("success w/ version override", func(t *testing.T) {
|
|
|
|
v, _ := version.NewVersion("0.13.0")
|
|
|
|
contextWithVersionOverride := models.ProjectCommandContext{
|
|
Workspace: "default",
|
|
ProjectName: "test",
|
|
Log: logger,
|
|
TerraformVersion: v,
|
|
}
|
|
|
|
When(mockExecutor.RunCommandWithVersion(
|
|
logger, path, []string{"show", "-no-color", "-json", filepath.Join(path, "test-default.tfplan")}, envs, v, context.Workspace,
|
|
)).ThenReturn("success", nil)
|
|
|
|
r, err := subject.Run(contextWithVersionOverride, []string{}, path, envs)
|
|
|
|
Ok(t, err)
|
|
|
|
actual, _ := ioutil.ReadFile(resultPath)
|
|
|
|
actualStr := string(actual)
|
|
Assert(t, actualStr == "success", "got expected result")
|
|
Assert(t, r == "success", "returned expected result")
|
|
|
|
})
|
|
|
|
t.Run("failure running command", func(t *testing.T) {
|
|
When(mockExecutor.RunCommandWithVersion(
|
|
logger, path, []string{"show", "-no-color", "-json", filepath.Join(path, "test-default.tfplan")}, envs, tfVersion, context.Workspace,
|
|
)).ThenReturn("success", errors.New("error"))
|
|
|
|
_, err := subject.Run(context, []string{}, path, envs)
|
|
|
|
Assert(t, err != nil, "error is returned")
|
|
|
|
})
|
|
|
|
}
|