package aws import ( "fmt" "os" "testing" "github.com/aws/aws-sdk-go/aws" "github.com/aws/aws-sdk-go/service/ec2" "github.com/hashicorp/terraform/helper/resource" "github.com/hashicorp/terraform/terraform" ) func TestAccAWSFlowLog_basic(t *testing.T) { var flowLog ec2.FlowLog lgn := os.Getenv("LOG_GROUP_NAME") resource.Test(t, resource.TestCase{ PreCheck: func() { testAccPreCheck(t) }, Providers: testAccProviders, CheckDestroy: testAccCheckFlowLogDestroy, Steps: []resource.TestStep{ resource.TestStep{ Config: fmt.Sprintf(testAccFlowLogConfig_basic, lgn), Check: resource.ComposeTestCheckFunc( testAccCheckFlowLogExists("aws_flow_log.test_flow_log", &flowLog), testAccCheckAWSFlowLogAttributes(&flowLog), ), }, }, }) } func TestAccAWSFlowLog_subnet(t *testing.T) { var flowLog ec2.FlowLog lgn := os.Getenv("LOG_GROUP_NAME") resource.Test(t, resource.TestCase{ PreCheck: func() { testAccPreCheck(t) }, Providers: testAccProviders, CheckDestroy: testAccCheckFlowLogDestroy, Steps: []resource.TestStep{ resource.TestStep{ Config: fmt.Sprintf(testAccFlowLogConfig_subnet, lgn), Check: resource.ComposeTestCheckFunc( testAccCheckFlowLogExists("aws_flow_log.test_flow_log_subnet", &flowLog), testAccCheckAWSFlowLogAttributes(&flowLog), ), }, }, }) } func testAccCheckFlowLogExists(n string, flowLog *ec2.FlowLog) 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 Flow Log ID is set") } conn := testAccProvider.Meta().(*AWSClient).ec2conn describeOpts := &ec2.DescribeFlowLogsInput{ FlowLogIDs: []*string{aws.String(rs.Primary.ID)}, } resp, err := conn.DescribeFlowLogs(describeOpts) if err != nil { return err } if len(resp.FlowLogs) > 0 { *flowLog = *resp.FlowLogs[0] return nil } return fmt.Errorf("No Flow Logs found for id (%s)", rs.Primary.ID) } } func testAccCheckAWSFlowLogAttributes(flowLog *ec2.FlowLog) resource.TestCheckFunc { return func(s *terraform.State) error { if flowLog.FlowLogStatus != nil && *flowLog.FlowLogStatus == "ACTIVE" { return nil } if flowLog.FlowLogStatus == nil { return fmt.Errorf("Flow Log status is not ACTIVE, is nil") } else { return fmt.Errorf("Flow Log status is not ACTIVE, got: %s", *flowLog.FlowLogStatus) } } } func testAccCheckFlowLogDestroy(s *terraform.State) error { for _, rs := range s.RootModule().Resources { if rs.Type != "aws_flow_log" { continue } return nil } return nil } var testAccFlowLogConfig_basic = ` resource "aws_vpc" "default" { cidr_block = "10.0.0.0/16" tags { Name = "tf-flow-log-test" } } resource "aws_subnet" "test_subnet" { vpc_id = "${aws_vpc.default.id}" cidr_block = "10.0.1.0/24" tags { Name = "tf-flow-test" } } resource "aws_iam_role" "test_role" { name = "test_role" assume_role_policy = <