Adds headers field for CheckHealthRequest

backend/convert_from_protobuf.go

@@ -172,8 +172,13 @@ func (f ConvertFromProtobuf) CallResourceResponse(protoResp *pluginv2.CallResour
 
 // CheckHealthRequest converts protobuf version of a CheckHealthRequest to the SDK version.
 func (f ConvertFromProtobuf) CheckHealthRequest(protoReq *pluginv2.CheckHealthRequest) *CheckHealthRequest {
+	if protoReq.Headers == nil {
+		protoReq.Headers = map[string]string{}
+	}
+
 	return &CheckHealthRequest{
 		PluginContext: f.PluginContext(protoReq.PluginContext),
+		Headers:       protoReq.Headers,
 	}
 }
 

backend/convert_from_protobuf_test.go

@@ -429,3 +429,31 @@ func TestConvertFromProtobufQueryDataRequest(t *testing.T) {
 	//
 	require.Equal(t, requireCounter.Count, sdkWalker.FieldCount-6, "untested fields in conversion") // -6 Struct Fields
 }
+
+func TestConvertFromProtobufCheckHealthRequest(t *testing.T) {
+	t.Run("Should convert provided headers", func(t *testing.T) {
+		protoReq := &pluginv2.CheckHealthRequest{
+			PluginContext: protoPluginContext,
+			Headers: map[string]string{
+				"foo": "fooVal",
+				"bar": "barVal",
+			},
+		}
+
+		req := FromProto().CheckHealthRequest(protoReq)
+		require.NotNil(t, req)
+		require.NotNil(t, req.PluginContext)
+		require.Equal(t, protoPluginContext.OrgId, req.PluginContext.OrgID)
+		require.Equal(t, protoReq.Headers, req.Headers)
+	})
+
+	t.Run("Should handle nil-provided headers", func(t *testing.T) {
+		protoReq := &pluginv2.CheckHealthRequest{
+			PluginContext: protoPluginContext,
+		}
+
+		req := FromProto().CheckHealthRequest(protoReq)
+		require.NotNil(t, req)
+		require.NotNil(t, req.Headers)
+	})
+}

backend/diagnostics.go

@@ -54,6 +54,7 @@ func (hs HealthStatus) String() string {
 // CheckHealthRequest contains the healthcheck request
 type CheckHealthRequest struct {
 	PluginContext PluginContext
+	Headers       map[string]string
 }
 
 // CheckHealthResult contains the healthcheck response