Function toHTTPResponseWithFormat

modules/frontend/combiner/common_test.go:290–316

Helper functions for creating test responses with different marshaling formats

(t *testing.T, pb proto.Message, statusCode int, responseData any, format api.MarshallingFormat)

// Helper functions for creating test responses with different marshaling formats
func toHTTPResponseWithFormat(t *testing.T, pb proto.Message, statusCode int, responseData any, format api.MarshallingFormat) PipelineResponse {
	var body []byte
	var err error

	if pb != nil {
		if format == api.MarshallingFormatProtobuf {
			body, err = proto.Marshal(pb)
			require.NoError(t, err)
		} else {
			m := jsonpb.Marshaler{}
			bodyStr, marshalErr := m.MarshalToString(pb)
			require.NoError(t, marshalErr)
			body = []byte(bodyStr)
		}
	}

	return &testPipelineResponse{
		responseData: responseData,
		r: &http.Response{
			Body:       io.NopCloser(strings.NewReader(string(body))),
			StatusCode: statusCode,
			Header: http.Header{
				api.HeaderContentType: {string(format)},
			},
		},
	}
}

func toHTTPResponse(t *testing.T, pb proto.Message, statusCode int) PipelineResponse {
	return toHTTPResponseWithFormat(t, pb, statusCode, nil, api.HeaderAcceptJSON)