109 lines
2.0 KiB
Go
109 lines
2.0 KiB
Go
|
package command
|
||
|
|
|
||
|
|
import (
|
||
|
|
"path/filepath"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/hashicorp/terraform/terraform"
|
||
|
|
"github.com/mitchellh/cli"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestStateRm(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",
|
||
|
|
Attributes: map[string]string{
|
||
|
|
"foo": "value",
|
||
|
|
"bar": "value",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
|
||
|
|
"test_instance.bar": &terraform.ResourceState{
|
||
|
|
Type: "test_instance",
|
||
|
|
Primary: &terraform.InstanceState{
|
||
|
|
ID: "foo",
|
||
|
|
Attributes: map[string]string{
|
||
|
|
"foo": "value",
|
||
|
|
"bar": "value",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
statePath := testStateFile(t, state)
|
||
|
|
|
||
|
|
p := testProvider()
|
||
|
|
ui := new(cli.MockUi)
|
||
|
|
c := &StateRmCommand{
|
||
|
|
Meta: Meta{
|
||
|
|
ContextOpts: testCtxConfig(p),
|
||
|
|
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())
|
||
|
|
}
|
||
|
|
|
||
|
|
// Test it is correct
|
||
|
|
testStateOutput(t, statePath, testStateRmOutput)
|
||
|
|
|
||
|
|
// Test we have backups
|
||
|
|
backups := testStateBackups(t, filepath.Dir(statePath))
|
||
|
|
if len(backups) != 1 {
|
||
|
|
t.Fatalf("bad: %#v", backups)
|
||
|
|
}
|
||
|
|
testStateOutput(t, backups[0], testStateRmOutputOriginal)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestStateRm_noState(t *testing.T) {
|
||
|
|
tmp, cwd := testCwd(t)
|
||
|
|
defer testFixCwd(t, tmp, cwd)
|
||
|
|
|
||
|
|
p := testProvider()
|
||
|
|
ui := new(cli.MockUi)
|
||
|
|
c := &StateRmCommand{
|
||
|
|
Meta: Meta{
|
||
|
|
ContextOpts: testCtxConfig(p),
|
||
|
|
Ui: ui,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
args := []string{}
|
||
|
|
if code := c.Run(args); code != 1 {
|
||
|
|
t.Fatalf("bad: %d\n\n%s", code, ui.ErrorWriter.String())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
const testStateRmOutputOriginal = `
|
||
|
|
test_instance.bar:
|
||
|
|
ID = foo
|
||
|
|
bar = value
|
||
|
|
foo = value
|
||
|
|
test_instance.foo:
|
||
|
|
ID = bar
|
||
|
|
bar = value
|
||
|
|
foo = value
|
||
|
|
`
|
||
|
|
|
||
|
|
const testStateRmOutput = `
|
||
|
|
test_instance.bar:
|
||
|
|
ID = foo
|
||
|
|
bar = value
|
||
|
|
foo = value
|
||
|
|
`
|