245 lines
5.8 KiB
Go
245 lines
5.8 KiB
Go
![]() |
package azurerm
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"math/rand"
|
||
|
"net/http"
|
||
|
"testing"
|
||
|
"time"
|
||
|
|
||
|
"github.com/hashicorp/terraform/helper/resource"
|
||
|
"github.com/hashicorp/terraform/terraform"
|
||
|
)
|
||
|
|
||
|
func TestResourceAzureRMPublicIpAllocation_validation(t *testing.T) {
|
||
|
cases := []struct {
|
||
|
Value string
|
||
|
ErrCount int
|
||
|
}{
|
||
|
{
|
||
|
Value: "Random",
|
||
|
ErrCount: 1,
|
||
|
},
|
||
|
{
|
||
|
Value: "Static",
|
||
|
ErrCount: 0,
|
||
|
},
|
||
|
{
|
||
|
Value: "Dynamic",
|
||
|
ErrCount: 0,
|
||
|
},
|
||
|
{
|
||
|
Value: "STATIC",
|
||
|
ErrCount: 0,
|
||
|
},
|
||
|
{
|
||
|
Value: "static",
|
||
|
ErrCount: 0,
|
||
|
},
|
||
|
}
|
||
|
|
||
|
for _, tc := range cases {
|
||
|
_, errors := validatePublicIpAllocation(tc.Value, "azurerm_public_ip")
|
||
|
|
||
|
if len(errors) != tc.ErrCount {
|
||
|
t.Fatalf("Expected the Azure RM Public IP allocation to trigger a validation error")
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestResourceAzureRMPublicIpDomainNameLabel_validation(t *testing.T) {
|
||
|
cases := []struct {
|
||
|
Value string
|
||
|
ErrCount int
|
||
|
}{
|
||
|
{
|
||
|
Value: "tEsting123",
|
||
|
ErrCount: 1,
|
||
|
},
|
||
|
{
|
||
|
Value: "testing123!",
|
||
|
ErrCount: 1,
|
||
|
},
|
||
|
{
|
||
|
Value: "testing123-",
|
||
|
ErrCount: 1,
|
||
|
},
|
||
|
{
|
||
|
Value: randomString(80),
|
||
|
ErrCount: 1,
|
||
|
},
|
||
|
}
|
||
|
|
||
|
for _, tc := range cases {
|
||
|
_, errors := validatePublicIpDomainNameLabel(tc.Value, "azurerm_public_ip")
|
||
|
|
||
|
if len(errors) != tc.ErrCount {
|
||
|
t.Fatalf("Expected the Azure RM Public IP Domain Name Label to trigger a validation error")
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestAccAzureRMPublicIpStatic_basic(t *testing.T) {
|
||
|
|
||
|
resource.Test(t, resource.TestCase{
|
||
|
PreCheck: func() { testAccPreCheck(t) },
|
||
|
Providers: testAccProviders,
|
||
|
CheckDestroy: testCheckAzureRMPublicIpDestroy,
|
||
|
Steps: []resource.TestStep{
|
||
|
resource.TestStep{
|
||
|
Config: testAccAzureRMVPublicIpStatic_basic,
|
||
|
Check: resource.ComposeTestCheckFunc(
|
||
|
testCheckAzureRMPublicIpExists("azurerm_public_ip.test"),
|
||
|
),
|
||
|
},
|
||
|
},
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func TestAccAzureRMPublicIpStatic_update(t *testing.T) {
|
||
|
|
||
|
resource.Test(t, resource.TestCase{
|
||
|
PreCheck: func() { testAccPreCheck(t) },
|
||
|
Providers: testAccProviders,
|
||
|
CheckDestroy: testCheckAzureRMPublicIpDestroy,
|
||
|
Steps: []resource.TestStep{
|
||
|
resource.TestStep{
|
||
|
Config: testAccAzureRMVPublicIpStatic_basic,
|
||
|
Check: resource.ComposeTestCheckFunc(
|
||
|
testCheckAzureRMPublicIpExists("azurerm_public_ip.test"),
|
||
|
),
|
||
|
},
|
||
|
|
||
|
resource.TestStep{
|
||
|
Config: testAccAzureRMVPublicIpStatic_update,
|
||
|
Check: resource.ComposeTestCheckFunc(
|
||
|
testCheckAzureRMPublicIpExists("azurerm_public_ip.test"),
|
||
|
resource.TestCheckResourceAttr(
|
||
|
"azurerm_public_ip.test", "domain_name_label", "mylabel01"),
|
||
|
),
|
||
|
},
|
||
|
},
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func TestAccAzureRMPublicIpDynamic_basic(t *testing.T) {
|
||
|
|
||
|
resource.Test(t, resource.TestCase{
|
||
|
PreCheck: func() { testAccPreCheck(t) },
|
||
|
Providers: testAccProviders,
|
||
|
CheckDestroy: testCheckAzureRMPublicIpDestroy,
|
||
|
Steps: []resource.TestStep{
|
||
|
resource.TestStep{
|
||
|
Config: testAccAzureRMVPublicIpDynamic_basic,
|
||
|
Check: resource.ComposeTestCheckFunc(
|
||
|
testCheckAzureRMPublicIpExists("azurerm_public_ip.test"),
|
||
|
),
|
||
|
},
|
||
|
},
|
||
|
})
|
||
|
}
|
||
|
|
||
|
func testCheckAzureRMPublicIpExists(name string) resource.TestCheckFunc {
|
||
|
return func(s *terraform.State) error {
|
||
|
// Ensure we have enough information in state to look up in API
|
||
|
rs, ok := s.RootModule().Resources[name]
|
||
|
if !ok {
|
||
|
return fmt.Errorf("Not found: %s", name)
|
||
|
}
|
||
|
|
||
|
availSetName := rs.Primary.Attributes["name"]
|
||
|
resourceGroup, hasResourceGroup := rs.Primary.Attributes["resource_group_name"]
|
||
|
if !hasResourceGroup {
|
||
|
return fmt.Errorf("Bad: no resource group found in state for public ip: %s", availSetName)
|
||
|
}
|
||
|
|
||
|
conn := testAccProvider.Meta().(*ArmClient).publicIPClient
|
||
|
|
||
|
resp, err := conn.Get(resourceGroup, availSetName)
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("Bad: Get on publicIPClient: %s", err)
|
||
|
}
|
||
|
|
||
|
if resp.StatusCode == http.StatusNotFound {
|
||
|
return fmt.Errorf("Bad: Public IP %q (resource group: %q) does not exist", name, resourceGroup)
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func testCheckAzureRMPublicIpDestroy(s *terraform.State) error {
|
||
|
conn := testAccProvider.Meta().(*ArmClient).publicIPClient
|
||
|
|
||
|
for _, rs := range s.RootModule().Resources {
|
||
|
if rs.Type != "azurerm_public_ip" {
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
name := rs.Primary.Attributes["name"]
|
||
|
resourceGroup := rs.Primary.Attributes["resource_group_name"]
|
||
|
|
||
|
resp, err := conn.Get(resourceGroup, name)
|
||
|
|
||
|
if err != nil {
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
if resp.StatusCode != http.StatusNotFound {
|
||
|
return fmt.Errorf("Public IP still exists:\n%#v", resp.Properties)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func randomString(strlen int) string {
|
||
|
rand.Seed(time.Now().UTC().UnixNano())
|
||
|
const chars = "abcdefghijklmnopqrstuvwxyz0123456789-"
|
||
|
result := make([]byte, strlen)
|
||
|
for i := 0; i < strlen; i++ {
|
||
|
result[i] = chars[rand.Intn(len(chars))]
|
||
|
}
|
||
|
return string(result)
|
||
|
}
|
||
|
|
||
|
var testAccAzureRMVPublicIpStatic_basic = `
|
||
|
resource "azurerm_resource_group" "test" {
|
||
|
name = "acceptanceTestResourceGroup1"
|
||
|
location = "West US"
|
||
|
}
|
||
|
resource "azurerm_public_ip" "test" {
|
||
|
name = "acceptanceTestPublicIp1"
|
||
|
location = "West US"
|
||
|
resource_group_name = "${azurerm_resource_group.test.name}"
|
||
|
public_ip_address_allocation = "static"
|
||
|
}
|
||
|
`
|
||
|
|
||
|
var testAccAzureRMVPublicIpStatic_update = `
|
||
|
resource "azurerm_resource_group" "test" {
|
||
|
name = "acceptanceTestResourceGroup1"
|
||
|
location = "West US"
|
||
|
}
|
||
|
resource "azurerm_public_ip" "test" {
|
||
|
name = "acceptanceTestPublicIp1"
|
||
|
location = "West US"
|
||
|
resource_group_name = "${azurerm_resource_group.test.name}"
|
||
|
public_ip_address_allocation = "static"
|
||
|
domain_name_label = "mylabel01"
|
||
|
}
|
||
|
`
|
||
|
|
||
|
var testAccAzureRMVPublicIpDynamic_basic = `
|
||
|
resource "azurerm_resource_group" "test" {
|
||
|
name = "acceptanceTestResourceGroup2"
|
||
|
location = "West US"
|
||
|
}
|
||
|
resource "azurerm_public_ip" "test" {
|
||
|
name = "acceptanceTestPublicIp2"
|
||
|
location = "West US"
|
||
|
resource_group_name = "${azurerm_resource_group.test.name}"
|
||
|
public_ip_address_allocation = "dynamic"
|
||
|
}
|
||
|
`
|