From 94e13671726abbcf766f9b4aacd2ee04de59dcbd Mon Sep 17 00:00:00 2001 From: Aleksandr Nogikh Date: Fri, 24 Jan 2025 17:17:53 +0100 Subject: pkg/rpcserver: refactor to remove Fatalf calls Apply necessary changes to pkg/flatrpc and pkg/manager as well. --- pkg/rpcserver/rpcserver_test.go | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'pkg/rpcserver/rpcserver_test.go') diff --git a/pkg/rpcserver/rpcserver_test.go b/pkg/rpcserver/rpcserver_test.go index a885ad720..0c4984e93 100644 --- a/pkg/rpcserver/rpcserver_test.go +++ b/pkg/rpcserver/rpcserver_test.go @@ -4,10 +4,12 @@ package rpcserver import ( + "context" "net" "testing" "github.com/stretchr/testify/assert" + "golang.org/x/sync/errgroup" "github.com/google/syzkaller/pkg/flatrpc" "github.com/google/syzkaller/pkg/mgrconfig" @@ -212,7 +214,11 @@ func TestHandleConn(t *testing.T) { serv.CreateInstance(1, injectExec, nil) go flatrpc.Send(clientConn, tt.req) - serv.handleConn(serverConn) + var eg errgroup.Group + serv.handleConn(context.Background(), &eg, serverConn) + if err := eg.Wait(); err != nil { + t.Fatal(err) + } }) } } -- cgit mrf-deployment