Function TestRPCModeInvokeReaderErrorPropogated

lambda/rpc_function_test.go:274–288

(t *testing.T)

}

func TestRPCModeInvokeReaderErrorPropogated(t *testing.T) {
	handlerResource := &closeableResponse{
		reader: &readerError{errors.New("yolo")},
		closed: false,
	}
	srv := NewFunction(newHandler(func() (interface{}, error) {
		return handlerResource, nil
	}))
	var response messages.InvokeResponse
	err := srv.Invoke(&messages.InvokeRequest{}, &response)
	require.NoError(t, err)
	assert.Equal(t, "", string(response.Payload))
	assert.Equal(t, "yolo", response.Error.Message)
	assert.True(t, handlerResource.closed)
}