Merge pull request #4017 from hashicorp/b-vars-refresh

core: Validate context after input of vars on refresh
This commit is contained in:
James Nugent 2015-11-23 16:19:36 +02:00
commit b044340e75
3 changed files with 41 additions and 3 deletions

View File

@ -87,14 +87,16 @@ func (c *RefreshCommand) Run(args []string) int {
c.Ui.Error(err.Error())
return 1
}
if !validateContext(ctx, c.Ui) {
return 1
}
if err := ctx.Input(c.InputMode()); err != nil {
c.Ui.Error(fmt.Sprintf("Error configuring: %s", err))
return 1
}
if !validateContext(ctx, c.Ui) {
return 1
}
newState, err := ctx.Refresh()
if err != nil {
c.Ui.Error(fmt.Sprintf("Error refreshing state: %s", err))

View File

@ -8,6 +8,7 @@ import (
"strings"
"testing"
"bytes"
"github.com/hashicorp/terraform/terraform"
"github.com/mitchellh/cli"
)
@ -413,6 +414,34 @@ func TestRefresh_varFileDefault(t *testing.T) {
}
}
func TestRefresh_varsUnset(t *testing.T) {
// Disable test mode so input would be asked
test = false
defer func() { test = true }()
defaultInputReader = bytes.NewBufferString("bar\n")
state := testState()
statePath := testStateFile(t, state)
p := testProvider()
ui := new(cli.MockUi)
c := &RefreshCommand{
Meta: Meta{
ContextOpts: testCtxConfig(p),
Ui: ui,
},
}
args := []string{
"-state", statePath,
testFixturePath("refresh-unset-var"),
}
if code := c.Run(args); code != 0 {
t.Fatalf("bad: %d\n\n%s", code, ui.ErrorWriter.String())
}
}
func TestRefresh_backup(t *testing.T) {
state := testState()
statePath := testStateFile(t, state)

View File

@ -0,0 +1,7 @@
variable "should_ask" {}
provider "test" {}
resource "test_instance" "foo" {
ami = "${var.should_ask}"
}