servicenetworking - fixup test checks (hashicorp#23410)

opentofu · Sep 28, 2023 · 7d44f7e · 7d44f7e
1 parent f93d250
commit 7d44f7e
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/internal/services/servicenetworking/application_load_balancer_resource_test.go b/internal/services/servicenetworking/application_load_balancer_resource_test.go
@@ -3,6 +3,7 @@ package servicenetworking_test
 import (
 	"context"
 	"fmt"
+	"regexp"
 	"testing"
 
 	"github.com/hashicorp/go-azure-helpers/lang/pointer"
@@ -41,7 +42,7 @@ func TestAccApplicationLoadBalancer_basic(t *testing.T) {
 			Config: r.basic(data),
 			Check: acceptance.ComposeTestCheckFunc(
 				check.That(data.ResourceName).ExistsInAzure(r),
-				check.That(data.ResourceName).Key("configuration_endpoint.#").HasValue("1"),
+				check.That(data.ResourceName).Key("primary_configuration_endpoint").MatchesRegex(regexp.MustCompile(`^.+\.alb.azure.com$`)),
 			),
 		},
 		data.ImportStep(),
@@ -57,7 +58,7 @@ func TestAccApplicationLoadBalancer_complete(t *testing.T) {
 			Config: r.complete(data),
 			Check: acceptance.ComposeTestCheckFunc(
 				check.That(data.ResourceName).ExistsInAzure(r),
-				check.That(data.ResourceName).Key("primary_configuration_endpoint").HasValue("1"),
+				check.That(data.ResourceName).Key("primary_configuration_endpoint").MatchesRegex(regexp.MustCompile(`^.+\.alb.azure.com$`)),
 			),
 		},
 		data.ImportStep(),
@@ -73,15 +74,15 @@ func TestAccApplicationLoadBalancer_update(t *testing.T) {
 			Config: r.basic(data),
 			Check: acceptance.ComposeTestCheckFunc(
 				check.That(data.ResourceName).ExistsInAzure(r),
-				check.That(data.ResourceName).Key("configuration_endpoint.#").HasValue("1"),
+				check.That(data.ResourceName).Key("primary_configuration_endpoint").MatchesRegex(regexp.MustCompile(`^.+\.alb.azure.com$`)),
 			),
 		},
 		data.ImportStep(),
 		{
 			Config: r.complete(data),
 			Check: acceptance.ComposeTestCheckFunc(
 				check.That(data.ResourceName).ExistsInAzure(r),
-				check.That(data.ResourceName).Key("configuration_endpoint.#").HasValue("1"),
+				check.That(data.ResourceName).Key("primary_configuration_endpoint").MatchesRegex(regexp.MustCompile(`^.+\.alb.azure.com$`)),
 			),
 		},
 		data.ImportStep(),
@@ -97,7 +98,7 @@ func TestAccApplicationLoadBalancer_requiresImport(t *testing.T) {
 			Config: r.basic(data),
 			Check: acceptance.ComposeTestCheckFunc(
 				check.That(data.ResourceName).ExistsInAzure(r),
-				check.That(data.ResourceName).Key("configuration_endpoint.#").HasValue("1"),
+				check.That(data.ResourceName).Key("primary_configuration_endpoint").MatchesRegex(regexp.MustCompile(`^.+\.alb.azure.com$`)),
 			),
 		},
 		data.RequiresImportErrorStep(r.requiresImport),