456 lines
12 KiB
Go
456 lines
12 KiB
Go
package azure
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/terraform/helper/resource"
|
|
"github.com/hashicorp/terraform/terraform"
|
|
"github.com/svanharmelen/azure-sdk-for-go/management"
|
|
"github.com/svanharmelen/azure-sdk-for-go/management/hostedservice"
|
|
"github.com/svanharmelen/azure-sdk-for-go/management/virtualmachine"
|
|
)
|
|
|
|
func TestAccAzureInstance_basic(t *testing.T) {
|
|
var dpmt virtualmachine.DeploymentResponse
|
|
|
|
resource.Test(t, resource.TestCase{
|
|
PreCheck: func() { testAccPreCheck(t) },
|
|
Providers: testAccProviders,
|
|
CheckDestroy: testAccCheckAzureInstanceDestroy,
|
|
Steps: []resource.TestStep{
|
|
resource.TestStep{
|
|
Config: testAccAzureInstance_basic,
|
|
Check: resource.ComposeTestCheckFunc(
|
|
testAccCheckAzureInstanceExists(
|
|
"azure_instance.foo", &dpmt),
|
|
testAccCheckAzureInstanceBasicAttributes(&dpmt),
|
|
resource.TestCheckResourceAttr(
|
|
"azure_instance.foo", "name", "terraform-test"),
|
|
resource.TestCheckResourceAttr(
|
|
"azure_instance.foo", "location", "West US"),
|
|
resource.TestCheckResourceAttr(
|
|
"azure_instance.foo", "endpoint.2462817782.public_port", "22"),
|
|
),
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
func TestAccAzureInstance_advanced(t *testing.T) {
|
|
var dpmt virtualmachine.DeploymentResponse
|
|
|
|
resource.Test(t, resource.TestCase{
|
|
PreCheck: func() { testAccPreCheck(t) },
|
|
Providers: testAccProviders,
|
|
CheckDestroy: testAccCheckAzureInstanceDestroy,
|
|
Steps: []resource.TestStep{
|
|
resource.TestStep{
|
|
Config: testAccAzureInstance_advanced,
|
|
Check: resource.ComposeTestCheckFunc(
|
|
testAccCheckAzureInstanceExists(
|
|
"azure_instance.foo", &dpmt),
|
|
testAccCheckAzureInstanceAdvancedAttributes(&dpmt),
|
|
resource.TestCheckResourceAttr(
|
|
"azure_instance.foo", "name", "terraform-test1"),
|
|
resource.TestCheckResourceAttr(
|
|
"azure_instance.foo", "size", "Basic_A1"),
|
|
resource.TestCheckResourceAttr(
|
|
"azure_instance.foo", "subnet", "subnet1"),
|
|
resource.TestCheckResourceAttr(
|
|
"azure_instance.foo", "virtual_network", "terraform-vnet"),
|
|
resource.TestCheckResourceAttr(
|
|
"azure_instance.foo", "security_group", "terraform-security-group1"),
|
|
resource.TestCheckResourceAttr(
|
|
"azure_instance.foo", "endpoint.1814039778.public_port", "3389"),
|
|
),
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
func TestAccAzureInstance_update(t *testing.T) {
|
|
var dpmt virtualmachine.DeploymentResponse
|
|
|
|
resource.Test(t, resource.TestCase{
|
|
PreCheck: func() { testAccPreCheck(t) },
|
|
Providers: testAccProviders,
|
|
CheckDestroy: testAccCheckAzureInstanceDestroy,
|
|
Steps: []resource.TestStep{
|
|
resource.TestStep{
|
|
Config: testAccAzureInstance_advanced,
|
|
Check: resource.ComposeTestCheckFunc(
|
|
testAccCheckAzureInstanceExists(
|
|
"azure_instance.foo", &dpmt),
|
|
testAccCheckAzureInstanceAdvancedAttributes(&dpmt),
|
|
resource.TestCheckResourceAttr(
|
|
"azure_instance.foo", "name", "terraform-test1"),
|
|
resource.TestCheckResourceAttr(
|
|
"azure_instance.foo", "size", "Basic_A1"),
|
|
resource.TestCheckResourceAttr(
|
|
"azure_instance.foo", "subnet", "subnet1"),
|
|
resource.TestCheckResourceAttr(
|
|
"azure_instance.foo", "virtual_network", "terraform-vnet"),
|
|
resource.TestCheckResourceAttr(
|
|
"azure_instance.foo", "security_group", "terraform-security-group1"),
|
|
resource.TestCheckResourceAttr(
|
|
"azure_instance.foo", "endpoint.1814039778.public_port", "3389"),
|
|
),
|
|
},
|
|
|
|
resource.TestStep{
|
|
Config: testAccAzureInstance_update,
|
|
Check: resource.ComposeTestCheckFunc(
|
|
testAccCheckAzureInstanceExists(
|
|
"azure_instance.foo", &dpmt),
|
|
testAccCheckAzureInstanceUpdatedAttributes(&dpmt),
|
|
resource.TestCheckResourceAttr(
|
|
"azure_instance.foo", "size", "Basic_A2"),
|
|
resource.TestCheckResourceAttr(
|
|
"azure_instance.foo", "security_group", "terraform-security-group2"),
|
|
resource.TestCheckResourceAttr(
|
|
"azure_instance.foo", "endpoint.1814039778.public_port", "3389"),
|
|
resource.TestCheckResourceAttr(
|
|
"azure_instance.foo", "endpoint.3713350066.public_port", "5985"),
|
|
),
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
func testAccCheckAzureInstanceExists(
|
|
n string,
|
|
dpmt *virtualmachine.DeploymentResponse) 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 instance ID is set")
|
|
}
|
|
|
|
mc := testAccProvider.Meta().(*Client).mgmtClient
|
|
vm, err := virtualmachine.NewClient(mc).GetDeployment(rs.Primary.ID, rs.Primary.ID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if vm.Name != rs.Primary.ID {
|
|
return fmt.Errorf("Instance not found")
|
|
}
|
|
|
|
*dpmt = vm
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func testAccCheckAzureInstanceBasicAttributes(
|
|
dpmt *virtualmachine.DeploymentResponse) resource.TestCheckFunc {
|
|
return func(s *terraform.State) error {
|
|
|
|
if dpmt.Name != "terraform-test" {
|
|
return fmt.Errorf("Bad name: %s", dpmt.Name)
|
|
}
|
|
|
|
if len(dpmt.RoleList) != 1 {
|
|
return fmt.Errorf(
|
|
"Instance %s has an unexpected number of roles: %d", dpmt.Name, len(dpmt.RoleList))
|
|
}
|
|
|
|
if dpmt.RoleList[0].RoleSize != "Basic_A1" {
|
|
return fmt.Errorf("Bad size: %s", dpmt.RoleList[0].RoleSize)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func testAccCheckAzureInstanceAdvancedAttributes(
|
|
dpmt *virtualmachine.DeploymentResponse) resource.TestCheckFunc {
|
|
return func(s *terraform.State) error {
|
|
|
|
if dpmt.Name != "terraform-test1" {
|
|
return fmt.Errorf("Bad name: %s", dpmt.Name)
|
|
}
|
|
|
|
if dpmt.VirtualNetworkName != "terraform-vnet" {
|
|
return fmt.Errorf("Bad virtual network: %s", dpmt.VirtualNetworkName)
|
|
}
|
|
|
|
if len(dpmt.RoleList) != 1 {
|
|
return fmt.Errorf(
|
|
"Instance %s has an unexpected number of roles: %d", dpmt.Name, len(dpmt.RoleList))
|
|
}
|
|
|
|
if dpmt.RoleList[0].RoleSize != "Basic_A1" {
|
|
return fmt.Errorf("Bad size: %s", dpmt.RoleList[0].RoleSize)
|
|
}
|
|
|
|
for _, c := range dpmt.RoleList[0].ConfigurationSets {
|
|
if c.ConfigurationSetType == virtualmachine.ConfigurationSetTypeNetwork {
|
|
if len(c.InputEndpoints) != 1 {
|
|
return fmt.Errorf(
|
|
"Instance %s has an unexpected number of endpoints %d",
|
|
dpmt.Name, len(c.InputEndpoints))
|
|
}
|
|
|
|
if c.InputEndpoints[0].Name != "RDP" {
|
|
return fmt.Errorf("Bad endpoint name: %s", c.InputEndpoints[0].Name)
|
|
}
|
|
|
|
if c.InputEndpoints[0].Port != 3389 {
|
|
return fmt.Errorf("Bad endpoint port: %d", c.InputEndpoints[0].Port)
|
|
}
|
|
|
|
if len(c.SubnetNames) != 1 {
|
|
return fmt.Errorf(
|
|
"Instance %s has an unexpected number of associated subnets %d",
|
|
dpmt.Name, len(c.SubnetNames))
|
|
}
|
|
|
|
if c.SubnetNames[0] != "subnet1" {
|
|
return fmt.Errorf("Bad subnet: %s", c.SubnetNames[0])
|
|
}
|
|
|
|
if c.NetworkSecurityGroup != "terraform-security-group1" {
|
|
return fmt.Errorf("Bad security group: %s", c.NetworkSecurityGroup)
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func testAccCheckAzureInstanceUpdatedAttributes(
|
|
dpmt *virtualmachine.DeploymentResponse) resource.TestCheckFunc {
|
|
return func(s *terraform.State) error {
|
|
|
|
if dpmt.Name != "terraform-test1" {
|
|
return fmt.Errorf("Bad name: %s", dpmt.Name)
|
|
}
|
|
|
|
if dpmt.VirtualNetworkName != "terraform-vnet" {
|
|
return fmt.Errorf("Bad virtual network: %s", dpmt.VirtualNetworkName)
|
|
}
|
|
|
|
if len(dpmt.RoleList) != 1 {
|
|
return fmt.Errorf(
|
|
"Instance %s has an unexpected number of roles: %d", dpmt.Name, len(dpmt.RoleList))
|
|
}
|
|
|
|
if dpmt.RoleList[0].RoleSize != "Basic_A2" {
|
|
return fmt.Errorf("Bad size: %s", dpmt.RoleList[0].RoleSize)
|
|
}
|
|
|
|
for _, c := range dpmt.RoleList[0].ConfigurationSets {
|
|
if c.ConfigurationSetType == virtualmachine.ConfigurationSetTypeNetwork {
|
|
if len(c.InputEndpoints) != 2 {
|
|
return fmt.Errorf(
|
|
"Instance %s has an unexpected number of endpoints %d",
|
|
dpmt.Name, len(c.InputEndpoints))
|
|
}
|
|
|
|
if c.InputEndpoints[1].Name != "WINRM" {
|
|
return fmt.Errorf("Bad endpoint name: %s", c.InputEndpoints[1].Name)
|
|
}
|
|
|
|
if c.InputEndpoints[1].Port != 5985 {
|
|
return fmt.Errorf("Bad endpoint port: %d", c.InputEndpoints[1].Port)
|
|
}
|
|
|
|
if len(c.SubnetNames) != 1 {
|
|
return fmt.Errorf(
|
|
"Instance %s has an unexpected number of associated subnets %d",
|
|
dpmt.Name, len(c.SubnetNames))
|
|
}
|
|
|
|
if c.SubnetNames[0] != "subnet1" {
|
|
return fmt.Errorf("Bad subnet: %s", c.SubnetNames[0])
|
|
}
|
|
|
|
if c.NetworkSecurityGroup != "terraform-security-group2" {
|
|
return fmt.Errorf("Bad security group: %s", c.NetworkSecurityGroup)
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func testAccCheckAzureInstanceDestroy(s *terraform.State) error {
|
|
mc := testAccProvider.Meta().(*Client).mgmtClient
|
|
|
|
for _, rs := range s.RootModule().Resources {
|
|
if rs.Type != "azure_instance" {
|
|
continue
|
|
}
|
|
|
|
if rs.Primary.ID == "" {
|
|
return fmt.Errorf("No instance ID is set")
|
|
}
|
|
|
|
_, err := hostedservice.NewClient(mc).GetHostedService(rs.Primary.ID)
|
|
if err == nil {
|
|
return fmt.Errorf("Resource %s still exists", rs.Primary.ID)
|
|
}
|
|
|
|
if !management.IsResourceNotFoundError(err) {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
var testAccAzureInstance_basic = fmt.Sprintf(`
|
|
resource "azure_instance" "foo" {
|
|
name = "terraform-test"
|
|
image = "Ubuntu Server 14.04 LTS"
|
|
size = "Basic_A1"
|
|
storage = "%s"
|
|
location = "West US"
|
|
username = "terraform"
|
|
password = "Pass!admin123"
|
|
|
|
endpoint {
|
|
name = "SSH"
|
|
protocol = "tcp"
|
|
public_port = 22
|
|
private_port = 22
|
|
}
|
|
}`, os.Getenv("AZURE_STORAGE"))
|
|
|
|
var testAccAzureInstance_advanced = fmt.Sprintf(`
|
|
resource "azure_virtual_network" "foo" {
|
|
name = "terraform-vnet"
|
|
address_space = ["10.1.2.0/24"]
|
|
location = "West US"
|
|
|
|
subnet {
|
|
name = "subnet1"
|
|
address_prefix = "10.1.2.0/25"
|
|
}
|
|
|
|
subnet {
|
|
name = "subnet2"
|
|
address_prefix = "10.1.2.128/25"
|
|
}
|
|
}
|
|
|
|
resource "azure_security_group" "foo" {
|
|
name = "terraform-security-group1"
|
|
location = "West US"
|
|
|
|
rule {
|
|
name = "rdp"
|
|
priority = 101
|
|
source_cidr = "*"
|
|
source_port = "*"
|
|
destination_cidr = "*"
|
|
destination_port = 3389
|
|
protocol = "TCP"
|
|
}
|
|
}
|
|
|
|
resource "azure_instance" "foo" {
|
|
name = "terraform-test1"
|
|
image = "Windows Server 2012 R2 Datacenter, April 2015"
|
|
size = "Basic_A1"
|
|
storage = "%s"
|
|
location = "West US"
|
|
time_zone = "America/Los_Angeles"
|
|
subnet = "subnet1"
|
|
virtual_network = "${azure_virtual_network.foo.name}"
|
|
security_group = "${azure_security_group.foo.name}"
|
|
username = "terraform"
|
|
password = "Pass!admin123"
|
|
|
|
endpoint {
|
|
name = "RDP"
|
|
protocol = "tcp"
|
|
public_port = 3389
|
|
private_port = 3389
|
|
}
|
|
}`, os.Getenv("AZURE_STORAGE"))
|
|
|
|
var testAccAzureInstance_update = fmt.Sprintf(`
|
|
resource "azure_virtual_network" "foo" {
|
|
name = "terraform-vnet"
|
|
address_space = ["10.1.2.0/24"]
|
|
location = "West US"
|
|
|
|
subnet {
|
|
name = "subnet1"
|
|
address_prefix = "10.1.2.0/25"
|
|
}
|
|
|
|
subnet {
|
|
name = "subnet2"
|
|
address_prefix = "10.1.2.128/25"
|
|
}
|
|
}
|
|
|
|
resource "azure_security_group" "foo" {
|
|
name = "terraform-security-group1"
|
|
location = "West US"
|
|
|
|
rule {
|
|
name = "rdp"
|
|
priority = 101
|
|
source_cidr = "*"
|
|
source_port = "*"
|
|
destination_cidr = "*"
|
|
destination_port = 3389
|
|
protocol = "TCP"
|
|
}
|
|
}
|
|
|
|
resource "azure_security_group" "bar" {
|
|
name = "terraform-security-group2"
|
|
location = "West US"
|
|
|
|
rule {
|
|
name = "rdp"
|
|
priority = 101
|
|
source_cidr = "192.168.0.0/24"
|
|
source_port = "*"
|
|
destination_cidr = "*"
|
|
destination_port = 3389
|
|
protocol = "TCP"
|
|
}
|
|
}
|
|
|
|
resource "azure_instance" "foo" {
|
|
name = "terraform-test1"
|
|
image = "Windows Server 2012 R2 Datacenter, April 2015"
|
|
size = "Basic_A2"
|
|
storage = "%s"
|
|
location = "West US"
|
|
time_zone = "America/Los_Angeles"
|
|
subnet = "subnet1"
|
|
virtual_network = "${azure_virtual_network.foo.name}"
|
|
security_group = "${azure_security_group.bar.name}"
|
|
username = "terraform"
|
|
password = "Pass!admin123"
|
|
|
|
endpoint {
|
|
name = "RDP"
|
|
protocol = "tcp"
|
|
public_port = 3389
|
|
private_port = 3389
|
|
}
|
|
|
|
endpoint {
|
|
name = "WINRM"
|
|
protocol = "tcp"
|
|
public_port = 5985
|
|
private_port = 5985
|
|
}
|
|
}`, os.Getenv("AZURE_STORAGE"))
|