package test import ( "reflect" "testing" "github.com/davecgh/go-spew/spew" "github.com/hashicorp/terraform/helper/schema" "github.com/hashicorp/terraform/terraform" ) func TestDiffApply_set(t *testing.T) { priorAttrs := map[string]string{ "id": "testID", "egress.#": "1", "egress.2129912301.cidr_blocks.#": "1", "egress.2129912301.cidr_blocks.0": "10.0.0.0/8", "egress.2129912301.description": "Egress description", "egress.2129912301.from_port": "80", "egress.2129912301.ipv6_cidr_blocks.#": "0", "egress.2129912301.prefix_list_ids.#": "0", "egress.2129912301.protocol": "tcp", "egress.2129912301.security_groups.#": "0", "egress.2129912301.self": "false", "egress.2129912301.to_port": "8000", } diff := &terraform.InstanceDiff{ Attributes: map[string]*terraform.ResourceAttrDiff{ "egress.2129912301.cidr_blocks.#": {Old: "1", New: "0", NewComputed: false, NewRemoved: false}, "egress.2129912301.cidr_blocks.0": {Old: "10.0.0.0/8", New: "", NewComputed: false, NewRemoved: true}, "egress.2129912301.description": {Old: "Egress description", New: "", NewComputed: false, NewRemoved: true}, "egress.2129912301.from_port": {Old: "80", New: "0", NewComputed: false, NewRemoved: true}, "egress.2129912301.ipv6_cidr_blocks.#": {Old: "0", New: "0", NewComputed: false, NewRemoved: false}, "egress.2129912301.prefix_list_ids.#": {Old: "0", New: "0", NewComputed: false, NewRemoved: false}, "egress.2129912301.protocol": {Old: "tcp", New: "", NewComputed: false, NewRemoved: true}, "egress.2129912301.security_groups.#": {Old: "0", New: "0", NewComputed: false, NewRemoved: false}, "egress.2129912301.self": {Old: "false", New: "false", NewComputed: false, NewRemoved: true}, "egress.2129912301.to_port": {Old: "8000", New: "0", NewComputed: false, NewRemoved: true}, "egress.746197026.cidr_blocks.#": {Old: "", New: "1", NewComputed: false, NewRemoved: false}, "egress.746197026.cidr_blocks.0": {Old: "", New: "10.0.0.0/8", NewComputed: false, NewRemoved: false}, "egress.746197026.description": {Old: "", New: "New egress description", NewComputed: false, NewRemoved: false}, "egress.746197026.from_port": {Old: "", New: "80", NewComputed: false, NewRemoved: false}, "egress.746197026.ipv6_cidr_blocks.#": {Old: "", New: "0", NewComputed: false, NewRemoved: false}, "egress.746197026.prefix_list_ids.#": {Old: "", New: "0", NewComputed: false, NewRemoved: false}, "egress.746197026.protocol": {Old: "", New: "tcp", NewComputed: false, NewRemoved: false, NewExtra: "tcp"}, "egress.746197026.security_groups.#": {Old: "", New: "0", NewComputed: false, NewRemoved: false}, "egress.746197026.self": {Old: "", New: "false", NewComputed: false, NewRemoved: false}, "egress.746197026.to_port": {Old: "", New: "8000", NewComputed: false, NewRemoved: false}, }, } resSchema := map[string]*schema.Schema{ "egress": { Type: schema.TypeSet, Optional: true, Computed: true, ConfigMode: schema.SchemaConfigModeAttr, Elem: &schema.Resource{ Schema: map[string]*schema.Schema{ "from_port": { Type: schema.TypeInt, Required: true, }, "to_port": { Type: schema.TypeInt, Required: true, }, "protocol": { Type: schema.TypeString, Required: true, }, "cidr_blocks": { Type: schema.TypeList, Optional: true, Elem: &schema.Schema{ Type: schema.TypeString, }, }, "ipv6_cidr_blocks": { Type: schema.TypeList, Optional: true, Elem: &schema.Schema{ Type: schema.TypeString, }, }, "prefix_list_ids": { Type: schema.TypeList, Optional: true, Elem: &schema.Schema{Type: schema.TypeString}, }, "security_groups": { Type: schema.TypeSet, Optional: true, Elem: &schema.Schema{Type: schema.TypeString}, Set: schema.HashString, }, "self": { Type: schema.TypeBool, Optional: true, Default: false, }, "description": { Type: schema.TypeString, Optional: true, }, }, }, }, } expected := map[string]string{ "egress.#": "1", "egress.746197026.cidr_blocks.#": "1", "egress.746197026.cidr_blocks.0": "10.0.0.0/8", "egress.746197026.description": "New egress description", "egress.746197026.from_port": "80", "egress.746197026.ipv6_cidr_blocks.#": "0", "egress.746197026.prefix_list_ids.#": "0", "egress.746197026.protocol": "tcp", "egress.746197026.security_groups.#": "0", "egress.746197026.self": "false", "egress.746197026.to_port": "8000", "id": "testID", } attrs, err := diff.Apply(priorAttrs, schema.LegacyResourceSchema(&schema.Resource{Schema: resSchema}).CoreConfigSchema()) if err != nil { t.Fatal(err) } if !reflect.DeepEqual(attrs, expected) { t.Fatalf("wrong result\ngot: %s\nwant: %s\n", spew.Sdump(attrs), spew.Sdump(expected)) } }