package aws import ( "reflect" "strings" "testing" "github.com/aws/aws-sdk-go/aws" "github.com/aws/aws-sdk-go/service/apigateway" "github.com/aws/aws-sdk-go/service/autoscaling" "github.com/aws/aws-sdk-go/service/ec2" "github.com/aws/aws-sdk-go/service/elasticache" "github.com/aws/aws-sdk-go/service/elb" "github.com/aws/aws-sdk-go/service/kinesis" "github.com/aws/aws-sdk-go/service/rds" "github.com/aws/aws-sdk-go/service/redshift" "github.com/aws/aws-sdk-go/service/route53" "github.com/hashicorp/terraform/flatmap" "github.com/hashicorp/terraform/helper/schema" ) // Returns test configuration func testConf() map[string]string { return map[string]string{ "listener.#": "1", "listener.0.lb_port": "80", "listener.0.lb_protocol": "http", "listener.0.instance_port": "8000", "listener.0.instance_protocol": "http", "availability_zones.#": "2", "availability_zones.0": "us-east-1a", "availability_zones.1": "us-east-1b", "ingress.#": "1", "ingress.0.protocol": "icmp", "ingress.0.from_port": "1", "ingress.0.to_port": "-1", "ingress.0.cidr_blocks.#": "1", "ingress.0.cidr_blocks.0": "0.0.0.0/0", "ingress.0.security_groups.#": "2", "ingress.0.security_groups.0": "sg-11111", "ingress.0.security_groups.1": "foo/sg-22222", } } func TestExpandIPPerms(t *testing.T) { hash := schema.HashString expanded := []interface{}{ map[string]interface{}{ "protocol": "icmp", "from_port": 1, "to_port": -1, "cidr_blocks": []interface{}{"0.0.0.0/0"}, "security_groups": schema.NewSet(hash, []interface{}{ "sg-11111", "foo/sg-22222", }), }, map[string]interface{}{ "protocol": "icmp", "from_port": 1, "to_port": -1, "self": true, }, } group := &ec2.SecurityGroup{ GroupId: aws.String("foo"), VpcId: aws.String("bar"), } perms, err := expandIPPerms(group, expanded) if err != nil { t.Fatalf("error expanding perms: %v", err) } expected := []ec2.IpPermission{ ec2.IpPermission{ IpProtocol: aws.String("icmp"), FromPort: aws.Int64(int64(1)), ToPort: aws.Int64(int64(-1)), IpRanges: []*ec2.IpRange{&ec2.IpRange{CidrIp: aws.String("0.0.0.0/0")}}, UserIdGroupPairs: []*ec2.UserIdGroupPair{ &ec2.UserIdGroupPair{ UserId: aws.String("foo"), GroupId: aws.String("sg-22222"), }, &ec2.UserIdGroupPair{ GroupId: aws.String("sg-11111"), }, }, }, ec2.IpPermission{ IpProtocol: aws.String("icmp"), FromPort: aws.Int64(int64(1)), ToPort: aws.Int64(int64(-1)), UserIdGroupPairs: []*ec2.UserIdGroupPair{ &ec2.UserIdGroupPair{ GroupId: aws.String("foo"), }, }, }, } exp := expected[0] perm := perms[0] if *exp.FromPort != *perm.FromPort { t.Fatalf( "Got:\n\n%#v\n\nExpected:\n\n%#v\n", *perm.FromPort, *exp.FromPort) } if *exp.IpRanges[0].CidrIp != *perm.IpRanges[0].CidrIp { t.Fatalf( "Got:\n\n%#v\n\nExpected:\n\n%#v\n", *perm.IpRanges[0].CidrIp, *exp.IpRanges[0].CidrIp) } if *exp.UserIdGroupPairs[0].UserId != *perm.UserIdGroupPairs[0].UserId { t.Fatalf( "Got:\n\n%#v\n\nExpected:\n\n%#v\n", *perm.UserIdGroupPairs[0].UserId, *exp.UserIdGroupPairs[0].UserId) } if *exp.UserIdGroupPairs[0].GroupId != *perm.UserIdGroupPairs[0].GroupId { t.Fatalf( "Got:\n\n%#v\n\nExpected:\n\n%#v\n", *perm.UserIdGroupPairs[0].GroupId, *exp.UserIdGroupPairs[0].GroupId) } if *exp.UserIdGroupPairs[1].GroupId != *perm.UserIdGroupPairs[1].GroupId { t.Fatalf( "Got:\n\n%#v\n\nExpected:\n\n%#v\n", *perm.UserIdGroupPairs[1].GroupId, *exp.UserIdGroupPairs[1].GroupId) } exp = expected[1] perm = perms[1] if *exp.UserIdGroupPairs[0].GroupId != *perm.UserIdGroupPairs[0].GroupId { t.Fatalf( "Got:\n\n%#v\n\nExpected:\n\n%#v\n", *perm.UserIdGroupPairs[0].GroupId, *exp.UserIdGroupPairs[0].GroupId) } } func TestExpandIPPerms_NegOneProtocol(t *testing.T) { hash := schema.HashString expanded := []interface{}{ map[string]interface{}{ "protocol": "-1", "from_port": 0, "to_port": 0, "cidr_blocks": []interface{}{"0.0.0.0/0"}, "security_groups": schema.NewSet(hash, []interface{}{ "sg-11111", "foo/sg-22222", }), }, } group := &ec2.SecurityGroup{ GroupId: aws.String("foo"), VpcId: aws.String("bar"), } perms, err := expandIPPerms(group, expanded) if err != nil { t.Fatalf("error expanding perms: %v", err) } expected := []ec2.IpPermission{ ec2.IpPermission{ IpProtocol: aws.String("-1"), FromPort: aws.Int64(int64(0)), ToPort: aws.Int64(int64(0)), IpRanges: []*ec2.IpRange{&ec2.IpRange{CidrIp: aws.String("0.0.0.0/0")}}, UserIdGroupPairs: []*ec2.UserIdGroupPair{ &ec2.UserIdGroupPair{ UserId: aws.String("foo"), GroupId: aws.String("sg-22222"), }, &ec2.UserIdGroupPair{ GroupId: aws.String("sg-11111"), }, }, }, } exp := expected[0] perm := perms[0] if *exp.FromPort != *perm.FromPort { t.Fatalf( "Got:\n\n%#v\n\nExpected:\n\n%#v\n", *perm.FromPort, *exp.FromPort) } if *exp.IpRanges[0].CidrIp != *perm.IpRanges[0].CidrIp { t.Fatalf( "Got:\n\n%#v\n\nExpected:\n\n%#v\n", *perm.IpRanges[0].CidrIp, *exp.IpRanges[0].CidrIp) } if *exp.UserIdGroupPairs[0].UserId != *perm.UserIdGroupPairs[0].UserId { t.Fatalf( "Got:\n\n%#v\n\nExpected:\n\n%#v\n", *perm.UserIdGroupPairs[0].UserId, *exp.UserIdGroupPairs[0].UserId) } // Now test the error case. This *should* error when either from_port // or to_port is not zero, but protocol is "-1". errorCase := []interface{}{ map[string]interface{}{ "protocol": "-1", "from_port": 0, "to_port": 65535, "cidr_blocks": []interface{}{"0.0.0.0/0"}, "security_groups": schema.NewSet(hash, []interface{}{ "sg-11111", "foo/sg-22222", }), }, } securityGroups := &ec2.SecurityGroup{ GroupId: aws.String("foo"), VpcId: aws.String("bar"), } _, expandErr := expandIPPerms(securityGroups, errorCase) if expandErr == nil { t.Fatal("expandIPPerms should have errored!") } } func TestExpandIPPerms_nonVPC(t *testing.T) { hash := schema.HashString expanded := []interface{}{ map[string]interface{}{ "protocol": "icmp", "from_port": 1, "to_port": -1, "cidr_blocks": []interface{}{"0.0.0.0/0"}, "security_groups": schema.NewSet(hash, []interface{}{ "sg-11111", "foo/sg-22222", }), }, map[string]interface{}{ "protocol": "icmp", "from_port": 1, "to_port": -1, "self": true, }, } group := &ec2.SecurityGroup{ GroupName: aws.String("foo"), } perms, err := expandIPPerms(group, expanded) if err != nil { t.Fatalf("error expanding perms: %v", err) } expected := []ec2.IpPermission{ ec2.IpPermission{ IpProtocol: aws.String("icmp"), FromPort: aws.Int64(int64(1)), ToPort: aws.Int64(int64(-1)), IpRanges: []*ec2.IpRange{&ec2.IpRange{CidrIp: aws.String("0.0.0.0/0")}}, UserIdGroupPairs: []*ec2.UserIdGroupPair{ &ec2.UserIdGroupPair{ GroupName: aws.String("sg-22222"), }, &ec2.UserIdGroupPair{ GroupName: aws.String("sg-11111"), }, }, }, ec2.IpPermission{ IpProtocol: aws.String("icmp"), FromPort: aws.Int64(int64(1)), ToPort: aws.Int64(int64(-1)), UserIdGroupPairs: []*ec2.UserIdGroupPair{ &ec2.UserIdGroupPair{ GroupName: aws.String("foo"), }, }, }, } exp := expected[0] perm := perms[0] if *exp.FromPort != *perm.FromPort { t.Fatalf( "Got:\n\n%#v\n\nExpected:\n\n%#v\n", *perm.FromPort, *exp.FromPort) } if *exp.IpRanges[0].CidrIp != *perm.IpRanges[0].CidrIp { t.Fatalf( "Got:\n\n%#v\n\nExpected:\n\n%#v\n", *perm.IpRanges[0].CidrIp, *exp.IpRanges[0].CidrIp) } if *exp.UserIdGroupPairs[0].GroupName != *perm.UserIdGroupPairs[0].GroupName { t.Fatalf( "Got:\n\n%#v\n\nExpected:\n\n%#v\n", *perm.UserIdGroupPairs[0].GroupName, *exp.UserIdGroupPairs[0].GroupName) } if *exp.UserIdGroupPairs[1].GroupName != *perm.UserIdGroupPairs[1].GroupName { t.Fatalf( "Got:\n\n%#v\n\nExpected:\n\n%#v\n", *perm.UserIdGroupPairs[1].GroupName, *exp.UserIdGroupPairs[1].GroupName) } exp = expected[1] perm = perms[1] if *exp.UserIdGroupPairs[0].GroupName != *perm.UserIdGroupPairs[0].GroupName { t.Fatalf( "Got:\n\n%#v\n\nExpected:\n\n%#v\n", *perm.UserIdGroupPairs[0].GroupName, *exp.UserIdGroupPairs[0].GroupName) } } func TestExpandListeners(t *testing.T) { expanded := []interface{}{ map[string]interface{}{ "instance_port": 8000, "lb_port": 80, "instance_protocol": "http", "lb_protocol": "http", }, map[string]interface{}{ "instance_port": 8000, "lb_port": 80, "instance_protocol": "https", "lb_protocol": "https", "ssl_certificate_id": "something", }, } listeners, err := expandListeners(expanded) if err != nil { t.Fatalf("bad: %#v", err) } expected := &elb.Listener{ InstancePort: aws.Int64(int64(8000)), LoadBalancerPort: aws.Int64(int64(80)), InstanceProtocol: aws.String("http"), Protocol: aws.String("http"), } if !reflect.DeepEqual(listeners[0], expected) { t.Fatalf( "Got:\n\n%#v\n\nExpected:\n\n%#v\n", listeners[0], expected) } } // this test should produce an error from expandlisteners on an invalid // combination func TestExpandListeners_invalid(t *testing.T) { expanded := []interface{}{ map[string]interface{}{ "instance_port": 8000, "lb_port": 80, "instance_protocol": "http", "lb_protocol": "http", "ssl_certificate_id": "something", }, } _, err := expandListeners(expanded) if err != nil { // Check the error we got if !strings.Contains(err.Error(), "ssl_certificate_id may be set only when protocol") { t.Fatalf("Got error in TestExpandListeners_invalid, but not what we expected: %s", err) } } if err == nil { t.Fatalf("Expected TestExpandListeners_invalid to fail, but passed") } } func TestFlattenHealthCheck(t *testing.T) { cases := []struct { Input *elb.HealthCheck Output []map[string]interface{} }{ { Input: &elb.HealthCheck{ UnhealthyThreshold: aws.Int64(int64(10)), HealthyThreshold: aws.Int64(int64(10)), Target: aws.String("HTTP:80/"), Timeout: aws.Int64(int64(30)), Interval: aws.Int64(int64(30)), }, Output: []map[string]interface{}{ map[string]interface{}{ "unhealthy_threshold": int64(10), "healthy_threshold": int64(10), "target": "HTTP:80/", "timeout": int64(30), "interval": int64(30), }, }, }, } for _, tc := range cases { output := flattenHealthCheck(tc.Input) if !reflect.DeepEqual(output, tc.Output) { t.Fatalf("Got:\n\n%#v\n\nExpected:\n\n%#v", output, tc.Output) } } } func TestExpandStringList(t *testing.T) { expanded := flatmap.Expand(testConf(), "availability_zones").([]interface{}) stringList := expandStringList(expanded) expected := []*string{ aws.String("us-east-1a"), aws.String("us-east-1b"), } if !reflect.DeepEqual(stringList, expected) { t.Fatalf( "Got:\n\n%#v\n\nExpected:\n\n%#v\n", stringList, expected) } } func TestExpandParameters(t *testing.T) { expanded := []interface{}{ map[string]interface{}{ "name": "character_set_client", "value": "utf8", "apply_method": "immediate", }, } parameters, err := expandParameters(expanded) if err != nil { t.Fatalf("bad: %#v", err) } expected := &rds.Parameter{ ParameterName: aws.String("character_set_client"), ParameterValue: aws.String("utf8"), ApplyMethod: aws.String("immediate"), } if !reflect.DeepEqual(parameters[0], expected) { t.Fatalf( "Got:\n\n%#v\n\nExpected:\n\n%#v\n", parameters[0], expected) } } func TestExpandRedshiftParameters(t *testing.T) { expanded := []interface{}{ map[string]interface{}{ "name": "character_set_client", "value": "utf8", }, } parameters, err := expandRedshiftParameters(expanded) if err != nil { t.Fatalf("bad: %#v", err) } expected := &redshift.Parameter{ ParameterName: aws.String("character_set_client"), ParameterValue: aws.String("utf8"), } if !reflect.DeepEqual(parameters[0], expected) { t.Fatalf( "Got:\n\n%#v\n\nExpected:\n\n%#v\n", parameters[0], expected) } } func TestExpandElasticacheParameters(t *testing.T) { expanded := []interface{}{ map[string]interface{}{ "name": "activerehashing", "value": "yes", "apply_method": "immediate", }, } parameters, err := expandElastiCacheParameters(expanded) if err != nil { t.Fatalf("bad: %#v", err) } expected := &elasticache.ParameterNameValue{ ParameterName: aws.String("activerehashing"), ParameterValue: aws.String("yes"), } if !reflect.DeepEqual(parameters[0], expected) { t.Fatalf( "Got:\n\n%#v\n\nExpected:\n\n%#v\n", parameters[0], expected) } } func TestExpandStepAdjustments(t *testing.T) { expanded := []interface{}{ map[string]interface{}{ "metric_interval_lower_bound": "1.0", "metric_interval_upper_bound": "2.0", "scaling_adjustment": 1, }, } parameters, err := expandStepAdjustments(expanded) if err != nil { t.Fatalf("bad: %#v", err) } expected := &autoscaling.StepAdjustment{ MetricIntervalLowerBound: aws.Float64(1.0), MetricIntervalUpperBound: aws.Float64(2.0), ScalingAdjustment: aws.Int64(int64(1)), } if !reflect.DeepEqual(parameters[0], expected) { t.Fatalf( "Got:\n\n%#v\n\nExpected:\n\n%#v\n", parameters[0], expected) } } func TestFlattenParameters(t *testing.T) { cases := []struct { Input []*rds.Parameter Output []map[string]interface{} }{ { Input: []*rds.Parameter{ &rds.Parameter{ ParameterName: aws.String("character_set_client"), ParameterValue: aws.String("utf8"), }, }, Output: []map[string]interface{}{ map[string]interface{}{ "name": "character_set_client", "value": "utf8", }, }, }, } for _, tc := range cases { output := flattenParameters(tc.Input) if !reflect.DeepEqual(output, tc.Output) { t.Fatalf("Got:\n\n%#v\n\nExpected:\n\n%#v", output, tc.Output) } } } func TestFlattenRedshiftParameters(t *testing.T) { cases := []struct { Input []*redshift.Parameter Output []map[string]interface{} }{ { Input: []*redshift.Parameter{ &redshift.Parameter{ ParameterName: aws.String("character_set_client"), ParameterValue: aws.String("utf8"), }, }, Output: []map[string]interface{}{ map[string]interface{}{ "name": "character_set_client", "value": "utf8", }, }, }, } for _, tc := range cases { output := flattenRedshiftParameters(tc.Input) if !reflect.DeepEqual(output, tc.Output) { t.Fatalf("Got:\n\n%#v\n\nExpected:\n\n%#v", output, tc.Output) } } } func TestFlattenElasticacheParameters(t *testing.T) { cases := []struct { Input []*elasticache.Parameter Output []map[string]interface{} }{ { Input: []*elasticache.Parameter{ &elasticache.Parameter{ ParameterName: aws.String("activerehashing"), ParameterValue: aws.String("yes"), }, }, Output: []map[string]interface{}{ map[string]interface{}{ "name": "activerehashing", "value": "yes", }, }, }, } for _, tc := range cases { output := flattenElastiCacheParameters(tc.Input) if !reflect.DeepEqual(output, tc.Output) { t.Fatalf("Got:\n\n%#v\n\nExpected:\n\n%#v", output, tc.Output) } } } func TestExpandInstanceString(t *testing.T) { expected := []*elb.Instance{ &elb.Instance{InstanceId: aws.String("test-one")}, &elb.Instance{InstanceId: aws.String("test-two")}, } ids := []interface{}{ "test-one", "test-two", } expanded := expandInstanceString(ids) if !reflect.DeepEqual(expanded, expected) { t.Fatalf("Expand Instance String output did not match.\nGot:\n%#v\n\nexpected:\n%#v", expanded, expected) } } func TestFlattenNetworkInterfacesPrivateIPAddresses(t *testing.T) { expanded := []*ec2.NetworkInterfacePrivateIpAddress{ &ec2.NetworkInterfacePrivateIpAddress{PrivateIpAddress: aws.String("192.168.0.1")}, &ec2.NetworkInterfacePrivateIpAddress{PrivateIpAddress: aws.String("192.168.0.2")}, } result := flattenNetworkInterfacesPrivateIPAddresses(expanded) if result == nil { t.Fatal("result was nil") } if len(result) != 2 { t.Fatalf("expected result had %d elements, but got %d", 2, len(result)) } if result[0] != "192.168.0.1" { t.Fatalf("expected ip to be 192.168.0.1, but was %s", result[0]) } if result[1] != "192.168.0.2" { t.Fatalf("expected ip to be 192.168.0.2, but was %s", result[1]) } } func TestFlattenGroupIdentifiers(t *testing.T) { expanded := []*ec2.GroupIdentifier{ &ec2.GroupIdentifier{GroupId: aws.String("sg-001")}, &ec2.GroupIdentifier{GroupId: aws.String("sg-002")}, } result := flattenGroupIdentifiers(expanded) if len(result) != 2 { t.Fatalf("expected result had %d elements, but got %d", 2, len(result)) } if result[0] != "sg-001" { t.Fatalf("expected id to be sg-001, but was %s", result[0]) } if result[1] != "sg-002" { t.Fatalf("expected id to be sg-002, but was %s", result[1]) } } func TestExpandPrivateIPAddresses(t *testing.T) { ip1 := "192.168.0.1" ip2 := "192.168.0.2" flattened := []interface{}{ ip1, ip2, } result := expandPrivateIPAddresses(flattened) if len(result) != 2 { t.Fatalf("expected result had %d elements, but got %d", 2, len(result)) } if *result[0].PrivateIpAddress != "192.168.0.1" || !*result[0].Primary { t.Fatalf("expected ip to be 192.168.0.1 and Primary, but got %v, %t", *result[0].PrivateIpAddress, *result[0].Primary) } if *result[1].PrivateIpAddress != "192.168.0.2" || *result[1].Primary { t.Fatalf("expected ip to be 192.168.0.2 and not Primary, but got %v, %t", *result[1].PrivateIpAddress, *result[1].Primary) } } func TestFlattenAttachment(t *testing.T) { expanded := &ec2.NetworkInterfaceAttachment{ InstanceId: aws.String("i-00001"), DeviceIndex: aws.Int64(int64(1)), AttachmentId: aws.String("at-002"), } result := flattenAttachment(expanded) if result == nil { t.Fatal("expected result to have value, but got nil") } if result["instance"] != "i-00001" { t.Fatalf("expected instance to be i-00001, but got %s", result["instance"]) } if result["device_index"] != int64(1) { t.Fatalf("expected device_index to be 1, but got %d", result["device_index"]) } if result["attachment_id"] != "at-002" { t.Fatalf("expected attachment_id to be at-002, but got %s", result["attachment_id"]) } } func TestFlattenAttachmentWhenNoInstanceId(t *testing.T) { expanded := &ec2.NetworkInterfaceAttachment{ DeviceIndex: aws.Int64(int64(1)), AttachmentId: aws.String("at-002"), } result := flattenAttachment(expanded) if result == nil { t.Fatal("expected result to have value, but got nil") } if result["instance"] != nil { t.Fatalf("expected instance to be nil, but got %s", result["instance"]) } } func TestFlattenStepAdjustments(t *testing.T) { expanded := []*autoscaling.StepAdjustment{ &autoscaling.StepAdjustment{ MetricIntervalLowerBound: aws.Float64(1.0), MetricIntervalUpperBound: aws.Float64(2.0), ScalingAdjustment: aws.Int64(int64(1)), }, } result := flattenStepAdjustments(expanded)[0] if result == nil { t.Fatal("expected result to have value, but got nil") } if result["metric_interval_lower_bound"] != float64(1.0) { t.Fatalf("expected metric_interval_lower_bound to be 1.0, but got %d", result["metric_interval_lower_bound"]) } if result["metric_interval_upper_bound"] != float64(2.0) { t.Fatalf("expected metric_interval_upper_bound to be 1.0, but got %d", result["metric_interval_upper_bound"]) } if result["scaling_adjustment"] != int64(1) { t.Fatalf("expected scaling_adjustment to be 1, but got %d", result["scaling_adjustment"]) } } func TestFlattenResourceRecords(t *testing.T) { expanded := []*route53.ResourceRecord{ &route53.ResourceRecord{ Value: aws.String("127.0.0.1"), }, &route53.ResourceRecord{ Value: aws.String("127.0.0.3"), }, } result := flattenResourceRecords(expanded) if result == nil { t.Fatal("expected result to have value, but got nil") } if len(result) != 2 { t.Fatal("expected result to have value, but got nil") } } func TestFlattenAsgEnabledMetrics(t *testing.T) { expanded := []*autoscaling.EnabledMetric{ &autoscaling.EnabledMetric{Granularity: aws.String("1Minute"), Metric: aws.String("GroupTotalInstances")}, &autoscaling.EnabledMetric{Granularity: aws.String("1Minute"), Metric: aws.String("GroupMaxSize")}, } result := flattenAsgEnabledMetrics(expanded) if len(result) != 2 { t.Fatalf("expected result had %d elements, but got %d", 2, len(result)) } if result[0] != "GroupTotalInstances" { t.Fatalf("expected id to be GroupTotalInstances, but was %s", result[0]) } if result[1] != "GroupMaxSize" { t.Fatalf("expected id to be GroupMaxSize, but was %s", result[1]) } } func TestFlattenKinesisShardLevelMetrics(t *testing.T) { expanded := []*kinesis.EnhancedMetrics{ &kinesis.EnhancedMetrics{ ShardLevelMetrics: []*string{ aws.String("IncomingBytes"), aws.String("IncomingRecords"), }, }, } result := flattenKinesisShardLevelMetrics(expanded) if len(result) != 2 { t.Fatalf("expected result had %d elements, but got %d", 2, len(result)) } if result[0] != "IncomingBytes" { t.Fatalf("expected element 0 to be IncomingBytes, but was %s", result[0]) } if result[1] != "IncomingRecords" { t.Fatalf("expected element 0 to be IncomingRecords, but was %s", result[1]) } } func TestFlattenSecurityGroups(t *testing.T) { cases := []struct { ownerId *string pairs []*ec2.UserIdGroupPair expected []*ec2.GroupIdentifier }{ // simple, no user id included (we ignore it mostly) { ownerId: aws.String("user1234"), pairs: []*ec2.UserIdGroupPair{ &ec2.UserIdGroupPair{ GroupId: aws.String("sg-12345"), }, }, expected: []*ec2.GroupIdentifier{ &ec2.GroupIdentifier{ GroupId: aws.String("sg-12345"), }, }, }, // include the owner id, but keep it consitent with the same account. Tests // EC2 classic situation { ownerId: aws.String("user1234"), pairs: []*ec2.UserIdGroupPair{ &ec2.UserIdGroupPair{ GroupId: aws.String("sg-12345"), UserId: aws.String("user1234"), }, }, expected: []*ec2.GroupIdentifier{ &ec2.GroupIdentifier{ GroupId: aws.String("sg-12345"), }, }, }, // include the owner id, but from a different account. This is reflects // EC2 Classic when refering to groups by name { ownerId: aws.String("user1234"), pairs: []*ec2.UserIdGroupPair{ &ec2.UserIdGroupPair{ GroupId: aws.String("sg-12345"), GroupName: aws.String("somegroup"), // GroupName is only included in Classic UserId: aws.String("user4321"), }, }, expected: []*ec2.GroupIdentifier{ &ec2.GroupIdentifier{ GroupId: aws.String("sg-12345"), GroupName: aws.String("user4321/somegroup"), }, }, }, // include the owner id, but from a different account. This reflects in // EC2 VPC when refering to groups by id { ownerId: aws.String("user1234"), pairs: []*ec2.UserIdGroupPair{ &ec2.UserIdGroupPair{ GroupId: aws.String("sg-12345"), UserId: aws.String("user4321"), }, }, expected: []*ec2.GroupIdentifier{ &ec2.GroupIdentifier{ GroupId: aws.String("user4321/sg-12345"), }, }, }, } for _, c := range cases { out := flattenSecurityGroups(c.pairs, c.ownerId) if !reflect.DeepEqual(out, c.expected) { t.Fatalf("Error matching output and expected: %#v vs %#v", out, c.expected) } } } func TestFlattenApiGatewayThrottleSettings(t *testing.T) { expectedBurstLimit := int64(140) expectedRateLimit := 120.0 ts := &apigateway.ThrottleSettings{ BurstLimit: aws.Int64(expectedBurstLimit), RateLimit: aws.Float64(expectedRateLimit), } result := flattenApiGatewayThrottleSettings(ts) if len(result) != 1 { t.Fatalf("Expected map to have exactly 1 element, got %d", len(result)) } burstLimit, ok := result[0]["burst_limit"] if !ok { t.Fatal("Expected 'burst_limit' key in the map") } burstLimitInt, ok := burstLimit.(int64) if !ok { t.Fatal("Expected 'burst_limit' to be int") } if burstLimitInt != expectedBurstLimit { t.Fatalf("Expected 'burst_limit' to equal %d, got %d", expectedBurstLimit, burstLimitInt) } rateLimit, ok := result[0]["rate_limit"] if !ok { t.Fatal("Expected 'rate_limit' key in the map") } rateLimitFloat, ok := rateLimit.(float64) if !ok { t.Fatal("Expected 'rate_limit' to be float64") } if rateLimitFloat != expectedRateLimit { t.Fatalf("Expected 'rate_limit' to equal %f, got %f", expectedRateLimit, rateLimitFloat) } } func TestFlattenApiGatewayStageKeys(t *testing.T) { cases := []struct { Input []*string Output []map[string]interface{} }{ { Input: []*string{ aws.String("a1b2c3d4e5/dev"), aws.String("e5d4c3b2a1/test"), }, Output: []map[string]interface{}{ map[string]interface{}{ "stage_name": "dev", "rest_api_id": "a1b2c3d4e5", }, map[string]interface{}{ "stage_name": "test", "rest_api_id": "e5d4c3b2a1", }, }, }, } for _, tc := range cases { output := flattenApiGatewayStageKeys(tc.Input) if !reflect.DeepEqual(output, tc.Output) { t.Fatalf("Got:\n\n%#v\n\nExpected:\n\n%#v", output, tc.Output) } } }