From 599e326718b25f8877bad8e5c148997ccf1e15fc Mon Sep 17 00:00:00 2001
From: Rajagopalan Subrahmanian <rsubrahmania@juniper.net>
Date: Tue, 21 Nov 2023 11:25:33 -0500
Subject: [PATCH] add property set tests

---
 .../resource_datacenter_property_set_test.go  | 81 +++++++++++++++++++
 1 file changed, 81 insertions(+)
 create mode 100644 apstra/resource_datacenter_property_set_test.go

diff --git a/apstra/resource_datacenter_property_set_test.go b/apstra/resource_datacenter_property_set_test.go
new file mode 100644
index 00000000..050d4d79
--- /dev/null
+++ b/apstra/resource_datacenter_property_set_test.go
@@ -0,0 +1,81 @@
+package tfapstra
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	testutils "github.com/Juniper/terraform-provider-apstra/apstra/test_utils"
+	"github.com/Juniper/terraform-provider-apstra/apstra/utils"
+	"github.com/hashicorp/terraform-plugin-framework/diag"
+	"github.com/hashicorp/terraform-plugin-framework/types"
+	"github.com/hashicorp/terraform-plugin-testing/helper/resource"
+	"testing"
+)
+
+const (
+	data_string = `
+					{
+						"value_str":"str",
+						"value_int":42
+					}`
+	resourceDatacenterPropertySetTemplateHCL = `
+  	resource "apstra_property_set" "test" {
+  		name = "TEST_PS"
+  		data = jsonencode(%s)
+		}
+
+	resource "apstra_datacenter_property_set" "test" {
+  		blueprint_id = "%s"
+        id = apstra_property_set.test.id
+		sync_with_catalog = true
+  	}
+	`
+)
+
+func TestAccResourceDatacenterPropertySet(t *testing.T) {
+	ctx := context.Background()
+
+	// BlueprintA returns a bpClient and the template from which the blueprint was created
+	bpClient, bpDelete, err := testutils.MakeOrFindBlueprint(ctx, "BPA", testutils.BlueprintA)
+
+	if err != nil {
+		t.Fatal(errors.Join(err, bpDelete(ctx)))
+	}
+	defer func() {
+		err = bpDelete(ctx)
+		if err != nil {
+			t.Error(err)
+		}
+	}()
+
+	if err != nil {
+		t.Fatal(errors.Join(err, bpDelete(ctx)))
+	}
+
+	d := diag.Diagnostics{}
+	TestPSData := func(state string) error {
+		if !utils.JSONEqual(types.StringValue(state), types.StringValue(data_string), &d) {
+			return fmt.Errorf("input Data does not match output Input %v. Output %v", data_string, state)
+		}
+		return nil
+	}
+	resource.Test(t, resource.TestCase{
+		// PreCheck:                 setup,
+		ProtoV6ProviderFactories: testAccProtoV6ProviderFactories,
+		Steps: []resource.TestStep{
+			// Import the property set
+			{
+				Config: insecureProviderConfigHCL + fmt.Sprintf(resourceDatacenterPropertySetTemplateHCL,
+					data_string, bpClient.Id()),
+				Check: resource.ComposeAggregateTestCheckFunc(
+					resource.TestCheckResourceAttrSet("apstra_datacenter_property_set.test", "id"),
+					resource.TestCheckResourceAttr("apstra_datacenter_property_set.test", "name", "TEST_PS"),
+					resource.TestCheckResourceAttrWith("apstra_datacenter_property_set.test", "data",
+						TestPSData),
+					resource.TestCheckResourceAttr("apstra_datacenter_property_set.test", "stale", "false"),
+					resource.TestCheckResourceAttr("apstra_datacenter_property_set.test", "sync_required", "false"),
+				),
+			},
+		},
+	})
+}