This change includes a handful of stack-related CLI formatting improvements that I've been noodling on in the background for a while, based on things that tend to trip up demos and the inner loop workflow. This includes: * If `pulumi stack select` is run by itself, use an interactive CLI menu to let the user select an existing stack, or choose to create a new one. This looks as follows $ pulumi stack select Please choose a stack, or choose to create a new one: abcdef babblabblabble > currentlyselected defcon <create a new stack> and is navigated in the usual way (key up, down, enter). * If a stack name is passed that does not exist, prompt the user to ask whether s/he wants to create one on-demand. This hooks interesting moments in time, like `pulumi stack select foo`, and cuts down on the need to run additional commands. * If a current stack is required, but none is currently selected, then pop the same interactive menu shown above to select one. Depending on the command being run, we may or may not show the option to create a new stack (e.g., that doesn't make much sense when you're running `pulumi destroy`, but might when you're running `pulumi stack`). This again lets you do with a single command what would have otherwise entailed an error with multiple commands to recover from it. * If you run `pulumi stack init` without any additional arguments, we interactively prompt for the stack name. Before, we would error and you'd then need to run `pulumi stack init <name>`. * Colorize some things nicely; for example, now all prompts will by default become bright white.
278 lines
10 KiB
Go
278 lines
10 KiB
Go
// Copyright 2016-2017, Pulumi Corporation. All rights reserved.
|
|
|
|
package tests
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/pulumi/pulumi/pkg/backend"
|
|
ptesting "github.com/pulumi/pulumi/pkg/testing"
|
|
"github.com/pulumi/pulumi/pkg/testing/integration"
|
|
)
|
|
|
|
// deleteIfNotFailed deletes the files in the testing environment if the testcase has
|
|
// not failed. (Otherwise they are left to aid debugging.)
|
|
func deleteIfNotFailed(e *ptesting.Environment) {
|
|
if !e.T.Failed() {
|
|
e.DeleteEnvironment()
|
|
}
|
|
}
|
|
|
|
// assertHasNoHistory runs `pulumi history` and confirms an error that the stack has not
|
|
// ever been updated.
|
|
func assertHasNoHistory(e *ptesting.Environment) {
|
|
// NOTE: pulumi returns with exit code 0 in this scenario.
|
|
out, err := e.RunCommand("pulumi", "history")
|
|
assert.Equal(e.T, "", err)
|
|
assert.Equal(e.T, "Stack has never been updated\n", out)
|
|
}
|
|
|
|
func assertEnvValue(t *testing.T, update backend.UpdateInfo, key, val string) {
|
|
t.Helper()
|
|
v, ok := update.Environment[key]
|
|
if !ok {
|
|
t.Errorf("Did not find key %q in update environment", key)
|
|
} else {
|
|
assert.Equal(t, val, v, "Comparing Environment values for key %q", key)
|
|
}
|
|
}
|
|
|
|
func assertEnvKeyNotFound(t *testing.T, update backend.UpdateInfo, key string) {
|
|
t.Helper()
|
|
_, found := update.Environment[key]
|
|
assert.False(t, found, "Did not expect to find key %q in update environment", key)
|
|
}
|
|
|
|
func TestHistoryCommand(t *testing.T) {
|
|
// The --output-json flag on pulumi history is hidden behind an environment variable.
|
|
os.Setenv("PULUMI_DEBUG_COMMANDS", "1")
|
|
defer os.Unsetenv("PULUMI_DEBUG_COMMANDS")
|
|
|
|
// We fail if no stack is selected.
|
|
t.Run("NoStackSelected", func(t *testing.T) {
|
|
e := ptesting.NewEnvironment(t)
|
|
defer deleteIfNotFailed(e)
|
|
integration.CreateBasicPulumiRepo(e)
|
|
|
|
out, err := e.RunCommandExpectError("pulumi", "history")
|
|
assert.Equal(t, "", out)
|
|
assert.NotEqual(t, err, "")
|
|
})
|
|
|
|
// We don't display any history for a stack that has never been updated.
|
|
t.Run("NoUpdates", func(t *testing.T) {
|
|
e := ptesting.NewEnvironment(t)
|
|
defer deleteIfNotFailed(e)
|
|
integration.CreateBasicPulumiRepo(e)
|
|
|
|
e.RunCommand("pulumi", "stack", "init", "no-updates-test", "--local")
|
|
assertHasNoHistory(e)
|
|
})
|
|
|
|
// The "history" command uses the currently selected stack.
|
|
t.Run("CurrentlySelectedStack", func(t *testing.T) {
|
|
e := ptesting.NewEnvironment(t)
|
|
defer deleteIfNotFailed(e)
|
|
integration.CreateBasicPulumiRepo(e)
|
|
e.ImportDirectory("integration/stack_outputs")
|
|
|
|
e.RunCommand("pulumi", "stack", "init", "stack-without-updates", "--local")
|
|
e.RunCommand("pulumi", "stack", "init", "history-test", "--local")
|
|
|
|
// Update the history-test stack.
|
|
e.RunCommand("yarn", "install")
|
|
e.RunCommand("yarn", "link", "@pulumi/pulumi")
|
|
e.RunCommand("yarn", "run", "build")
|
|
e.RunCommand("pulumi", "update", "-m", "updating stack...")
|
|
|
|
// Confirm we see the update message in thie history output.
|
|
out, err := e.RunCommand("pulumi", "history")
|
|
assert.Equal(t, "", err)
|
|
assert.Contains(t, out, "updating stack...")
|
|
|
|
// Change stack and confirm the history command honors the selected stack.
|
|
e.RunCommand("pulumi", "stack", "select", "stack-without-updates")
|
|
assertHasNoHistory(e)
|
|
|
|
// Change stack back, and confirm still has history.
|
|
e.RunCommand("pulumi", "stack", "select", "history-test")
|
|
out, err = e.RunCommand("pulumi", "history")
|
|
assert.Equal(t, "", err)
|
|
assert.Contains(t, out, "updating stack...")
|
|
})
|
|
|
|
// That the history command contains accurate data about the update history.
|
|
t.Run("Data(Deploy,Kind,Result)", func(t *testing.T) {
|
|
e := ptesting.NewEnvironment(t)
|
|
defer deleteIfNotFailed(e)
|
|
integration.CreateBasicPulumiRepo(e)
|
|
e.ImportDirectory("integration/stack_outputs")
|
|
|
|
e.RunCommand("pulumi", "stack", "init", "history-test", "--local")
|
|
|
|
// Update the history-test stack.
|
|
e.RunCommand("yarn", "install")
|
|
e.RunCommand("yarn", "link", "@pulumi/pulumi")
|
|
e.RunCommand("yarn", "run", "build")
|
|
e.RunCommand("pulumi", "update", "-m", "first update (successful)")
|
|
|
|
// Now we "break" the program, by adding gibberish to bin/index.js.
|
|
indexJS := path.Join(e.CWD, "bin", "index.js")
|
|
origContents, err := ioutil.ReadFile(indexJS)
|
|
assert.NoError(t, err, "Reading bin/index.js")
|
|
|
|
var invalidJS bytes.Buffer
|
|
invalidJS.Write(origContents)
|
|
invalidJS.WriteString("\n\n... with random text -> syntax error, too")
|
|
err = ioutil.WriteFile(indexJS, invalidJS.Bytes(), os.ModePerm)
|
|
assert.NoError(t, err, "Writing bin/index.js")
|
|
|
|
e.RunCommandExpectError("pulumi", "update", "-m", "second update (failure)")
|
|
|
|
// Fix it
|
|
err = ioutil.WriteFile(indexJS, origContents, os.ModePerm)
|
|
assert.NoError(t, err, "Writing bin/index.js")
|
|
e.RunCommand("pulumi", "update", "-m", "third update (successful)")
|
|
|
|
// Destroy
|
|
e.RunCommand("pulumi", "destroy", "-m", "fourth update (destroy)", "--yes")
|
|
|
|
// Confirm the history is as expected. Output as JSON and parse the result.
|
|
stdout, stderr := e.RunCommand("pulumi", "history", "--output-json")
|
|
assert.Equal(t, "", stderr)
|
|
|
|
var updateRecords []backend.UpdateInfo
|
|
err = json.Unmarshal([]byte(stdout), &updateRecords)
|
|
if err != nil {
|
|
t.Fatalf("Error marshalling `pulumi history` output as JSON: %v", err)
|
|
}
|
|
if len(updateRecords) != 4 {
|
|
t.Fatalf("didn't get expected number of updates from testcase. Raw history output:\n%v", stdout)
|
|
}
|
|
|
|
// The most recent updates are listed first.
|
|
update := updateRecords[0]
|
|
assert.Equal(t, "fourth update (destroy)", update.Message)
|
|
assert.True(t, backend.DestroyUpdate == update.Kind)
|
|
assert.True(t, backend.SucceededResult == update.Result)
|
|
|
|
update = updateRecords[1]
|
|
assert.Equal(t, "third update (successful)", update.Message)
|
|
assert.True(t, backend.DeployUpdate == update.Kind)
|
|
assert.True(t, backend.SucceededResult == update.Result)
|
|
|
|
update = updateRecords[2]
|
|
assert.Equal(t, "second update (failure)", update.Message)
|
|
assert.True(t, backend.DeployUpdate == update.Kind)
|
|
assert.True(t, backend.FailedResult == update.Result)
|
|
|
|
update = updateRecords[3]
|
|
assert.Equal(t, "first update (successful)", update.Message)
|
|
assert.True(t, backend.DeployUpdate == update.Kind)
|
|
assert.True(t, backend.SucceededResult == update.Result)
|
|
|
|
if t.Failed() {
|
|
t.Logf("Test failed. Printing raw history output:\n%v", stdout)
|
|
}
|
|
|
|
// Call stack rm to run the "delete history file too" codepath.
|
|
e.RunCommand("pulumi", "stack", "rm", "--yes")
|
|
})
|
|
|
|
// We include git-related data in the environment metadata.
|
|
t.Run("Data(Environment[Git])", func(t *testing.T) {
|
|
e := ptesting.NewEnvironment(t)
|
|
defer deleteIfNotFailed(e)
|
|
integration.CreateBasicPulumiRepo(e)
|
|
e.ImportDirectory("integration/stack_outputs")
|
|
|
|
e.RunCommand("pulumi", "stack", "init", "history-test", "--local")
|
|
e.RunCommand("yarn", "install")
|
|
e.RunCommand("yarn", "link", "@pulumi/pulumi")
|
|
e.RunCommand("yarn", "run", "build")
|
|
|
|
// Update 1, git repo that has no commits.
|
|
e.RunCommand("pulumi", "update", "-m", "first update (git repo has no commits)")
|
|
|
|
// Update 2, repo has commit, but no remote.
|
|
e.RunCommand("git", "add", ".")
|
|
e.RunCommand("git", "commit", "-m", "First commit of test files")
|
|
e.RunCommand("pulumi", "update", "-m", "second update (git commit, no remote)")
|
|
|
|
// Update 3, repo has remote and is dirty (by rewriting index.ts).
|
|
indexTS := path.Join(e.CWD, "index.ts")
|
|
origContents, err := ioutil.ReadFile(indexTS)
|
|
assert.NoError(t, err, "Reading index.ts")
|
|
|
|
err = ioutil.WriteFile(indexTS, []byte("change to file..."), os.ModePerm)
|
|
assert.NoError(t, err, "writing index.ts")
|
|
|
|
e.RunCommand("git", "remote", "add", "origin", "git@github.com:rick/c-132")
|
|
e.RunCommand("pulumi", "update", "-m", "third update (is dirty, has remote)")
|
|
|
|
// Update 4, repo is now clean again.
|
|
err = ioutil.WriteFile(indexTS, origContents, os.ModePerm)
|
|
assert.NoError(t, err, "writing index.ts")
|
|
|
|
e.RunCommand("pulumi", "update", "-m", "fourth update (is clean)")
|
|
|
|
// Confirm the history is as expected. Output as JSON and parse the result.
|
|
stdout, stderr := e.RunCommand("pulumi", "history", "--output-json")
|
|
assert.Equal(t, "", stderr)
|
|
|
|
var updateRecords []backend.UpdateInfo
|
|
err = json.Unmarshal([]byte(stdout), &updateRecords)
|
|
if err != nil {
|
|
t.Fatalf("Error marshalling `pulumi history` output as JSON: %v", err)
|
|
}
|
|
if len(updateRecords) != 4 {
|
|
t.Fatalf("didn't get expected number of updates from testcase. Raw history output:\n%v", stdout)
|
|
}
|
|
|
|
// The first update doesn't have any git information, since
|
|
// nothing has been committed yet.
|
|
t.Log("Checking first update...")
|
|
update := updateRecords[3]
|
|
assertEnvKeyNotFound(e.T, update, backend.GitHead)
|
|
|
|
// The second update has a commit.
|
|
t.Log("Checking second update...")
|
|
update = updateRecords[2]
|
|
// We don't know what the SHA will be ahead of time, since the code we use to call `pulumi init`
|
|
// uses the current time as part of the --name flag.
|
|
headSHA, ok := update.Environment[backend.GitHead]
|
|
assert.True(t, ok, "Didn't find %s in environment", backend.GitHead)
|
|
assert.Equal(t, 40, len(headSHA), "Commit SHA was not expected length")
|
|
assertEnvValue(e.T, update, backend.GitDirty, "false")
|
|
// The github-related info is still not set though.
|
|
assertEnvKeyNotFound(e.T, update, backend.GitHubLogin)
|
|
assertEnvKeyNotFound(e.T, update, backend.GitHubRepo)
|
|
|
|
// The third commit sets a remote (which we detect as a GitHub repo) and is now dirty.
|
|
t.Log("Checking third update...")
|
|
update = updateRecords[1]
|
|
assertEnvValue(e.T, update, backend.GitHead, headSHA)
|
|
assertEnvValue(e.T, update, backend.GitDirty, "true")
|
|
assertEnvValue(e.T, update, backend.GitHubLogin, "rick")
|
|
assertEnvValue(e.T, update, backend.GitHubRepo, "c-132")
|
|
|
|
// The fourth commit is clean (by restoring to the previous commit).
|
|
t.Log("Checking fourth update...")
|
|
update = updateRecords[0]
|
|
assertEnvValue(e.T, update, backend.GitHead, headSHA)
|
|
assertEnvValue(e.T, update, backend.GitDirty, "false")
|
|
assertEnvValue(e.T, update, backend.GitHubLogin, "rick")
|
|
assertEnvValue(e.T, update, backend.GitHubRepo, "c-132")
|
|
|
|
if t.Failed() {
|
|
t.Logf("Test failed. Printing raw history output:\n%v\n", stdout)
|
|
}
|
|
})
|
|
}
|