add tests

Author: xbucks

.vscode/launch.json

@@ -0,0 +1,15 @@
+{
+    // Use IntelliSense to learn about possible attributes.
+    // Hover to view descriptions of existing attributes.
+    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+    "version": "0.2.0",
+    "configurations": [
+        {
+            "name": "Launch file",
+            "type": "go",
+            "request": "launch",
+            "mode": "debug",
+            "program": "${file}"
+        }
+    ]
+}

internal/server/router_test.go

@@ -2,11 +2,13 @@ package server
 
 import (
 	"context"
-	"github.com/rameshsunkara/go-rest-api-example/internal/models"
 	"net/http"
 	"testing"
 	"time"
 
+	"github.com/rameshsunkara/go-rest-api-example/internal/models"
+	"github.com/stretchr/testify/assert"
+
 	"github.com/gin-gonic/gin"
 	"go.mongodb.org/mongo-driver/mongo"
 	"go.mongodb.org/mongo-driver/mongo/options"
@@ -37,12 +39,20 @@ func (m *MockMongoDataBase) Collection(name string, opts ...*options.CollectionO
 func TestListOfRoutes(t *testing.T) {
 	router := WebRouter(svcInfo, &MockMongoDBClient{}, &MockMongoDataBase{})
 	list := router.Routes()
+	mode := gin.Mode()
+
+	assert.Equal(t, gin.ReleaseMode, mode)
 
 	assertRoutePresent(t, list, gin.RouteInfo{
 		Method: http.MethodGet,
 		Path:   "/status",
 	})
 
+	assertRouteNotPresent(t, list, gin.RouteInfo{
+		Method: http.MethodPost,
+		Path:   "/seedDB",
+	})
+
 	assertRoutePresent(t, list, gin.RouteInfo{
 		Method: http.MethodGet,
 		Path:   "/api/v1/orders",
@@ -70,6 +80,20 @@ func TestListOfRoutes(t *testing.T) {
 
 }
 
+func TestModeSpecificRoutes(t *testing.T) {
+	svcInfo.Environment = "dev"
+	router := WebRouter(svcInfo, &MockMongoDBClient{}, &MockMongoDataBase{})
+	list := router.Routes()
+	mode := gin.Mode()
+
+	assert.Equal(t, gin.DebugMode, mode)
+
+	assertRoutePresent(t, list, gin.RouteInfo{
+		Method: http.MethodPost,
+		Path:   "/seedDB",
+	})
+}
+
 func assertRoutePresent(t *testing.T, gotRoutes gin.RoutesInfo, wantRoute gin.RouteInfo) {
 	for _, gotRoute := range gotRoutes {
 		if gotRoute.Path == wantRoute.Path && gotRoute.Method == wantRoute.Method {
@@ -78,3 +102,11 @@ func assertRoutePresent(t *testing.T, gotRoutes gin.RoutesInfo, wantRoute gin.Ro
 	}
 	t.Errorf("route not found: %v", wantRoute)
 }
+
+func assertRouteNotPresent(t *testing.T, gotRoutes gin.RoutesInfo, wantRoute gin.RouteInfo) {
+	for _, gotRoute := range gotRoutes {
+		if gotRoute.Path == wantRoute.Path && gotRoute.Method == wantRoute.Method {
+			t.Errorf("route found: %v", wantRoute)
+		}
+	}
+}