50 lines
1.0 KiB
Go
50 lines
1.0 KiB
Go
|
package aws
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/hashicorp/terraform/terraform"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestAWSVpcMigrateState(t *testing.T) {
|
||
|
|
cases := map[string]struct {
|
||
|
|
StateVersion int
|
||
|
|
ID string
|
||
|
|
Attributes map[string]string
|
||
|
|
Expected string
|
||
|
|
Meta interface{}
|
||
|
|
}{
|
||
|
|
"v0_1": {
|
||
|
|
StateVersion: 0,
|
||
|
|
ID: "some_id",
|
||
|
|
Attributes: map[string]string{
|
||
|
|
"assign_generated_ipv6_cidr_block": "true",
|
||
|
|
},
|
||
|
|
Expected: "false",
|
||
|
|
},
|
||
|
|
"v0_1_without_value": {
|
||
|
|
StateVersion: 0,
|
||
|
|
ID: "some_id",
|
||
|
|
Attributes: map[string]string{},
|
||
|
|
Expected: "false",
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
for tn, tc := range cases {
|
||
|
|
is := &terraform.InstanceState{
|
||
|
|
ID: tc.ID,
|
||
|
|
Attributes: tc.Attributes,
|
||
|
|
}
|
||
|
|
is, err := resourceAwsVpcMigrateState(
|
||
|
|
tc.StateVersion, is, tc.Meta)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("bad: %s, err: %#v", tn, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if is.Attributes["assign_generated_ipv6_cidr_block"] != tc.Expected {
|
||
|
|
t.Fatalf("bad VPC Migrate: %s\n\n expected: %s", is.Attributes["assign_generated_ipv6_cidr_block"], tc.Expected)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|