From 6e9ac4418d9dd2fff23902400d990663c0411572 Mon Sep 17 00:00:00 2001
From: Wilken Rivera <wilken@hashicorp.com>
Date: Fri, 21 Apr 2023 14:36:30 -0400
Subject: [PATCH] Add recover to fail datasource test on cty panic

---
 rpc/datasource_test.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/rpc/datasource_test.go b/rpc/datasource_test.go
index bc98a2780..180adae2f 100644
--- a/rpc/datasource_test.go
+++ b/rpc/datasource_test.go
@@ -66,6 +66,12 @@ func TestDatasource(t *testing.T) {
 	d.outputSpec = map[string]hcldec.Spec{
 		"foo": &hcldec.AttrSpec{Name: "foo", Type: cty.String, Required: false},
 	}
+
+	defer func() {
+		if r := recover(); r != nil {
+			t.Errorf("Call to ConfigSpec for datasource panicked: %v", r)
+		}
+	}()
 	spec := dsClient.OutputSpec()
 	if !reflect.DeepEqual(spec, d.outputSpec) {
 		t.Fatalf("unknown outputSpec value: %#v", spec)