add test of api

internal/sbi/api_hi.go

@@ -8,14 +8,28 @@ import (
 func (s *Server) getHiRoute() []Route {
 	return []Route {
 		{
-			Name: "Hi",
+			Name: "sayHello",
 			Method: http.MethodGet,
 			Pattern: "/",
-			APIFunc: func(c *gin.Context) {
-				c.JSON(http.StatusOK, "Today is a good day.")
-			},
+			APIFunc: s.SayHello,
 			// Use
 			// curl -X GET http://127.0.0.163:8000/hi/ -w "\n"
 		},
+		{
+			Name: "sayGoodBye",
+			Method: http.MethodPost,
+			Pattern: "/",
+			APIFunc: s.SayGoodBye,
+			// Use
+			// curl -X POST http://127.0.0.163:8000/hi/ -w "\n"
+		},
 	}
 }
+
+func(s *Server)SayHello(c *gin.Context) {
+	c.String(http.StatusOK, "Hello!\n")
+}
+
+func(s *Server)SayGoodBye(c *gin.Context) {
+	c.String(http.StatusOK, "Good-bye!\n")
+}

internal/sbi/api_hi_test.go

@@ -0,0 +1,77 @@
+package sbi_test
+
+import (
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/andy89923/nf-example/internal/sbi"
+	"github.com/andy89923/nf-example/pkg/factory"
+	"github.com/gin-gonic/gin"
+	"go.uber.org/mock/gomock"
+)
+
+func Test_getHiRoutes(t *testing.T) {
+	gin.SetMode(gin.TestMode)
+
+	mockCtrl := gomock.NewController(t)
+	nfApp := sbi.NewMocknfApp(mockCtrl)
+	nfApp.EXPECT().Config().Return(&factory.Config{
+		Configuration: &factory.Configuration{
+			Sbi: &factory.Sbi{
+				Port: 8000,
+			},
+		},
+	}).AnyTimes()
+	server := sbi.NewServer(nfApp, "")
+
+	t.Run("sayHello", func(t *testing.T) {
+		const EXPECTED_STATUS = http.StatusOK
+		const EXPECTED_BODY = "Hello!\n"
+
+		httpRecorder := httptest.NewRecorder()
+		ginCtx, _ := gin.CreateTestContext(httpRecorder)
+
+		var err error
+		ginCtx.Request, err = http.NewRequest("GET", "/hi", nil)
+		if err != nil {
+			t.Errorf("Failed to create request: %s", err)
+			return
+		}
+
+		server.SayHello(ginCtx)
+
+		if httpRecorder.Code != EXPECTED_STATUS {
+			t.Errorf("Expected status code %d, got %d", EXPECTED_STATUS, httpRecorder.Code)
+		}
+
+		if httpRecorder.Body.String() != EXPECTED_BODY {
+			t.Errorf("Expected body %s, got %s", EXPECTED_BODY, httpRecorder.Body.String())
+		}
+	})
+
+	t.Run("sayGoodBye", func(t *testing.T) {
+		const EXPECTED_STATUS = http.StatusOK
+		const EXPECTED_BODY = "Good-bye!\n"
+
+		httpRecorder := httptest.NewRecorder()
+		ginCtx, _ := gin.CreateTestContext(httpRecorder)
+
+		var err error
+		ginCtx.Request, err = http.NewRequest("POST", "/hi", nil)
+		if err != nil {
+			t.Errorf("Failed to create request: %s", err)
+			return
+		}
+
+		server.SayGoodBye(ginCtx)
+
+		if httpRecorder.Code != EXPECTED_STATUS {
+			t.Errorf("Expected status code %d, got %d", EXPECTED_STATUS, httpRecorder.Code)
+		}
+
+		if httpRecorder.Body.String() != EXPECTED_BODY {
+			t.Errorf("Expected body %s, got %s", EXPECTED_BODY, httpRecorder.Body.String())
+		}
+	})
+}