providers/aws: fixing faililng test

This commit is contained in:
Mitchell Hashimoto 2014-10-21 10:57:55 -07:00
parent 61811fce23
commit 81913c58fb
2 changed files with 22 additions and 5 deletions

View File

@ -5,6 +5,8 @@ import (
"testing"
"github.com/hashicorp/terraform/flatmap"
"github.com/hashicorp/terraform/helper/hashcode"
"github.com/hashicorp/terraform/helper/schema"
"github.com/mitchellh/goamz/ec2"
"github.com/mitchellh/goamz/elb"
)
@ -33,16 +35,20 @@ func testConf() map[string]string {
}
func Test_expandIPPerms(t *testing.T) {
hash := func(v interface{}) int {
return hashcode.String(v.(string))
}
expanded := []interface{}{
map[string]interface{}{
"protocol": "icmp",
"from_port": 1,
"to_port": -1,
"cidr_blocks": []interface{}{"0.0.0.0/0"},
"security_groups": []interface{}{
"security_groups": schema.NewSet(hash, []interface{}{
"sg-11111",
"foo/sg-22222",
},
}),
},
map[string]interface{}{
"protocol": "icmp",
@ -60,13 +66,13 @@ func Test_expandIPPerms(t *testing.T) {
ToPort: -1,
SourceIPs: []string{"0.0.0.0/0"},
SourceGroups: []ec2.UserSecurityGroup{
ec2.UserSecurityGroup{
Id: "sg-11111",
},
ec2.UserSecurityGroup{
OwnerId: "foo",
Id: "sg-22222",
},
ec2.UserSecurityGroup{
Id: "sg-11111",
},
},
},
ec2.IPPerm{

View File

@ -14,6 +14,17 @@ type Set struct {
once sync.Once
}
// NewSet is a convenience method for creating a new set with the given
// items.
func NewSet(f SchemaSetFunc, items []interface{}) *Set {
s := &Set{F: f}
for _, i := range items {
s.Add(i)
}
return s
}
// Add adds an item to the set if it isn't already in the set.
func (s *Set) Add(item interface{}) {
s.add(item)