482 lines
11 KiB
Go
482 lines
11 KiB
Go
package aws
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/terraform/helper/resource"
|
|
"github.com/hashicorp/terraform/helper/schema"
|
|
"github.com/hashicorp/terraform/terraform"
|
|
"github.com/mitchellh/goamz/ec2"
|
|
)
|
|
|
|
func TestAccAWSInstance_normal(t *testing.T) {
|
|
var v ec2.Instance
|
|
|
|
testCheck := func(*terraform.State) error {
|
|
if v.AvailZone != "us-west-2a" {
|
|
return fmt.Errorf("bad availability zone: %#v", v.AvailZone)
|
|
}
|
|
|
|
if len(v.SecurityGroups) == 0 {
|
|
return fmt.Errorf("no security groups: %#v", v.SecurityGroups)
|
|
}
|
|
if v.SecurityGroups[0].Name != "tf_test_foo" {
|
|
return fmt.Errorf("no security groups: %#v", v.SecurityGroups)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
resource.Test(t, resource.TestCase{
|
|
PreCheck: func() { testAccPreCheck(t) },
|
|
Providers: testAccProviders,
|
|
CheckDestroy: testAccCheckInstanceDestroy,
|
|
Steps: []resource.TestStep{
|
|
resource.TestStep{
|
|
Config: testAccInstanceConfig,
|
|
Check: resource.ComposeTestCheckFunc(
|
|
testAccCheckInstanceExists(
|
|
"aws_instance.foo", &v),
|
|
testCheck,
|
|
resource.TestCheckResourceAttr(
|
|
"aws_instance.foo",
|
|
"user_data",
|
|
"0beec7b5ea3f0fdbc95d0dd47f3c5bc275da8a33"),
|
|
),
|
|
},
|
|
|
|
// We repeat the exact same test so that we can be sure
|
|
// that the user data hash stuff is working without generating
|
|
// an incorrect diff.
|
|
resource.TestStep{
|
|
Config: testAccInstanceConfig,
|
|
Check: resource.ComposeTestCheckFunc(
|
|
testAccCheckInstanceExists(
|
|
"aws_instance.foo", &v),
|
|
testCheck,
|
|
resource.TestCheckResourceAttr(
|
|
"aws_instance.foo",
|
|
"user_data",
|
|
"0beec7b5ea3f0fdbc95d0dd47f3c5bc275da8a33"),
|
|
),
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
func TestAccAWSInstance_blockDevicesCheck(t *testing.T) {
|
|
var v ec2.Instance
|
|
|
|
testCheck := func() resource.TestCheckFunc {
|
|
return func(*terraform.State) error {
|
|
|
|
// Map out the block devices by name, which should be unique.
|
|
blockDevices := make(map[string]ec2.BlockDevice)
|
|
for _, blockDevice := range v.BlockDevices {
|
|
blockDevices[blockDevice.DeviceName] = blockDevice
|
|
}
|
|
|
|
// Check if the secondary block device exists.
|
|
if _, ok := blockDevices["/dev/sdb"]; !ok {
|
|
fmt.Errorf("block device doesn't exist: /dev/sdb")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
resource.Test(t, resource.TestCase{
|
|
PreCheck: func() { testAccPreCheck(t) },
|
|
Providers: testAccProviders,
|
|
CheckDestroy: testAccCheckInstanceDestroy,
|
|
Steps: []resource.TestStep{
|
|
resource.TestStep{
|
|
Config: testAccInstanceConfigBlockDevices,
|
|
Check: resource.ComposeTestCheckFunc(
|
|
testAccCheckInstanceExists(
|
|
"aws_instance.foo", &v),
|
|
// though two block devices exist in EC2, terraform state should only
|
|
// have the one block device we created, as terraform does not manage
|
|
// the root device
|
|
resource.TestCheckResourceAttr(
|
|
"aws_instance.foo", "block_device.#", "1"),
|
|
testCheck(),
|
|
),
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
func TestAccAWSInstance_sourceDestCheck(t *testing.T) {
|
|
var v ec2.Instance
|
|
|
|
testCheck := func(enabled bool) resource.TestCheckFunc {
|
|
return func(*terraform.State) error {
|
|
if v.SourceDestCheck != enabled {
|
|
return fmt.Errorf("bad source_dest_check: %#v", v.SourceDestCheck)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
resource.Test(t, resource.TestCase{
|
|
PreCheck: func() { testAccPreCheck(t) },
|
|
Providers: testAccProviders,
|
|
CheckDestroy: testAccCheckInstanceDestroy,
|
|
Steps: []resource.TestStep{
|
|
resource.TestStep{
|
|
Config: testAccInstanceConfigSourceDest,
|
|
Check: resource.ComposeTestCheckFunc(
|
|
testAccCheckInstanceExists(
|
|
"aws_instance.foo", &v),
|
|
testCheck(true),
|
|
),
|
|
},
|
|
|
|
resource.TestStep{
|
|
Config: testAccInstanceConfigSourceDestDisable,
|
|
Check: resource.ComposeTestCheckFunc(
|
|
testAccCheckInstanceExists(
|
|
"aws_instance.foo", &v),
|
|
testCheck(false),
|
|
),
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
func TestAccAWSInstance_vpc(t *testing.T) {
|
|
var v ec2.Instance
|
|
|
|
resource.Test(t, resource.TestCase{
|
|
PreCheck: func() { testAccPreCheck(t) },
|
|
Providers: testAccProviders,
|
|
CheckDestroy: testAccCheckInstanceDestroy,
|
|
Steps: []resource.TestStep{
|
|
resource.TestStep{
|
|
Config: testAccInstanceConfigVPC,
|
|
Check: resource.ComposeTestCheckFunc(
|
|
testAccCheckInstanceExists(
|
|
"aws_instance.foo", &v),
|
|
),
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
func TestAccInstance_tags(t *testing.T) {
|
|
var v ec2.Instance
|
|
|
|
resource.Test(t, resource.TestCase{
|
|
PreCheck: func() { testAccPreCheck(t) },
|
|
Providers: testAccProviders,
|
|
CheckDestroy: testAccCheckInstanceDestroy,
|
|
Steps: []resource.TestStep{
|
|
resource.TestStep{
|
|
Config: testAccCheckInstanceConfigTags,
|
|
Check: resource.ComposeTestCheckFunc(
|
|
testAccCheckInstanceExists("aws_instance.foo", &v),
|
|
testAccCheckTags(&v.Tags, "foo", "bar"),
|
|
),
|
|
},
|
|
|
|
resource.TestStep{
|
|
Config: testAccCheckInstanceConfigTagsUpdate,
|
|
Check: resource.ComposeTestCheckFunc(
|
|
testAccCheckInstanceExists("aws_instance.foo", &v),
|
|
testAccCheckTags(&v.Tags, "foo", ""),
|
|
testAccCheckTags(&v.Tags, "bar", "baz"),
|
|
),
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
func TestAccInstance_privateIP(t *testing.T) {
|
|
var v ec2.Instance
|
|
|
|
testCheckPrivateIP := func() resource.TestCheckFunc {
|
|
return func(*terraform.State) error {
|
|
if v.PrivateIpAddress != "10.1.1.42" {
|
|
return fmt.Errorf("bad private IP: %s", v.PrivateIpAddress)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
resource.Test(t, resource.TestCase{
|
|
PreCheck: func() { testAccPreCheck(t) },
|
|
Providers: testAccProviders,
|
|
CheckDestroy: testAccCheckInstanceDestroy,
|
|
Steps: []resource.TestStep{
|
|
resource.TestStep{
|
|
Config: testAccInstanceConfigPrivateIP,
|
|
Check: resource.ComposeTestCheckFunc(
|
|
testAccCheckInstanceExists("aws_instance.foo", &v),
|
|
testCheckPrivateIP(),
|
|
),
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
func TestAccInstance_associatePublicIPAndPrivateIP(t *testing.T) {
|
|
var v ec2.Instance
|
|
|
|
testCheckPrivateIP := func() resource.TestCheckFunc {
|
|
return func(*terraform.State) error {
|
|
if v.PrivateIpAddress != "10.1.1.42" {
|
|
return fmt.Errorf("bad private IP: %s", v.PrivateIpAddress)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
resource.Test(t, resource.TestCase{
|
|
PreCheck: func() { testAccPreCheck(t) },
|
|
Providers: testAccProviders,
|
|
CheckDestroy: testAccCheckInstanceDestroy,
|
|
Steps: []resource.TestStep{
|
|
resource.TestStep{
|
|
Config: testAccInstanceConfigAssociatePublicIPAndPrivateIP,
|
|
Check: resource.ComposeTestCheckFunc(
|
|
testAccCheckInstanceExists("aws_instance.foo", &v),
|
|
testCheckPrivateIP(),
|
|
),
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
func testAccCheckInstanceDestroy(s *terraform.State) error {
|
|
conn := testAccProvider.Meta().(*AWSClient).ec2conn
|
|
|
|
for _, rs := range s.RootModule().Resources {
|
|
if rs.Type != "aws_instance" {
|
|
continue
|
|
}
|
|
|
|
// Try to find the resource
|
|
resp, err := conn.Instances(
|
|
[]string{rs.Primary.ID}, ec2.NewFilter())
|
|
if err == nil {
|
|
if len(resp.Reservations) > 0 {
|
|
return fmt.Errorf("still exist.")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Verify the error is what we want
|
|
ec2err, ok := err.(*ec2.Error)
|
|
if !ok {
|
|
return err
|
|
}
|
|
if ec2err.Code != "InvalidInstanceID.NotFound" {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func testAccCheckInstanceExists(n string, i *ec2.Instance) resource.TestCheckFunc {
|
|
return func(s *terraform.State) error {
|
|
rs, ok := s.RootModule().Resources[n]
|
|
if !ok {
|
|
return fmt.Errorf("Not found: %s", n)
|
|
}
|
|
|
|
if rs.Primary.ID == "" {
|
|
return fmt.Errorf("No ID is set")
|
|
}
|
|
|
|
conn := testAccProvider.Meta().(*AWSClient).ec2conn
|
|
resp, err := conn.Instances(
|
|
[]string{rs.Primary.ID}, ec2.NewFilter())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if len(resp.Reservations) == 0 {
|
|
return fmt.Errorf("Instance not found")
|
|
}
|
|
|
|
*i = resp.Reservations[0].Instances[0]
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func TestInstanceTenancySchema(t *testing.T) {
|
|
actualSchema := resourceAwsInstance().Schema["tenancy"]
|
|
expectedSchema := &schema.Schema{
|
|
Type: schema.TypeString,
|
|
Optional: true,
|
|
Computed: true,
|
|
ForceNew: true,
|
|
}
|
|
if !reflect.DeepEqual(actualSchema, expectedSchema) {
|
|
t.Fatalf(
|
|
"Got:\n\n%#v\n\nExpected:\n\n%#v\n",
|
|
actualSchema,
|
|
expectedSchema)
|
|
}
|
|
}
|
|
|
|
const testAccInstanceConfig = `
|
|
resource "aws_security_group" "tf_test_foo" {
|
|
name = "tf_test_foo"
|
|
description = "foo"
|
|
|
|
ingress {
|
|
protocol = "icmp"
|
|
from_port = -1
|
|
to_port = -1
|
|
cidr_blocks = ["0.0.0.0/0"]
|
|
}
|
|
}
|
|
|
|
resource "aws_instance" "foo" {
|
|
# us-west-2
|
|
ami = "ami-4fccb37f"
|
|
availability_zone = "us-west-2a"
|
|
|
|
instance_type = "m1.small"
|
|
security_groups = ["${aws_security_group.tf_test_foo.name}"]
|
|
user_data = "foo"
|
|
}
|
|
`
|
|
|
|
const testAccInstanceConfigBlockDevices = `
|
|
resource "aws_instance" "foo" {
|
|
# us-west-2
|
|
ami = "ami-55a7ea65"
|
|
instance_type = "m1.small"
|
|
block_device {
|
|
device_name = "/dev/sdb"
|
|
volume_type = "gp2"
|
|
volume_size = 10
|
|
}
|
|
}
|
|
`
|
|
|
|
const testAccInstanceConfigSourceDest = `
|
|
resource "aws_vpc" "foo" {
|
|
cidr_block = "10.1.0.0/16"
|
|
}
|
|
|
|
resource "aws_subnet" "foo" {
|
|
cidr_block = "10.1.1.0/24"
|
|
vpc_id = "${aws_vpc.foo.id}"
|
|
}
|
|
|
|
resource "aws_instance" "foo" {
|
|
# us-west-2
|
|
ami = "ami-4fccb37f"
|
|
instance_type = "m1.small"
|
|
subnet_id = "${aws_subnet.foo.id}"
|
|
source_dest_check = true
|
|
}
|
|
`
|
|
|
|
const testAccInstanceConfigSourceDestDisable = `
|
|
resource "aws_vpc" "foo" {
|
|
cidr_block = "10.1.0.0/16"
|
|
}
|
|
|
|
resource "aws_subnet" "foo" {
|
|
cidr_block = "10.1.1.0/24"
|
|
vpc_id = "${aws_vpc.foo.id}"
|
|
}
|
|
|
|
resource "aws_instance" "foo" {
|
|
# us-west-2
|
|
ami = "ami-4fccb37f"
|
|
instance_type = "m1.small"
|
|
subnet_id = "${aws_subnet.foo.id}"
|
|
source_dest_check = false
|
|
}
|
|
`
|
|
|
|
const testAccInstanceConfigVPC = `
|
|
resource "aws_vpc" "foo" {
|
|
cidr_block = "10.1.0.0/16"
|
|
}
|
|
|
|
resource "aws_subnet" "foo" {
|
|
cidr_block = "10.1.1.0/24"
|
|
vpc_id = "${aws_vpc.foo.id}"
|
|
}
|
|
|
|
resource "aws_instance" "foo" {
|
|
# us-west-2
|
|
ami = "ami-4fccb37f"
|
|
instance_type = "m1.small"
|
|
subnet_id = "${aws_subnet.foo.id}"
|
|
associate_public_ip_address = true
|
|
tenancy = "dedicated"
|
|
}
|
|
`
|
|
|
|
const testAccCheckInstanceConfigTags = `
|
|
resource "aws_instance" "foo" {
|
|
ami = "ami-4fccb37f"
|
|
instance_type = "m1.small"
|
|
tags {
|
|
foo = "bar"
|
|
}
|
|
}
|
|
`
|
|
|
|
const testAccCheckInstanceConfigTagsUpdate = `
|
|
resource "aws_instance" "foo" {
|
|
ami = "ami-4fccb37f"
|
|
instance_type = "m1.small"
|
|
tags {
|
|
bar = "baz"
|
|
}
|
|
}
|
|
`
|
|
|
|
const testAccInstanceConfigPrivateIP = `
|
|
resource "aws_vpc" "foo" {
|
|
cidr_block = "10.1.0.0/16"
|
|
}
|
|
|
|
resource "aws_subnet" "foo" {
|
|
cidr_block = "10.1.1.0/24"
|
|
vpc_id = "${aws_vpc.foo.id}"
|
|
}
|
|
|
|
resource "aws_instance" "foo" {
|
|
ami = "ami-c5eabbf5"
|
|
instance_type = "t2.micro"
|
|
subnet_id = "${aws_subnet.foo.id}"
|
|
private_ip = "10.1.1.42"
|
|
}
|
|
`
|
|
|
|
const testAccInstanceConfigAssociatePublicIPAndPrivateIP = `
|
|
resource "aws_vpc" "foo" {
|
|
cidr_block = "10.1.0.0/16"
|
|
}
|
|
|
|
resource "aws_subnet" "foo" {
|
|
cidr_block = "10.1.1.0/24"
|
|
vpc_id = "${aws_vpc.foo.id}"
|
|
}
|
|
|
|
resource "aws_instance" "foo" {
|
|
ami = "ami-c5eabbf5"
|
|
instance_type = "t2.micro"
|
|
subnet_id = "${aws_subnet.foo.id}"
|
|
associate_public_ip_address = true
|
|
private_ip = "10.1.1.42"
|
|
}
|
|
`
|