add v4 examples to acc tests

Author: lonegunmanb

test/e2e/terraform_aks_test.go

@@ -4,18 +4,17 @@ import (
 	"fmt"
 	"io"
 	"os"
+	"path/filepath"
 	"regexp"
 	"strings"
 	"testing"
 	"time"
 
-	"github.com/hashicorp/go-retryablehttp"
-
-	"github.com/stretchr/testify/require"
-
 	test_helper "github.com/Azure/terraform-module-test-helper"
 	"github.com/gruntwork-io/terratest/modules/terraform"
+	"github.com/hashicorp/go-retryablehttp"
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 )
 
 func TestExamplesStartup(t *testing.T) {
@@ -178,6 +177,41 @@ func TestExamples_applicationGatewayIngress(t *testing.T) {
 	}
 }
 
+func TestExamplesForV4(t *testing.T) {
+	managedIdentityId := os.Getenv("MSI_ID")
+	if managedIdentityId != "" {
+		t.Setenv("TF_VAR_managed_identity_principal_id", managedIdentityId)
+	}
+	tmp, err := os.MkdirTemp("", "")
+	require.NoError(t, err)
+	defer func() {
+		_ = os.RemoveAll(tmp)
+	}()
+	tfvars := filepath.Join(tmp, "terraform.tfvars")
+	require.NoError(t, os.WriteFile(tfvars, []byte(`
+	client_id = ""
+	client_secret = ""
+`), 0o644))
+	t.Setenv("TF_VAR_client_id", "")
+	t.Setenv("TF_VAR_client_secret", "")
+	examples, err := os.ReadDir("../../examples")
+	require.NoError(t, err)
+	for _, example := range examples {
+		if !example.IsDir() {
+			continue
+		}
+		if !strings.HasSuffix(example.Name(), "_v4") {
+			continue
+		}
+		t.Run(example.Name(), func(t *testing.T) {
+			test_helper.RunE2ETest(t, fmt.Sprintf("../../"), fmt.Sprintf("examples/%s", example.Name()), terraform.Options{
+				Upgrade:  true,
+				VarFiles: []string{tfvars},
+			}, nil)
+		})
+	}
+}
+
 func getHTML(url string) (string, error) {
 	client := retryablehttp.NewClient()
 	client.RetryMax = 10

test/upgrade/upgrade_test.go

@@ -3,9 +3,13 @@ package upgrade
 import (
 	"fmt"
 	"os"
+	"path/filepath"
+	"strings"
 	"testing"
 	"time"
 
+	"github.com/stretchr/testify/require"
+
 	test_helper "github.com/Azure/terraform-module-test-helper"
 	"github.com/gruntwork-io/terratest/modules/terraform"
 )
@@ -148,3 +152,45 @@ func TestExampleUpgrade_applicationGw(t *testing.T) {
 		})
 	}
 }
+
+func TestExamplesForV4(t *testing.T) {
+	managedIdentityId := os.Getenv("MSI_ID")
+	if managedIdentityId != "" {
+		t.Setenv("TF_VAR_managed_identity_principal_id", managedIdentityId)
+	}
+	tmp, err := os.MkdirTemp("", "")
+	require.NoError(t, err)
+	defer func() {
+		_ = os.RemoveAll(tmp)
+	}()
+	tfvars := filepath.Join(tmp, "terraform.tfvars")
+	require.NoError(t, os.WriteFile(tfvars, []byte(`
+	client_id = ""
+	client_secret = ""
+`), 0o644))
+	t.Setenv("TF_VAR_client_id", "")
+	t.Setenv("TF_VAR_client_secret", "")
+	examples, err := os.ReadDir("../../examples")
+	require.NoError(t, err)
+	currentRoot, err := test_helper.GetCurrentModuleRootPath()
+	if err != nil {
+		t.FailNow()
+	}
+	currentMajorVersion, err := test_helper.GetCurrentMajorVersionFromEnv()
+	if err != nil {
+		t.FailNow()
+	}
+	for _, example := range examples {
+		if !example.IsDir() {
+			continue
+		}
+		if strings.HasSuffix(example.Name(), "_v4") {
+			continue
+		}
+		t.Run(example.Name(), func(t *testing.T) {
+			test_helper.ModuleUpgradeTest(t, "Azure", "terraform-azurerm-aks", fmt.Sprintf("examples/%s", example.Name()), currentRoot, terraform.Options{
+				VarFiles: []string{tfvars},
+			}, currentMajorVersion)
+		})
+	}
+}