new test and test update for http server

pkg/eventingest/grpc_forwarder_test.go

@@ -0,0 +1,50 @@
+package eventingest
+
+import (
+	"testing"
+
+	"github.com/google/uuid"
+	"github.com/stretchr/testify/assert"
+	"go.uber.org/mock/gomock"
+	"gofr.dev/pkg/gofr"
+	"google.golang.org/protobuf/types/known/emptypb"
+)
+
+func TestGRPCEventForwarder_ForwardEvent(t *testing.T) {
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+
+	mockClient := NewMockServiceClient(ctrl)
+	forwarder := &GRPCEventForwarder{
+		client: mockClient,
+	}
+
+	ctx := &gofr.Context{}
+	tenantID := uuid.New()
+	customerID := uuid.New()
+	eventData := []byte("test event data")
+
+	expectedRequest := &IngestEventRequest{
+		TenantID:   tenantID,
+		CustomerID: customerID,
+		EventData:  eventData,
+	}
+
+	// Test successful forward
+	mockClient.EXPECT().
+		IngestEvent(gomock.Any(), gomock.Eq(expectedRequest), gomock.Any()).
+		Return(&emptypb.Empty{}, nil).
+		Times(1)
+
+	err := forwarder.ForwardEvent(ctx, tenantID, customerID, eventData)
+	assert.NoError(t, err)
+
+	// Test error case
+	mockClient.EXPECT().
+		IngestEvent(gomock.Any(), gomock.Eq(expectedRequest), gomock.Any()).
+		Return(nil, assert.AnError).
+		Times(1)
+
+	err = forwarder.ForwardEvent(ctx, tenantID, customerID, eventData)
+	assert.Error(t, err)
+}

pkg/eventingest/http_server_test.go

@@ -21,7 +21,7 @@ func TestHandleEvent(t *testing.T) {
 		{
 			name: "Success",
 			setupMocks: func(ctrl *gomock.Controller) (customctx.Context, EventForwarder) {
-				mockCtx := customctx.NewMockInterface(ctrl)
+				mockCtx := customctx.NewMockContext(ctrl)
 				mockEF := NewMockEventForwarder(ctrl)
 
 				tenantID := uuid.New()
@@ -43,7 +43,7 @@ func TestHandleEvent(t *testing.T) {
 		{
 			name: "Missing tenant ID",
 			setupMocks: func(ctrl *gomock.Controller) (customctx.Context, EventForwarder) {
-				mockCtx := customctx.NewMockInterface(ctrl)
+				mockCtx := customctx.NewMockContext(ctrl)
 				mockEF := NewMockEventForwarder(ctrl)
 
 				mockCtx.EXPECT().GetUUIDClaim("tenant_id").Return(uuid.Nil, false)
@@ -56,7 +56,7 @@ func TestHandleEvent(t *testing.T) {
 		{
 			name: "Missing customer ID",
 			setupMocks: func(ctrl *gomock.Controller) (customctx.Context, EventForwarder) {
-				mockCtx := customctx.NewMockInterface(ctrl)
+				mockCtx := customctx.NewMockContext(ctrl)
 				mockEF := NewMockEventForwarder(ctrl)
 
 				mockCtx.EXPECT().GetUUIDClaim("tenant_id").Return(uuid.New(), true)
@@ -70,7 +70,7 @@ func TestHandleEvent(t *testing.T) {
 		{
 			name: "Invalid request body",
 			setupMocks: func(ctrl *gomock.Controller) (customctx.Context, EventForwarder) {
-				mockCtx := customctx.NewMockInterface(ctrl)
+				mockCtx := customctx.NewMockContext(ctrl)
 				mockEF := NewMockEventForwarder(ctrl)
 
 				mockCtx.EXPECT().GetUUIDClaim("tenant_id").Return(uuid.New(), true)
@@ -85,7 +85,7 @@ func TestHandleEvent(t *testing.T) {
 		{
 			name: "Forward event failure",
 			setupMocks: func(ctrl *gomock.Controller) (customctx.Context, EventForwarder) {
-				mockCtx := customctx.NewMockInterface(ctrl)
+				mockCtx := customctx.NewMockContext(ctrl)
 				mockEF := NewMockEventForwarder(ctrl)
 
 				tenantID := uuid.New()
@@ -114,7 +114,7 @@ func TestHandleEvent(t *testing.T) {
 			mockCtx, mockEF := tt.setupMocks(ctrl)
 
 			server := NewHTTPServer(&gofr.App{}, mockEF)
-			result, err := server.HandleEvent(&mockCtx)
+			result, err := server.HandleEvent(mockCtx)
 
 			assert.Equal(t, tt.expectedResult, result)
 			assert.Equal(t, tt.expectedError, err)