Add test

Author: miguno

cmd/golang-docker-build-tutorial/main.go

@@ -4,33 +4,52 @@ import (
 	"net/http"
 
 	"github.com/go-chi/chi/v5"
+	"github.com/go-chi/chi/v5/middleware"
 	"github.com/go-chi/render"
 
 	internal "github.com/miguno/golang-docker-build-tutorial/internal/pkg"
 )
 
-// Response is just a basic example.
-type Response struct {
-	Status string `json:"status,omitempty"`
+type Server struct {
+	Router *chi.Mux
+	// Other configs such as DB settings can be added here
 }
 
-func main() {
-	r := chi.NewRouter()
-
+func CreateNewServer() *Server {
+	s := &Server{}
+	s.Router = chi.NewRouter()
 	// Uncomment to enable logging of incoming HTTP requests to STDOUT.
 	// Requires `import "github.com/go-chi/chi/v5/middleware"`.
-	//r.Use(middleware.Logger)
+	//s.Router.Use(middleware.Logger)
+	return s
+}
+
+// StatusResponse is just a basic example.
+type StatusResponse struct {
+	Status string `json:"status,omitempty"`
+}
+
+func (s *Server) MountHandlers() {
+	// Mount all Middleware here
+	s.Router.Use(middleware.Logger)
 
-	r.Get("/status", func(w http.ResponseWriter, r *http.Request) {
+	// Mount all handlers here
+	s.Router.Get("/status", func(w http.ResponseWriter, r *http.Request) {
 		// Create a Response object
-		var response Response
+		var response StatusResponse
 		if internal.IsIdleToyFunction() {
-			response = Response{Status: "idle"}
+			response = StatusResponse{Status: "idle"}
 		} else {
-			response = Response{Status: "busy"}
+			response = StatusResponse{Status: "busy"}
 		}
 
 		render.JSON(w, r, response)
 	})
-	_ = http.ListenAndServe(":8123", r)
+
+}
+
+func main() {
+	s := CreateNewServer()
+	s.MountHandlers()
+	_ = http.ListenAndServe(":8123", s.Router)
 }

cmd/golang-docker-build-tutorial/main_test.go

@@ -0,0 +1,36 @@
+package main
+
+import (
+	"encoding/json"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func executeRequest(req *http.Request, s *Server) *httptest.ResponseRecorder {
+	rr := httptest.NewRecorder()
+	s.Router.ServeHTTP(rr, req)
+	return rr
+}
+
+func verifyResponseCode(t *testing.T, expected, actual int) {
+	if expected != actual {
+		t.Errorf("Expected HTTP response code %d but got %d\n", expected, actual)
+	}
+}
+
+func TestStatusEndpoint(t *testing.T) {
+	s := CreateNewServer()
+	s.MountHandlers()
+
+	req, _ := http.NewRequest("GET", "/status", nil)
+	res := executeRequest(req, s)
+
+	verifyResponseCode(t, http.StatusOK, res.Code)
+	var sr StatusResponse
+	_ = json.Unmarshal(res.Body.Bytes(), &sr)
+	expectedResponse := StatusResponse{Status: "idle"}
+	require.Equal(t, expectedResponse, sr)
+}