chore: added jsonrpc server test

Author: ascandone

internal/jsonrpc2/jsonrpc2.go

@@ -65,7 +65,9 @@ func HandleNotification[Params any](s *Server, method string, handler func(param
 }
 
 // Send a json rpc request and wait for the response. Thread safe.
-func SendRequest(s *Server, method string, params any) (any, error) {
+//
+// TODO change the API in order to make sure SendRequest is never called before .Listen() is called
+func SendRequest[Res any](s *Server, method string, params any) (*Res, error) {
 	bytes, err := json.Marshal(params)
 	if err != nil {
 		return nil, err
@@ -91,7 +93,9 @@ func SendRequest(s *Server, method string, params any) (any, error) {
 	delete(s.pendingRequests, freshId)
 	s.pendingRequestMu.Unlock()
 
-	return response, nil
+	var res Res
+	json.Unmarshal(response.Result, &res)
+	return &res, nil
 }
 
 // Send a json rpc request and wait for the message to be sent. Thread safe

internal/jsonrpc2/jsonrpc2_test.go

@@ -0,0 +1,34 @@
+package jsonrpc2_test
+
+import (
+	"testing"
+
+	"github.com/formancehq/numscript/internal/jsonrpc2"
+	"github.com/stretchr/testify/require"
+)
+
+func TestHandleRequest(t *testing.T) {
+	type SumParams struct {
+		X int `json:"x"`
+		Y int `json:"y"`
+	}
+
+	in := make(chan jsonrpc2.Message)
+	out := make(chan jsonrpc2.Message)
+
+	server := jsonrpc2.NewServer(jsonrpc2.NewChanObjStream(in, out))
+	jsonrpc2.HandleRequest(server, "sum", func(p SumParams) any {
+		return p.X + p.Y
+	})
+	go server.Listen()
+	defer server.Close()
+
+	client := jsonrpc2.NewServer(jsonrpc2.NewChanObjStream(out, in))
+	go client.Listen()
+	defer server.Close()
+
+	res, err := jsonrpc2.SendRequest[int](client, "sum", SumParams{X: 100, Y: 42})
+	require.Nil(t, err)
+
+	require.Equal(t, 142, *res)
+}