router test

internal/router/answer_router.go

@@ -10,8 +10,8 @@ import (
 	"go.mongodb.org/mongo-driver/mongo"
 )
 
-func InitAnswerRoutes(r *gin.Engine, db *mongo.Database, conf *config.Config) {
-	forumRepository := forum_repository.NewForumRepository(db.Collection("forum"), conf)
+func InitAnswerRoutes(r *gin.Engine, forumCollection *mongo.Collection, conf *config.Config) {
+	forumRepository := forum_repository.NewForumRepository(forumCollection, conf)
 	pushNotifier := notification.NewPushNotifier(&conf.NOTIFICATIONS_SERVICE, conf)
 	questionService := answer_service.NewAnswerService(forumRepository, pushNotifier, conf)
 	answerHandler := answer_handler.NewAnswerHandler(questionService, conf)

internal/router/question_router.go

@@ -11,8 +11,8 @@ import (
 	"go.mongodb.org/mongo-driver/mongo"
 )
 
-func InitQuestionRoutes(r *gin.Engine, db *mongo.Database, conf *config.Config) {
-	forumRepository := forum_repository.NewForumRepository(db.Collection("forum"), conf)
+func InitQuestionRoutes(r *gin.Engine, forumCollection *mongo.Collection, conf *config.Config) {
+	forumRepository := forum_repository.NewForumRepository(forumCollection, conf)
 	pushNotifier := notifications.NewPushNotifier(&conf.NOTIFICATIONS_SERVICE, conf)
 	questionService := question_service.NewQuestionService(forumRepository, pushNotifier, conf)
 	questionHandler := question_handler.NewQuestionHandler(questionService, conf)

internal/router/router.go

@@ -19,8 +19,14 @@
 	r.Use(middleware.ErrorHandlerMiddleware(conf))
 	r.Use(middleware.JWTAuthMiddleware(conf))
 
-	InitQuestionRoutes(r, db, conf)
-	InitAnswerRoutes(r, db, conf)
+	var forumCollection *mongo.Collection
+    if db != nil {
+        forumCollection = db.Collection("forum")
+    }
+
+
+	InitQuestionRoutes(r, forumCollection, conf)
+	InitAnswerRoutes(r, forumCollection, conf)
 	InitSwaggerRouter(r)
 
 	return r

internal/router/router_test.go

@@ -0,0 +1,51 @@
+package router
+
+import (
+	"fmt"
+	"testing"
+	"forum-microservice/config"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestSetupRouter_RegistersExpectedRoutes(t *testing.T) {
+	conf := &config.Config{
+		SV_ENVIRONMENT: config.DEVELOPMENT,
+	}
+	router := SetupRouter(nil, conf)
+
+	expectedRoutes := []string{
+		// Swagger
+		"GET /swagger/*any",
+
+		// Question
+		"POST /:courseId/question",
+		"PUT /question/:questionId",
+		"DELETE /question/:questionId",
+		"GET /:courseId/questions",
+		"GET /question/:questionId",
+		"POST /question/:questionId/upvote",
+		"POST /question/:questionId/downvote",
+		"DELETE /question/:questionId/vote",
+
+		// Answer
+		"POST /question/:questionId/answer",
+		"PUT /question/:questionId/answer/:answerId",
+		"DELETE /question/:questionId/answer/:answerId",
+		"POST /question/:questionId/answer/:answerId/upvote",
+		"POST /question/:questionId/answer/:answerId/downvote",
+		"DELETE /question/:questionId/answer/:answerId/vote",
+		"PUT /question/:questionId/answer/:answerId/correct",
+		"DELETE /question/:questionId/answer/correct",
+	}
+
+	actualRoutes := make(map[string]bool)
+	for _, route := range router.Routes() {
+		key := fmt.Sprintf("%s %s", route.Method, route.Path)
+		actualRoutes[key] = true
+	}
+
+	for _, expected := range expectedRoutes {
+		assert.True(t, actualRoutes[expected], "Expected route not registered: %s", expected)
+	}
+}