Update validate test to add case for marked count value

The diff here is largely a refactor to allow the test to run multiple cases

terraform/node_resource_validate_test.go

@@ -202,6 +202,7 @@ func TestNodeValidatableResource_ValidateResource_managedResource(t *testing.T)
 }
 
 func TestNodeValidatableResource_ValidateResource_managedResourceCount(t *testing.T) {
+	// Setup
 	mp := simpleMockProvider()
 	mp.ValidateResourceTypeConfigFn = func(req providers.ValidateResourceTypeConfigRequest) providers.ValidateResourceTypeConfigResponse {
 		if got, want := req.TypeName, "test_object"; got != want {
@@ -214,11 +215,33 @@ func TestNodeValidatableResource_ValidateResource_managedResourceCount(t *testin
 	}
 
 	p := providers.Interface(mp)
+
+	ctx := &MockEvalContext{}
+	ctx.installSimpleEval()
+	ctx.ProviderSchemaSchema = mp.GetSchemaReturn
+	ctx.ProviderProvider = p
+
+	tests := []struct {
+		name  string
+		count hcl.Expression
+	}{
+		{
+			"simple count",
+			hcltest.MockExprLiteral(cty.NumberIntVal(2)),
+		},
+		{
+			"marked count value",
+			hcltest.MockExprLiteral(cty.NumberIntVal(3).Mark("marked")),
+		},
+	}
+
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
 			rc := &configs.Resource{
 				Mode:  addrs.ManagedResourceMode,
 				Type:  "test_object",
 				Name:  "foo",
-		Count: hcltest.MockExprLiteral(cty.NumberIntVal(2)),
+				Count: test.count,
 				Config: configs.SynthBody("", map[string]cty.Value{
 					"test_string": cty.StringVal("bar"),
 				}),
@@ -231,11 +254,6 @@ func TestNodeValidatableResource_ValidateResource_managedResourceCount(t *testin
 				},
 			}
 
-	ctx := &MockEvalContext{}
-	ctx.installSimpleEval()
-	ctx.ProviderSchemaSchema = mp.GetSchemaReturn
-	ctx.ProviderProvider = p
-
 			diags := node.validateResource(ctx)
 			if diags.HasErrors() {
 				t.Fatalf("err: %s", diags.Err())
@@ -244,6 +262,8 @@ func TestNodeValidatableResource_ValidateResource_managedResourceCount(t *testin
 			if !mp.ValidateResourceTypeConfigCalled {
 				t.Fatal("Expected ValidateResourceTypeConfig to be called, but it was not!")
 			}
+		})
+	}
 }
 
 func TestNodeValidatableResource_ValidateResource_dataSource(t *testing.T) {