add tests for subscriber handler after seperating them

Author: Umang01-hash

examples/using-subscriber/main.go

@@ -10,41 +10,45 @@ func main() {
 
 	app.Migrate(migrations.All())
 
-	app.Subscribe("products", func(c *gofr.Context) error {
-		var productInfo struct {
-			ProductId string `json:"productId"`
-			Price     string `json:"price"`
-		}
+	app.Subscribe("products", productHandler)
 
-		err := c.Bind(&productInfo)
-		if err != nil {
-			c.Logger.Error(err)
+	app.Subscribe("order-logs", orderHandler)
 
-			return nil
-		}
+	app.Run()
+}
+
+func productHandler(c *gofr.Context) error {
+	var productInfo struct {
+		ProductId string `json:"productId"`
+		Price     string `json:"price"`
+	}
 
-		c.Logger.Info("Received product", productInfo)
+	err := c.Bind(&productInfo)
+	if err != nil {
+		c.Logger.Error(err)
 
 		return nil
-	})
+	}
 
-	app.Subscribe("order-logs", func(c *gofr.Context) error {
-		var orderStatus struct {
-			OrderId string `json:"orderId"`
-			Status  string `json:"status"`
-		}
+	c.Logger.Info("Received product", productInfo)
 
-		err := c.Bind(&orderStatus)
-		if err != nil {
-			c.Logger.Error(err)
+	return nil
+}
 
-			return nil
-		}
+func orderHandler(c *gofr.Context) error {
+	var orderStatus struct {
+		OrderId string `json:"orderId"`
+		Status  string `json:"status"`
+	}
 
-		c.Logger.Info("Received order", orderStatus)
+	err := c.Bind(&orderStatus)
+	if err != nil {
+		c.Logger.Error(err)
 
 		return nil
-	})
+	}
 
-	app.Run()
+	c.Logger.Info("Received order", orderStatus)
+
+	return nil
 }

examples/using-subscriber/main_test.go

@@ -2,11 +2,14 @@ package main
 
 import (
 	"context"
+	"errors"
 	"os"
 	"strings"
 	"testing"
 	"time"
 
+	"gofr.dev/pkg/gofr"
+	"gofr.dev/pkg/gofr/container"
 	"gofr.dev/pkg/gofr/datasource/pubsub/kafka"
 	"gofr.dev/pkg/gofr/logging"
 	"gofr.dev/pkg/gofr/testutil"
@@ -49,10 +52,10 @@ func TestExampleSubscriber(t *testing.T) {
 		testutil.NewServerConfigs(t)
 
 		go main()
-		time.Sleep(time.Second * 1) // Giving some time to start the server
+		time.Sleep(time.Second * 30) // Giving some time to start the server
 
 		initializeTest(t)
-		time.Sleep(time.Second * 20) // Giving some time to publish events
+		time.Sleep(time.Second * 5) // Giving some time to publish events
 	})
 
 	testCases := []struct {
@@ -75,3 +78,114 @@ func TestExampleSubscriber(t *testing.T) {
 		}
 	}
 }
+
+type errorRequest struct{}
+
+func (e *errorRequest) Context() context.Context {
+	return context.Background()
+}
+
+func (e *errorRequest) Bind(v interface{}) error    { return errors.New("bind error") }
+func (e *errorRequest) Param(key string) string     { return "" }
+func (e *errorRequest) PathParam(key string) string { return "" }
+func (e *errorRequest) HostName() string            { return "" }
+func (e *errorRequest) Params(key string) []string  { return nil }
+
+func TestProductSubscribe_BindError(t *testing.T) {
+	mockContainer, _ := container.NewMockContainer(t)
+
+	ctx := &gofr.Context{
+		Request:       &errorRequest{},
+		Container:     mockContainer,
+		ContextLogger: *logging.NewContextLogger(context.Background(), mockContainer.Logger),
+	}
+
+	err := productHandler(ctx)
+
+	if err != nil {
+		t.Errorf("Expected nil error, got %v", err)
+	}
+}
+
+func TestOrderSubscribe_BindError(t *testing.T) {
+	mockContainer, _ := container.NewMockContainer(t)
+
+	ctx := &gofr.Context{
+		Request:       &errorRequest{},
+		Container:     mockContainer,
+		ContextLogger: *logging.NewContextLogger(context.Background(), mockContainer.Logger),
+	}
+
+	err := orderHandler(ctx)
+	if err != nil {
+		t.Errorf("Expected nil error, got %v", err)
+	}
+}
+
+type successProductRequest struct{}
+
+func (r *successProductRequest) Context() context.Context {
+	return context.Background()
+}
+
+func (r *successProductRequest) Bind(v interface{}) error {
+	p := v.(*struct {
+		ProductId string `json:"productId"`
+		Price     string `json:"price"`
+	})
+	p.ProductId = "123"
+	p.Price = "599"
+	return nil
+}
+
+func (r *successProductRequest) Param(string) string     { return "" }
+func (r *successProductRequest) PathParam(string) string { return "" }
+func (r *successProductRequest) HostName() string        { return "" }
+func (r *successProductRequest) Params(string) []string  { return nil }
+
+func TestProductHandler_Success(t *testing.T) {
+	mockContainer, _ := container.NewMockContainer(t)
+	ctx := &gofr.Context{
+		Request:       &successProductRequest{},
+		Container:     mockContainer,
+		ContextLogger: *logging.NewContextLogger(context.Background(), mockContainer.Logger),
+	}
+
+	err := productHandler(ctx)
+	if err != nil {
+		t.Errorf("Expected nil error, got %v", err)
+	}
+}
+
+type successOrderRequest struct{}
+
+func (r *successOrderRequest) Context() context.Context {
+	return context.Background()
+}
+
+func (r *successOrderRequest) Bind(v interface{}) error {
+	o := v.(*struct {
+		OrderId string `json:"orderId"`
+		Status  string `json:"status"`
+	})
+	o.OrderId = "456"
+	o.Status = "pending"
+	return nil
+}
+func (r *successOrderRequest) Param(string) string     { return "" }
+func (r *successOrderRequest) PathParam(string) string { return "" }
+func (r *successOrderRequest) HostName() string        { return "" }
+func (r *successOrderRequest) Params(string) []string  { return nil }
+
+func TestOrderHandler_Success(t *testing.T) {
+	mockContainer, _ := container.NewMockContainer(t)
+	ctx := &gofr.Context{
+		Request:       &successOrderRequest{},
+		Container:     mockContainer,
+		ContextLogger: *logging.NewContextLogger(context.Background(), mockContainer.Logger),
+	}
+	err := orderHandler(ctx)
+	if err != nil {
+		t.Errorf("Expected nil error, got %v", err)
+	}
+}