Add route prefix (#2)

Author: vincetse

hello_world.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"flag"
 	"net/http"
 	"os"
 
@@ -28,16 +29,21 @@ func healthFunc(c *gin.Context) {
 	c.String(http.StatusOK, "")
 }
 
-func setupRouter() *gin.Engine {
+func setupRouter(routePrefix string) *gin.Engine {
 	router := gin.Default()
 	ginprom := ginprometheus.NewPrometheus("gin")
 	ginprom.Use(router)
-	router.GET("/", helloFunc)
 	router.GET("/health", healthFunc)
+
+	rg := router.Group(routePrefix)
+	rg.GET("/", helloFunc)
 	return router
 }
 
 func main() {
-	router := setupRouter()
+	optRoutePrefix := flag.String("route-prefix", "/", "Route prefix")
+	flag.Parse()
+
+	router := setupRouter(*optRoutePrefix)
 	router.Run()
 }

hello_world_test.go

@@ -9,7 +9,7 @@ import (
 )
 
 func TestDefaultRoute(t *testing.T) {
-	router := setupRouter()
+	router := setupRouter("/")
 
 	w := httptest.NewRecorder()
 	req, _ := http.NewRequest("GET", "/", nil)
@@ -20,7 +20,7 @@ func TestDefaultRoute(t *testing.T) {
 }
 
 func TestHealthRoute(t *testing.T) {
-	router := setupRouter()
+	router := setupRouter("/")
 
 	w := httptest.NewRecorder()
 	req, _ := http.NewRequest("GET", "/health", nil)
@@ -31,11 +31,23 @@ func TestHealthRoute(t *testing.T) {
 }
 
 func TestMetricsRoute(t *testing.T) {
-	router := setupRouter()
+	router := setupRouter("/")
 
 	w := httptest.NewRecorder()
 	req, _ := http.NewRequest("GET", "/metrics", nil)
 	router.ServeHTTP(w, req)
 
 	assert.Equal(t, 200, w.Code)
 }
+
+func TestRoutePrefix(t *testing.T) {
+	prefix := "/foobar/"
+	router := setupRouter(prefix)
+
+	w := httptest.NewRecorder()
+	req, _ := http.NewRequest("GET", prefix, nil)
+	router.ServeHTTP(w, req)
+
+	assert.Equal(t, 200, w.Code)
+	assert.Equal(t, "Hello, World!", w.Body.String())
+}