diff --git a/internal/lambda/interop/model.go b/internal/lambda/interop/model.go index d4c6d9ec..60aba410 100644 --- a/internal/lambda/interop/model.go +++ b/internal/lambda/interop/model.go @@ -359,7 +359,7 @@ type ErrorResponseTooLargeDI struct { // ErrorResponseTooLarge is returned when response provided by Runtime does not fit into shared memory buffer func (s *ErrorResponseTooLarge) Error() string { - return fmt.Sprintf("Response payload size (%d bytes) exceeded maximum allowed payload size (%d bytes).", s.ResponseSize, s.MaxResponseSize) + return fmt.Sprintf("Response payload size exceeded maximum allowed payload size (%d bytes).", s.MaxResponseSize) } // AsErrorResponse generates ErrorInvokeResponse from ErrorResponseTooLarge diff --git a/internal/lambda/rapi/handler/invocationresponse_test.go b/internal/lambda/rapi/handler/invocationresponse_test.go index 9ce0a3d9..628208f5 100644 --- a/internal/lambda/rapi/handler/invocationresponse_test.go +++ b/internal/lambda/rapi/handler/invocationresponse_test.go @@ -64,7 +64,7 @@ func TestResponseTooLarge(t *testing.T) { assert.NotNil(t, errorResponse) assert.Nil(t, flowTest.InteropServer.Response) assert.Equal(t, fatalerror.FunctionOversizedResponse, errorResponse.FunctionError.Type) - assert.Equal(t, "Response payload size (6291557 bytes) exceeded maximum allowed payload size (6291556 bytes).", errorResponse.FunctionError.Message) + assert.Equal(t, "Response payload size exceeded maximum allowed payload size (6291556 bytes).", errorResponse.FunctionError.Message) var errorPayload map[string]interface{} assert.NoError(t, json.Unmarshal(errorResponse.Payload, &errorPayload)) diff --git a/internal/lambda/rapi/rapi_fuzz_test.go b/internal/lambda/rapi/rapi_fuzz_test.go index 46f99f9f..b6651bca 100644 --- a/internal/lambda/rapi/rapi_fuzz_test.go +++ b/internal/lambda/rapi/rapi_fuzz_test.go @@ -340,7 +340,7 @@ func assertInvocationResponseTooLarge(t *testing.T, responseRecorder *httptest.R assert.NotNil(t, errorResponse) assert.Nil(t, flowTest.InteropServer.Response) assert.Equal(t, fatalerror.FunctionOversizedResponse, errorResponse.FunctionError.Type) - assert.Equal(t, fmt.Sprintf("Response payload size (%v bytes) exceeded maximum allowed payload size (6291556 bytes).", len(responseBody)), errorResponse.FunctionError.Message) + assert.Equal(t, "Response payload size exceeded maximum allowed payload size (6291556 bytes).", errorResponse.FunctionError.Message) var errorPayload map[string]interface{} assert.NoError(t, json.Unmarshal(errorResponse.Payload, &errorPayload)) diff --git a/internal/lambda/rapidcore/server.go b/internal/lambda/rapidcore/server.go index c0c24e34..b82d8276 100644 --- a/internal/lambda/rapidcore/server.go +++ b/internal/lambda/rapidcore/server.go @@ -658,7 +658,7 @@ func (s *Server) Invoke(responseWriter http.ResponseWriter, invoke *interop.Invo // The logic would be almost identical, except that init failures could manifest // through return values of FastInvoke and not Reserve() - reserveResp, err := s.Reserve(invoke.ID, "", "") + reserveResp, err := s.Reserve(invoke.ID, invoke.TraceID, invoke.LambdaSegmentID) if err != nil { log.Infof("ReserveFailed: %s", err) }