github.com/darmach/terratest@v0.34.8-0.20210517103231-80931f95e3ff/test/azure/terraform_azure_loadbalancer_example_test.go (about)

     1  // +build azure
     2  
     3  // NOTE: We use build tags to differentiate azure testing because we currently do not have azure access setup for
     4  // CircleCI.
     5  
     6  package test
     7  
     8  import (
     9  	"testing"
    10  
    11  	"github.com/gruntwork-io/terratest/modules/azure"
    12  	"github.com/gruntwork-io/terratest/modules/random"
    13  	"github.com/gruntwork-io/terratest/modules/terraform"
    14  	"github.com/stretchr/testify/assert"
    15  )
    16  
    17  func TestTerraformAzureLoadBalancerExample(t *testing.T) {
    18  	t.Parallel()
    19  
    20  	// subscriptionID is overridden by the environment variable "ARM_SUBSCRIPTION_ID"
    21  	subscriptionID := ""
    22  	uniquePostfix := random.UniqueId()
    23  	privateIPForLB02 := "10.200.2.10"
    24  
    25  	// Configure Terraform setting up a path to Terraform code.
    26  	terraformOptions := &terraform.Options{
    27  		// The path to where our Terraform code is located
    28  		TerraformDir: "../../examples/azure/terraform-azure-loadbalancer-example",
    29  
    30  		// Variables to pass to our Terraform code using -var options
    31  		Vars: map[string]interface{}{
    32  			"postfix":       uniquePostfix,
    33  			"lb_private_ip": privateIPForLB02,
    34  			// "location": "East US",
    35  		},
    36  	}
    37  
    38  	// At the end of the test, run `terraform destroy` to clean up any resources that were created.
    39  	defer terraform.Destroy(t, terraformOptions)
    40  
    41  	// Run `terraform init` and `terraform apply`. Fail the test if there are any errors.
    42  	terraform.InitAndApply(t, terraformOptions)
    43  
    44  	// Run `terraform output` to get the values of output variables
    45  	resourceGroupName := terraform.Output(t, terraformOptions, "resource_group_name")
    46  	expectedLBPublicName := terraform.Output(t, terraformOptions, "lb_public_name")
    47  	expectedLBPrivateName := terraform.Output(t, terraformOptions, "lb_private_name")
    48  	expectedLBNoFEConfigName := terraform.Output(t, terraformOptions, "lb_default_name")
    49  	expectedLBPublicFeConfigName := terraform.Output(t, terraformOptions, "lb_public_fe_config_name")
    50  	expectedLBPrivateFeConfigName := terraform.Output(t, terraformOptions, "lb_private_fe_config_static_name")
    51  	expectedLBPrivateIP := terraform.Output(t, terraformOptions, "lb_private_ip_static")
    52  
    53  	actualLBDoesNotExist := azure.LoadBalancerExists(t, "negative-test", resourceGroupName, subscriptionID)
    54  	assert.False(t, actualLBDoesNotExist)
    55  
    56  	t.Run("LoadBalancer_Public", func(t *testing.T) {
    57  		// Check Public Load Balancer exists.
    58  		actualLBPublicExists := azure.LoadBalancerExists(t, expectedLBPublicName, resourceGroupName, subscriptionID)
    59  		assert.True(t, actualLBPublicExists)
    60  
    61  		// Check Frontend Configuration for Load Balancer.
    62  		actualLBPublicFeConfigNames := azure.GetLoadBalancerFrontendIPConfigNames(t, expectedLBPublicName, resourceGroupName, subscriptionID)
    63  		assert.Contains(t, actualLBPublicFeConfigNames, expectedLBPublicFeConfigName)
    64  
    65  		// Check Frontend Configuration Public Address and Public IP assignment
    66  		actualLBPublicIPAddress, actualLBPublicIPType := azure.GetIPOfLoadBalancerFrontendIPConfig(t, expectedLBPublicFeConfigName, expectedLBPublicName, resourceGroupName, subscriptionID)
    67  		assert.NotEmpty(t, actualLBPublicIPAddress)
    68  		assert.Equal(t, azure.PublicIP, actualLBPublicIPType)
    69  	})
    70  
    71  	t.Run("LoadBalancer_Private", func(t *testing.T) {
    72  		// Check Private Load Balancer exists.
    73  		actualLBPrivateExists := azure.LoadBalancerExists(t, expectedLBPrivateName, resourceGroupName, subscriptionID)
    74  		assert.True(t, actualLBPrivateExists)
    75  
    76  		// Check Frontend Configuration for Load Balancer.
    77  		actualLBPrivateFeConfigNames := azure.GetLoadBalancerFrontendIPConfigNames(t, expectedLBPrivateName, resourceGroupName, subscriptionID)
    78  		assert.Equal(t, 2, len(actualLBPrivateFeConfigNames))
    79  		assert.Contains(t, actualLBPrivateFeConfigNames, expectedLBPrivateFeConfigName)
    80  
    81  		// Check Frontend Configuration Private IP Type and Address.
    82  		actualLBPrivateIPAddress, actualLBPrivateIPType := azure.GetIPOfLoadBalancerFrontendIPConfig(t, expectedLBPrivateFeConfigName, expectedLBPrivateName, resourceGroupName, subscriptionID)
    83  		assert.NotEmpty(t, actualLBPrivateIPAddress)
    84  		assert.Equal(t, expectedLBPrivateIP, actualLBPrivateIPAddress)
    85  		assert.Equal(t, azure.PrivateIP, actualLBPrivateIPType)
    86  	})
    87  
    88  	t.Run("LoadBalancer_Default", func(t *testing.T) {
    89  		// Check No Frontend Config Load Balancer exists.
    90  		actualLBNoFEConfigExists := azure.LoadBalancerExists(t, expectedLBNoFEConfigName, resourceGroupName, subscriptionID)
    91  		assert.True(t, actualLBNoFEConfigExists)
    92  
    93  		// Check for No Frontend Configuration for Load Balancer.
    94  		actualLBNoFEConfigFeConfigNames := azure.GetLoadBalancerFrontendIPConfigNames(t, expectedLBNoFEConfigName, resourceGroupName, subscriptionID)
    95  		assert.Equal(t, 0, len(actualLBNoFEConfigFeConfigNames))
    96  	})
    97  }