Refactor. Add tests for the home & api routes

main.go

@@ -94,6 +94,17 @@ func apiHandler(w http.ResponseWriter, r *http.Request) {
 	w.Write(payload) // nolint: errcheck
 }
 
+// RegisterAppHandlers registers the app handlers with the given mux
+func RegisterAppHandlers(mux *http.ServeMux) {
+	mux.Handle(StaticPath,
+		http.StripPrefix(StaticPath, http.FileServer(http.Dir("static"))),
+	)
+
+	mux.HandleFunc("/", indexHandler)
+	mux.HandleFunc(APIPath, apiHandler)
+
+}
+
 func main() {
 	config := zap.NewProductionConfig()
 	config.EncoderConfig.TimeKey = "timestamp"
@@ -120,15 +131,10 @@ func main() {
 	mux := http.NewServeMux()
 
 	RegisterDebugHandler(mux)
-	mux.Handle(StaticPath,
-		http.StripPrefix(StaticPath, http.FileServer(http.Dir("static"))),
-	)
-
-	mux.HandleFunc("/", indexHandler)
-	mux.HandleFunc(APIPath, apiHandler)
+	RegisterAppHandlers(mux)
 
 	server := http.Server{
-		Handler:      http.TimeoutHandler(mux, procTimeout, "Processing your request took too long!"),
+		Handler:      http.TimeoutHandler(mux, procTimeout, "Processing your request took too long."),
 		ReadTimeout:  readTimeout,
 		WriteTimeout: writeTimeout,
 	}

main_test.go

@@ -50,6 +50,13 @@ func TestAPIHandler(t *testing.T) {
 			response: "tokenizer parameter not found\n",
 			status:   http.StatusBadRequest,
 		},
+		{
+			name:     "Empty payload",
+			method:   "GET",
+			payload:  nil,
+			response: "str parameter not found\n",
+			status:   http.StatusBadRequest,
+		},
 	}
 
 	for _, tc := range testCases {
@@ -78,3 +85,50 @@ func TestAPIHandler(t *testing.T) {
 		})
 	}
 }
+
+func TestIndexRoute(t *testing.T) {
+	mux := http.NewServeMux()
+	RegisterAppHandlers(mux)
+
+	// The response recorder used to record HTTP responses
+	rr := httptest.NewRecorder()
+
+	req, err := http.NewRequest("GET", "/", nil)
+	if err != nil {
+		t.Fatal("Creating 'GET /' request failed,")
+	}
+
+	mux.ServeHTTP(rr, req)
+	if rr.Code != http.StatusOK {
+		t.Fatalf("'GET /' request should've a handler registered, got: %d, want: %d",
+			rr.Code, http.StatusOK,
+		)
+	}
+}
+
+func TestAPIRoute(t *testing.T) {
+	mux := http.NewServeMux()
+	RegisterAppHandlers(mux)
+
+	// The response recorder used to record HTTP responses
+	rr := httptest.NewRecorder()
+	payload := url.Values{
+		"tokenizer": {"%{key1} %{key2}"},
+		"str":       {"a b"},
+	}
+
+	req, err := http.NewRequest("POST", APIPath, strings.NewReader(
+		payload.Encode()))
+	if err != nil {
+		t.Fatalf("Creating 'POST /api' request failed.")
+	}
+	req.Header.Add("Content-Type",
+		"application/x-www-form-urlencoded;charset=UTF-8")
+
+	mux.ServeHTTP(rr, req)
+	if rr.Code != http.StatusOK {
+		t.Fatalf("'POST /api' request should've a handler registered, got: %d, want: %d",
+			rr.Code, http.StatusOK,
+		)
+	}
+}