회사들 리스트 조회 api 추가 & 테스트 db config설정

cmd/main.go

@@ -7,6 +7,7 @@ import (
 	"github.com/techbloghub/server/config"
 	_ "github.com/techbloghub/server/ent/runtime"
 	"github.com/techbloghub/server/internal/database"
+	"github.com/techbloghub/server/internal/http/router"
 
 	"github.com/gin-gonic/gin"
 	_ "github.com/lib/pq"
@@ -26,18 +27,11 @@ func main() {
 	defer client.Close()
 
 	// 서버 실행
-	r := setRouter()
+	r := gin.Default()
+	router.InitRouter(r)
 	routerErr := r.Run(":" + cfg.ServerConfig.Port)
 	if routerErr != nil {
 		fmt.Println("Error while running server: ", routerErr)
 		return
 	}
 }
-
-func setRouter() *gin.Engine {
-	r := gin.Default()
-	r.GET("/ping", func(context *gin.Context) {
-		context.String(200, "pong")
-	})
-	return r
-}

cmd/main_test.go

@@ -7,14 +7,16 @@ import (
 
 	"github.com/gin-gonic/gin"
 	"github.com/stretchr/testify/assert"
+	"github.com/techbloghub/server/internal/http/router"
 )
 
 func TestPingEndpoint(t *testing.T) {
 	// Set Gin to Test Mode
 	gin.SetMode(gin.TestMode)
 
-	// Create a new router instance using the private setRouter function
-	router := setRouter()
+	// Create a new r instance using the private setRouter function
+	r := gin.Default()
+	router.InitRouter(r)
 
 	// Create a test HTTP recorder
 	w := httptest.NewRecorder()
@@ -24,11 +26,11 @@ func TestPingEndpoint(t *testing.T) {
 	assert.NoError(t, err)
 
 	// Serve the request using the router
-	router.ServeHTTP(w, req)
+	r.ServeHTTP(w, req)
 
 	// Assert the response code is 200
 	assert.Equal(t, http.StatusOK, w.Code)
 
 	// Assert the response body is "pong"
-	assert.Equal(t, "pong", w.Body.String())
+	assert.Equal(t, "pong\n", w.Body.String())
 }

internal/http/handler/pinpong.go

@@ -0,0 +1,7 @@
+package handler
+
+import "github.com/gin-gonic/gin"
+
+func PingPong(context *gin.Context) {
+	context.String(200, "pong\n")
+}

internal/http/router/router.go

@@ -0,0 +1,10 @@
+package router
+
+import (
+	"github.com/gin-gonic/gin"
+	"github.com/techbloghub/server/internal/http/handler"
+)
+
+func InitRouter(r *gin.Engine) {
+	r.GET("/ping", handler.PingPong)
+}