From 4910018031a723c0c222e9dc80f420b828eef225 Mon Sep 17 00:00:00 2001 From: Yashash H L Date: Fri, 13 Sep 2024 22:05:21 +0530 Subject: [PATCH] fix unit tests Signed-off-by: Yashash H L --- pkg/mapper/server_test.go | 2 +- pkg/sourcer/service_test.go | 13 +++++++++++++ 2 files changed, 14 insertions(+), 1 deletion(-) diff --git a/pkg/mapper/server_test.go b/pkg/mapper/server_test.go index 8c661ff0..baa94d4a 100644 --- a/pkg/mapper/server_test.go +++ b/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()), ) diff --git a/pkg/sourcer/service_test.go b/pkg/sourcer/service_test.go index ae484fec..b16b4792 100644 --- a/pkg/sourcer/service_test.go +++ b/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 }