set anti affinity policy as separate action with Apstra 4.2.0

apstra/data_source_datacenter_blueprint_test.go

@@ -11,6 +11,7 @@ import (
 
 	"github.com/Juniper/apstra-go-sdk/apstra"
 	"github.com/Juniper/apstra-go-sdk/apstra/enum"
+	"github.com/Juniper/terraform-provider-apstra/apstra/compatibility"
 	testutils "github.com/Juniper/terraform-provider-apstra/apstra/test_utils"
 	"github.com/Juniper/terraform-provider-apstra/apstra/utils"
 	"github.com/hashicorp/go-version"
@@ -211,9 +212,13 @@ func TestDatasourceDatacenterBlueprint(t *testing.T) {
 			require.NoError(t, err)
 			t.Cleanup(func() { require.NoError(t, client.DeleteBlueprint(ctx, id)) })
 
+			var bpClient *apstra.TwoStageL3ClosClient
+
 			if tCase.ipv6 {
-				bpClient, err := client.NewTwoStageL3ClosClient(ctx, id)
-				require.NoError(t, err)
+				if bpClient == nil {
+					bpClient, err = client.NewTwoStageL3ClosClient(ctx, id)
+					require.NoError(t, err)
+				}
 
 				fs, err := bpClient.GetFabricSettings(ctx)
 				require.NoError(t, err)
@@ -223,6 +228,17 @@ func TestDatasourceDatacenterBlueprint(t *testing.T) {
 				require.NoError(t, err)
 			}
 
+			// set anti-affinity policy as needed with Apstra 4.2.0
+			if compatibility.TemplateRequiresAntiAffinityPolicy.Check(apiVersion) && tCase.fabricSettings.AntiAffinityPolicy != nil {
+				if bpClient == nil {
+					bpClient, err = client.NewTwoStageL3ClosClient(ctx, id)
+					require.NoError(t, err)
+				}
+
+				err = bpClient.SetFabricSettings(ctx, &tCase.fabricSettings)
+				require.NoError(t, err)
+			}
+
 			// add unpredictable name and ID to existing checks
 			tCase.checks = append(tCase.checks,
 				resource.TestCheckResourceAttr("data.apstra_datacenter_blueprint.test", "id", id.String()),