123 lines
3.0 KiB
Go
123 lines
3.0 KiB
Go
![]() |
package aws
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"reflect"
|
||
|
"testing"
|
||
|
|
||
|
"github.com/hashicorp/aws-sdk-go/gen/autoscaling"
|
||
|
"github.com/hashicorp/terraform/helper/resource"
|
||
|
"github.com/hashicorp/terraform/terraform"
|
||
|
)
|
||
|
|
||
|
func TestDiffAutoscalingTags(t *testing.T) {
|
||
|
cases := []struct {
|
||
|
Old, New map[string]interface{}
|
||
|
Create, Remove map[string]interface{}
|
||
|
}{
|
||
|
// Basic add/remove
|
||
|
{
|
||
|
Old: map[string]interface{}{
|
||
|
"Name": map[string]interface{}{
|
||
|
"value": "bar",
|
||
|
"propagate_at_launch": true,
|
||
|
},
|
||
|
},
|
||
|
New: map[string]interface{}{
|
||
|
"DifferentTag": map[string]interface{}{
|
||
|
"value": "baz",
|
||
|
"propagate_at_launch": true,
|
||
|
},
|
||
|
},
|
||
|
Create: map[string]interface{}{
|
||
|
"DifferentTag": map[string]interface{}{
|
||
|
"value": "baz",
|
||
|
"propagate_at_launch": true,
|
||
|
},
|
||
|
},
|
||
|
Remove: map[string]interface{}{
|
||
|
"Name": map[string]interface{}{
|
||
|
"value": "bar",
|
||
|
"propagate_at_launch": true,
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
|
||
|
// Modify
|
||
|
{
|
||
|
Old: map[string]interface{}{
|
||
|
"Name": map[string]interface{}{
|
||
|
"value": "bar",
|
||
|
"propagate_at_launch": true,
|
||
|
},
|
||
|
},
|
||
|
New: map[string]interface{}{
|
||
|
"Name": map[string]interface{}{
|
||
|
"value": "baz",
|
||
|
"propagate_at_launch": false,
|
||
|
},
|
||
|
},
|
||
|
Create: map[string]interface{}{
|
||
|
"Name": map[string]interface{}{
|
||
|
"value": "baz",
|
||
|
"propagate_at_launch": false,
|
||
|
},
|
||
|
},
|
||
|
Remove: map[string]interface{}{
|
||
|
"Name": map[string]interface{}{
|
||
|
"value": "bar",
|
||
|
"propagate_at_launch": true,
|
||
|
},
|
||
|
},
|
||
|
},
|
||
|
}
|
||
|
|
||
|
var resourceID = "sample"
|
||
|
|
||
|
for i, tc := range cases {
|
||
|
awsTagsOld := autoscalingTagsFromMap(tc.Old, resourceID)
|
||
|
awsTagsNew := autoscalingTagsFromMap(tc.New, resourceID)
|
||
|
|
||
|
c, r := diffAutoscalingTags(awsTagsOld, awsTagsNew, resourceID)
|
||
|
|
||
|
cm := autoscalingTagsToMap(c)
|
||
|
rm := autoscalingTagsToMap(r)
|
||
|
if !reflect.DeepEqual(cm, tc.Create) {
|
||
|
t.Fatalf("%d: bad create: \n%#v\n%#v", i, cm, tc.Create)
|
||
|
}
|
||
|
if !reflect.DeepEqual(rm, tc.Remove) {
|
||
|
t.Fatalf("%d: bad remove: \n%#v\n%#v", i, rm, tc.Remove)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// testAccCheckTags can be used to check the tags on a resource.
|
||
|
func testAccCheckAutoscalingTags(
|
||
|
ts *[]autoscaling.TagDescription, key string, expected map[string]interface{}) resource.TestCheckFunc {
|
||
|
return func(s *terraform.State) error {
|
||
|
m := autoscalingTagDescriptionsToMap(*ts)
|
||
|
v, ok := m[key]
|
||
|
if !ok {
|
||
|
return fmt.Errorf("Missing tag: %s", key)
|
||
|
}
|
||
|
|
||
|
if v["value"] != expected["value"].(string) ||
|
||
|
v["propagate_at_launch"] != expected["propagate_at_launch"].(bool) {
|
||
|
return fmt.Errorf("%s: bad value: %s", key, v)
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func testAccCheckAutoscalingTagNotExists(ts *[]autoscaling.TagDescription, key string) resource.TestCheckFunc {
|
||
|
return func(s *terraform.State) error {
|
||
|
m := autoscalingTagDescriptionsToMap(*ts)
|
||
|
if _, ok := m[key]; ok {
|
||
|
return fmt.Errorf("Tag exists when it should not: %s", key)
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
}
|