fix unit tests

Signed-off-by: Yashash H L <[email protected]>

pkg/mapper/server_test.go

@@ -73,7 +73,7 @@ func TestMapServer_GracefulShutdown(t *testing.T) {
 	time.Sleep(10 * time.Millisecond)
 
 	// create a client
-	conn, err := grpc.Dial(
+	conn, err := grpc.NewClient(
 		"unix://"+socketFile.Name(),
 		grpc.WithTransportCredentials(insecure.NewCredentials()),
 	)

pkg/sourcer/service_test.go

@@ -11,6 +11,7 @@ import (
 
 	"github.com/stretchr/testify/assert"
 	"google.golang.org/grpc"
+	"google.golang.org/grpc/metadata"
 	"google.golang.org/protobuf/types/known/emptypb"
 	"google.golang.org/protobuf/types/known/timestamppb"
 
@@ -61,6 +62,10 @@ type ReadFnServerTest struct {
 	index int
 }
 
+func (t *ReadFnServerTest) SendHeader(metadata.MD) error {
+	return nil
+}
+
 func (t *ReadFnServerTest) Recv() (*sourcepb.ReadRequest, error) {
 	if t.index >= len(t.requests) {
 		return nil, io.EOF
@@ -96,6 +101,10 @@ type ReadFnServerErrTest struct {
 	grpc.ServerStream
 }
 
+func (te *ReadFnServerErrTest) SendHeader(metadata.MD) error {
+	return nil
+}
+
 func (te *ReadFnServerErrTest) Recv() (*sourcepb.ReadRequest, error) {
 	return nil, fmt.Errorf("recv error")
 }
@@ -146,6 +155,10 @@ func NewAckFnServerTest(
 	}
 }
 
+func (a *AckFnServerTest) SendHeader(metadata.MD) error {
+	return nil
+}
+
 func (a *AckFnServerTest) SendAndClose(*sourcepb.AckResponse) error {
 	return nil
 }