242 lines
4.6 KiB
Go
242 lines
4.6 KiB
Go
![]() |
package command
|
||
|
|
||
|
import (
|
||
|
"os"
|
||
|
"testing"
|
||
|
|
||
|
"github.com/hashicorp/terraform/terraform"
|
||
|
"github.com/mitchellh/cli"
|
||
|
)
|
||
|
|
||
|
func TestTaint(t *testing.T) {
|
||
|
state := &terraform.State{
|
||
|
Modules: []*terraform.ModuleState{
|
||
|
&terraform.ModuleState{
|
||
|
Path: []string{"root"},
|
||
|
Resources: map[string]*terraform.ResourceState{
|
||
|
"test_instance.foo": &terraform.ResourceState{
|
||
|
Type: "test_instance",
|
||
|
Primary: &terraform.InstanceState{
|
||
|
ID: "bar",
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
}
|
||
|
statePath := testStateFile(t, state)
|
||
|
|
||
|
ui := new(cli.MockUi)
|
||
|
c := &TaintCommand{
|
||
|
Meta: Meta{
|
||
|
Ui: ui,
|
||
|
},
|
||
|
}
|
||
|
|
||
|
args := []string{
|
||
|
"-state", statePath,
|
||
|
"test_instance.foo",
|
||
|
}
|
||
|
if code := c.Run(args); code != 0 {
|
||
|
t.Fatalf("bad: %d\n\n%s", code, ui.ErrorWriter.String())
|
||
|
}
|
||
|
|
||
|
testStateOutput(t, statePath, testTaintStr)
|
||
|
}
|
||
|
|
||
|
func TestTaint_backup(t *testing.T) {
|
||
|
// Get a temp cwd
|
||
|
tmp, cwd := testCwd(t)
|
||
|
defer testFixCwd(t, tmp, cwd)
|
||
|
|
||
|
// Write the temp state
|
||
|
state := &terraform.State{
|
||
|
Modules: []*terraform.ModuleState{
|
||
|
&terraform.ModuleState{
|
||
|
Path: []string{"root"},
|
||
|
Resources: map[string]*terraform.ResourceState{
|
||
|
"test_instance.foo": &terraform.ResourceState{
|
||
|
Type: "test_instance",
|
||
|
Primary: &terraform.InstanceState{
|
||
|
ID: "bar",
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
}
|
||
|
path := testStateFileDefault(t, state)
|
||
|
|
||
|
ui := new(cli.MockUi)
|
||
|
c := &TaintCommand{
|
||
|
Meta: Meta{
|
||
|
Ui: ui,
|
||
|
},
|
||
|
}
|
||
|
|
||
|
args := []string{
|
||
|
"test_instance.foo",
|
||
|
}
|
||
|
if code := c.Run(args); code != 0 {
|
||
|
t.Fatalf("bad: %d\n\n%s", code, ui.ErrorWriter.String())
|
||
|
}
|
||
|
|
||
|
testStateOutput(t, path+".backup", testTaintDefaultStr)
|
||
|
testStateOutput(t, path, testTaintStr)
|
||
|
}
|
||
|
|
||
|
func TestTaint_backupDisable(t *testing.T) {
|
||
|
// Get a temp cwd
|
||
|
tmp, cwd := testCwd(t)
|
||
|
defer testFixCwd(t, tmp, cwd)
|
||
|
|
||
|
// Write the temp state
|
||
|
state := &terraform.State{
|
||
|
Modules: []*terraform.ModuleState{
|
||
|
&terraform.ModuleState{
|
||
|
Path: []string{"root"},
|
||
|
Resources: map[string]*terraform.ResourceState{
|
||
|
"test_instance.foo": &terraform.ResourceState{
|
||
|
Type: "test_instance",
|
||
|
Primary: &terraform.InstanceState{
|
||
|
ID: "bar",
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
}
|
||
|
path := testStateFileDefault(t, state)
|
||
|
|
||
|
ui := new(cli.MockUi)
|
||
|
c := &TaintCommand{
|
||
|
Meta: Meta{
|
||
|
Ui: ui,
|
||
|
},
|
||
|
}
|
||
|
|
||
|
args := []string{
|
||
|
"-backup", "-",
|
||
|
"test_instance.foo",
|
||
|
}
|
||
|
if code := c.Run(args); code != 0 {
|
||
|
t.Fatalf("bad: %d\n\n%s", code, ui.ErrorWriter.String())
|
||
|
}
|
||
|
|
||
|
if _, err := os.Stat(path + ".backup"); err == nil {
|
||
|
t.Fatal("backup path should not exist")
|
||
|
}
|
||
|
|
||
|
testStateOutput(t, path, testTaintStr)
|
||
|
}
|
||
|
|
||
|
func TestTaint_badState(t *testing.T) {
|
||
|
ui := new(cli.MockUi)
|
||
|
c := &TaintCommand{
|
||
|
Meta: Meta{
|
||
|
Ui: ui,
|
||
|
},
|
||
|
}
|
||
|
|
||
|
args := []string{
|
||
|
"-state", "i-should-not-exist-ever",
|
||
|
"foo",
|
||
|
}
|
||
|
if code := c.Run(args); code != 1 {
|
||
|
t.Fatalf("bad: %d\n\n%s", code, ui.ErrorWriter.String())
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestTaint_defaultState(t *testing.T) {
|
||
|
// Get a temp cwd
|
||
|
tmp, cwd := testCwd(t)
|
||
|
defer testFixCwd(t, tmp, cwd)
|
||
|
|
||
|
// Write the temp state
|
||
|
state := &terraform.State{
|
||
|
Modules: []*terraform.ModuleState{
|
||
|
&terraform.ModuleState{
|
||
|
Path: []string{"root"},
|
||
|
Resources: map[string]*terraform.ResourceState{
|
||
|
"test_instance.foo": &terraform.ResourceState{
|
||
|
Type: "test_instance",
|
||
|
Primary: &terraform.InstanceState{
|
||
|
ID: "bar",
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
}
|
||
|
path := testStateFileDefault(t, state)
|
||
|
|
||
|
ui := new(cli.MockUi)
|
||
|
c := &TaintCommand{
|
||
|
Meta: Meta{
|
||
|
Ui: ui,
|
||
|
},
|
||
|
}
|
||
|
|
||
|
args := []string{
|
||
|
"test_instance.foo",
|
||
|
}
|
||
|
if code := c.Run(args); code != 0 {
|
||
|
t.Fatalf("bad: %d\n\n%s", code, ui.ErrorWriter.String())
|
||
|
}
|
||
|
|
||
|
testStateOutput(t, path, testTaintStr)
|
||
|
}
|
||
|
|
||
|
func TestTaint_stateOut(t *testing.T) {
|
||
|
// Get a temp cwd
|
||
|
tmp, cwd := testCwd(t)
|
||
|
defer testFixCwd(t, tmp, cwd)
|
||
|
|
||
|
// Write the temp state
|
||
|
state := &terraform.State{
|
||
|
Modules: []*terraform.ModuleState{
|
||
|
&terraform.ModuleState{
|
||
|
Path: []string{"root"},
|
||
|
Resources: map[string]*terraform.ResourceState{
|
||
|
"test_instance.foo": &terraform.ResourceState{
|
||
|
Type: "test_instance",
|
||
|
Primary: &terraform.InstanceState{
|
||
|
ID: "bar",
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
}
|
||
|
path := testStateFileDefault(t, state)
|
||
|
|
||
|
ui := new(cli.MockUi)
|
||
|
c := &TaintCommand{
|
||
|
Meta: Meta{
|
||
|
Ui: ui,
|
||
|
},
|
||
|
}
|
||
|
|
||
|
args := []string{
|
||
|
"-state-out", "foo",
|
||
|
"test_instance.foo",
|
||
|
}
|
||
|
if code := c.Run(args); code != 0 {
|
||
|
t.Fatalf("bad: %d\n\n%s", code, ui.ErrorWriter.String())
|
||
|
}
|
||
|
|
||
|
testStateOutput(t, path, testTaintDefaultStr)
|
||
|
testStateOutput(t, "foo", testTaintStr)
|
||
|
}
|
||
|
|
||
|
const testTaintStr = `
|
||
|
test_instance.foo: (1 tainted)
|
||
|
ID = <not created>
|
||
|
Tainted ID 1 = bar
|
||
|
`
|
||
|
|
||
|
const testTaintDefaultStr = `
|
||
|
test_instance.foo:
|
||
|
ID = bar
|
||
|
`
|