test: fix test

IceWhaleTech · May 29, 2024 · 2a6545a · 2a6545a
1 parent 197adae
commit 2a6545a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/route/management_route_test.go b/route/management_route_test.go
@@ -10,6 +10,7 @@ import (
 	"testing"
 
 	"github.com/IceWhaleTech/CasaOS-Common/utils/logger"
+	"github.com/labstack/echo/v4"
 
 	"github.com/IceWhaleTech/CasaOS-Common/model"
 	"github.com/IceWhaleTech/CasaOS-Gateway/service"
@@ -69,6 +70,7 @@ func TestCreateRoute(t *testing.T) {
 
 	req, _ := http.NewRequest(http.MethodPost, "/v1/gateway/routes", bytes.NewReader(body))
 	req.RemoteAddr = "127.0.0.1:0"
+	req.Header.Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
 
 	w := httptest.NewRecorder()
 	_router.ServeHTTP(w, req)
@@ -112,6 +114,7 @@ func TestChangePort(t *testing.T) {
 
 	req, _ := http.NewRequest(http.MethodPut, "/v1/gateway/port", bytes.NewReader(body))
 	req.RemoteAddr = "127.0.0.1:0"
+	req.Header.Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
 
 	w := httptest.NewRecorder()
 	_router.ServeHTTP(w, req)
@@ -150,6 +153,7 @@ func TestChangePortNegative(t *testing.T) {
 
 	req, _ := http.NewRequest(http.MethodPut, "/v1/gateway/port", bytes.NewReader(body))
 	req.RemoteAddr = "127.0.0.1:0"
+	req.Header.Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
 
 	w := httptest.NewRecorder()
 	_router.ServeHTTP(w, req)
@@ -159,6 +163,7 @@ func TestChangePortNegative(t *testing.T) {
 
 	// get
 	req, _ = http.NewRequest(http.MethodGet, "/v1/gateway/port", nil)
+	req.Header.Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
 
 	w = httptest.NewRecorder()
 	_router.ServeHTTP(w, req)
@@ -185,6 +190,7 @@ func TestChangePortNegative(t *testing.T) {
 
 	req, _ = http.NewRequest(http.MethodPut, "/v1/gateway/port", bytes.NewReader(body))
 	req.RemoteAddr = "127.0.0.1:0"
+	req.Header.Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
 
 	w = httptest.NewRecorder()
 	_router.ServeHTTP(w, req)