github.com/vtorhonen/terraform@v0.9.0-beta2.0.20170307220345-5d894e4ffda7/builtin/providers/openstack/resource_openstack_networking_secgroup_v2_test.go (about)

     1  package openstack
     2  
     3  import (
     4  	"fmt"
     5  	"testing"
     6  
     7  	"github.com/hashicorp/terraform/helper/resource"
     8  	"github.com/hashicorp/terraform/terraform"
     9  
    10  	"github.com/gophercloud/gophercloud/openstack/networking/v2/extensions/security/groups"
    11  )
    12  
    13  func TestAccNetworkingV2SecGroup_basic(t *testing.T) {
    14  	var security_group groups.SecGroup
    15  
    16  	resource.Test(t, resource.TestCase{
    17  		PreCheck:     func() { testAccPreCheck(t) },
    18  		Providers:    testAccProviders,
    19  		CheckDestroy: testAccCheckNetworkingV2SecGroupDestroy,
    20  		Steps: []resource.TestStep{
    21  			resource.TestStep{
    22  				Config: testAccNetworkingV2SecGroup_basic,
    23  				Check: resource.ComposeTestCheckFunc(
    24  					testAccCheckNetworkingV2SecGroupExists(
    25  						"openstack_networking_secgroup_v2.secgroup_1", &security_group),
    26  					testAccCheckNetworkingV2SecGroupRuleCount(&security_group, 2),
    27  				),
    28  			},
    29  			resource.TestStep{
    30  				Config: testAccNetworkingV2SecGroup_update,
    31  				Check: resource.ComposeTestCheckFunc(
    32  					resource.TestCheckResourceAttr(
    33  						"openstack_networking_secgroup_v2.secgroup_1", "name", "security_group_2"),
    34  				),
    35  			},
    36  		},
    37  	})
    38  }
    39  
    40  func TestAccNetworkingV2SecGroup_noDefaultRules(t *testing.T) {
    41  	var security_group groups.SecGroup
    42  
    43  	resource.Test(t, resource.TestCase{
    44  		PreCheck:     func() { testAccPreCheck(t) },
    45  		Providers:    testAccProviders,
    46  		CheckDestroy: testAccCheckNetworkingV2SecGroupDestroy,
    47  		Steps: []resource.TestStep{
    48  			resource.TestStep{
    49  				Config: testAccNetworkingV2SecGroup_noDefaultRules,
    50  				Check: resource.ComposeTestCheckFunc(
    51  					testAccCheckNetworkingV2SecGroupExists(
    52  						"openstack_networking_secgroup_v2.secgroup_1", &security_group),
    53  					testAccCheckNetworkingV2SecGroupRuleCount(&security_group, 0),
    54  				),
    55  			},
    56  		},
    57  	})
    58  }
    59  
    60  func testAccCheckNetworkingV2SecGroupDestroy(s *terraform.State) error {
    61  	config := testAccProvider.Meta().(*Config)
    62  	networkingClient, err := config.networkingV2Client(OS_REGION_NAME)
    63  	if err != nil {
    64  		return fmt.Errorf("Error creating OpenStack networking client: %s", err)
    65  	}
    66  
    67  	for _, rs := range s.RootModule().Resources {
    68  		if rs.Type != "openstack_networking_secgroup_v2" {
    69  			continue
    70  		}
    71  
    72  		_, err := groups.Get(networkingClient, rs.Primary.ID).Extract()
    73  		if err == nil {
    74  			return fmt.Errorf("Security group still exists")
    75  		}
    76  	}
    77  
    78  	return nil
    79  }
    80  
    81  func testAccCheckNetworkingV2SecGroupExists(n string, security_group *groups.SecGroup) resource.TestCheckFunc {
    82  	return func(s *terraform.State) error {
    83  		rs, ok := s.RootModule().Resources[n]
    84  		if !ok {
    85  			return fmt.Errorf("Not found: %s", n)
    86  		}
    87  
    88  		if rs.Primary.ID == "" {
    89  			return fmt.Errorf("No ID is set")
    90  		}
    91  
    92  		config := testAccProvider.Meta().(*Config)
    93  		networkingClient, err := config.networkingV2Client(OS_REGION_NAME)
    94  		if err != nil {
    95  			return fmt.Errorf("Error creating OpenStack networking client: %s", err)
    96  		}
    97  
    98  		found, err := groups.Get(networkingClient, rs.Primary.ID).Extract()
    99  		if err != nil {
   100  			return err
   101  		}
   102  
   103  		if found.ID != rs.Primary.ID {
   104  			return fmt.Errorf("Security group not found")
   105  		}
   106  
   107  		*security_group = *found
   108  
   109  		return nil
   110  	}
   111  }
   112  
   113  func testAccCheckNetworkingV2SecGroupRuleCount(
   114  	sg *groups.SecGroup, count int) resource.TestCheckFunc {
   115  	return func(s *terraform.State) error {
   116  		if len(sg.Rules) == count {
   117  			return nil
   118  		}
   119  
   120  		return fmt.Errorf("Unexpected number of rules in group %s. Expected %d, got %d",
   121  			sg.ID, count, len(sg.Rules))
   122  	}
   123  }
   124  
   125  const testAccNetworkingV2SecGroup_basic = `
   126  resource "openstack_networking_secgroup_v2" "secgroup_1" {
   127    name = "security_group"
   128    description = "terraform security group acceptance test"
   129  }
   130  `
   131  
   132  const testAccNetworkingV2SecGroup_update = `
   133  resource "openstack_networking_secgroup_v2" "secgroup_1" {
   134    name = "security_group_2"
   135    description = "terraform security group acceptance test"
   136  }
   137  `
   138  
   139  const testAccNetworkingV2SecGroup_noDefaultRules = `
   140  resource "openstack_networking_secgroup_v2" "secgroup_1" {
   141  	name = "security_group_1"
   142  	description = "terraform security group acceptance test"
   143  	delete_default_rules = true
   144  }
   145  `