Add recover to fail datasource test on cty panic

hashicorp · Apr 25, 2023 · b08070c · b08070c
1 parent 21a1e5e
commit b08070c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/rpc/common_test.go b/rpc/common_test.go
@@ -7,8 +7,6 @@ import (
 	packersdk "github.com/hashicorp/packer-plugin-sdk/packer"
 )
 
-var b testing.B
-
 func TestCommonServer_ConfigSpec(t *testing.T) {
 	tt := []struct {
 		name      string

diff --git 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)