Add /erised/webpage route

EAddario · EAddario · commit 412c76b15d55 · 2024-06-28T12:21:29.000+01:00
diff --git a/cmd/erised/serverRoutes.go b/cmd/erised/serverRoutes.go
@@ -22,6 +22,8 @@ func (s *server) routes() {
 	go s.mux.HandleFunc("/erised/info", s.handleInfo())
 	go s.mux.HandleFunc("/erised/ip", s.handleIP())
 	go s.mux.HandleFunc("/erised/shutdown", s.handleShutdown())
+	go s.mux.HandleFunc("/erised/webpage", s.handleWebPage())
+	go s.mux.HandleFunc("/erised/webpage/{path...}", s.handleWebPage())
 	log.Debug().Msg("leaving routes")
 }
 
@@ -247,3 +249,30 @@ func (s *server) handleShutdown() http.HandlerFunc {
 		log.Debug().Msg("leaving handleShutdown")
 	}
 }
+
+func (s *server) handleWebPage() http.HandlerFunc {
+	log.Debug().Msg("entering handleWebPage")
+
+	return func(res http.ResponseWriter, req *http.Request) {
+		log.Info().
+			Str("protocol", req.Proto).
+			Str("remoteAddress", req.RemoteAddr).
+			Str("method", req.Method).
+			Str("host", req.Host).
+			Str("path", req.RequestURI).
+			Msg("handleWebPage")
+
+		res.Header().Set("Content-Type", "text/html")
+
+		data := "<!DOCTYPE html><html><head><title>Erised Webpage</title></head>"
+		data += "<body>"
+		data += "<center><h1>Host: " + req.Host + "</h1></center>"
+		data += "<center><h1>Method: " + req.Method + "</h1></center>"
+		data += "<center><h1>Protocol: " + req.Proto + "</h1></center>"
+		data += "<center><h1>Request Path: " + req.RequestURI + "</h1></center>"
+		data += "<hr><center><a href=\"https://github.com/EAddario/erised\">Erised: A nimble http server to test arbitrary REST API responses.</a></center>"
+		data += "</body></html>"
+		s.respond(res, encodingHTML, 0, data)
+		log.Debug().Msg("leaving handleWebPage")
+	}
+}
diff --git a/cmd/erised/serverRoutes_test.go b/cmd/erised/serverRoutes_test.go
@@ -154,6 +154,35 @@ func TestErisedShutdownRoute(t *testing.T) {
 	})
 }
 
+func TestErisedWebPageRoute(t *testing.T) {
+	zerolog.SetGlobalLevel(zerolog.Disabled)
+	g := goblin.Goblin(t)
+	RegisterFailHandler(func(m string, _ ...int) { g.Fail(m) })
+	svr := server{}
+
+	g.Describe("Test erised/webpage", func() {
+		g.It("Should return StatusOK", func() {
+			res := httptest.NewRecorder()
+			req := httptest.NewRequest(http.MethodGet, "http://localhost:8080/erised/webpage", nil)
+			svr.handleWebPage().ServeHTTP(res, req)
+
+			Ω(res.Code).Should(Equal(http.StatusOK))
+			Ω(res.Body.String()).ShouldNot(BeEmpty())
+			Ω(res.Header().Get("Content-Type")).Should(Equal("text/html"))
+		})
+
+		g.It("Should also return StatusOK", func() {
+			res := httptest.NewRecorder()
+			req := httptest.NewRequest(http.MethodPost, "http://localhost:8080/erised/webpage/any/path", nil)
+			svr.handleWebPage().ServeHTTP(res, req)
+
+			Ω(res.Code).Should(Equal(http.StatusOK))
+			Ω(res.Body.String()).ShouldNot(BeEmpty())
+			Ω(res.Header().Get("Content-Type")).Should(Equal("text/html"))
+		})
+	})
+}
+
 func TestErisedLandingRoute(t *testing.T) {
 	zerolog.SetGlobalLevel(zerolog.Disabled)
 	g := goblin.Goblin(t)
