Skip to content

Commit

Permalink
Merge pull request #4017 from hashicorp/b-vars-refresh
Browse files Browse the repository at this point in the history
core: Validate context after input of vars on refresh
  • Loading branch information
jen20 committed Nov 23, 2015
2 parents fd2c62b + 50b5e7c commit b044340
Show file tree
Hide file tree
Showing 3 changed files with 41 additions and 3 deletions.
8 changes: 5 additions & 3 deletions command/refresh.go
Expand Up @@ -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))
Expand Down
29 changes: 29 additions & 0 deletions command/refresh_test.go
Expand Up @@ -8,6 +8,7 @@ import (
"strings"
"testing"

"bytes"
"github.com/hashicorp/terraform/terraform"
"github.com/mitchellh/cli"
)
Expand Down Expand Up @@ -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)
Expand Down
7 changes: 7 additions & 0 deletions command/test-fixtures/refresh-unset-var/main.tf
@@ -0,0 +1,7 @@
variable "should_ask" {}

provider "test" {}

resource "test_instance" "foo" {
ami = "${var.should_ask}"
}

0 comments on commit b044340

Please sign in to comment.