wangtianqi/Golang_demo
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

go fmt

Browse Source
This commit is contained in:
wangtianqi 2025-06-25 15:12:40 +08:00
parent b433e690b6
commit 8c7c5c4287
2 changed files with 15 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
main.go
View File

@ -39,7 +39,7 @@ func main() {
// 获取端口 // 获取端口
port := getEnv("PORT", "8080") port := getEnv("PORT", "8080")
// 创建HTTP服务器 // 创建HTTP服务器
srv := &http.Server{ srv := &http.Server{
Addr: ":" + port, Addr: ":" + port,
@ -192,4 +192,4 @@ func ginZapLogger(logger *zap.Logger) gin.HandlerFunc {
func ginZapRecovery(logger *zap.Logger, stack bool) gin.HandlerFunc { func ginZapRecovery(logger *zap.Logger, stack bool) gin.HandlerFunc {
return gin.Recovery() return gin.Recovery()
} }

26
main_test.go
View File

@ -17,68 +17,68 @@ func TestMain(m *testing.M) {
func TestPingRoute(t *testing.T) { func TestPingRoute(t *testing.T) {
router := setupRouter() router := setupRouter()
w := httptest.NewRecorder() w := httptest.NewRecorder()
req, _ := http.NewRequest("GET", "/ping", nil) req, _ := http.NewRequest("GET", "/ping", nil)
router.ServeHTTP(w, req) router.ServeHTTP(w, req)
assert.Equal(t, 200, w.Code) assert.Equal(t, 200, w.Code)
assert.Contains(t, w.Body.String(), "pong") assert.Contains(t, w.Body.String(), "pong")
} }
func TestHealthRoute(t *testing.T) { func TestHealthRoute(t *testing.T) {
router := setupRouter() router := setupRouter()
w := httptest.NewRecorder() w := httptest.NewRecorder()
req, _ := http.NewRequest("GET", "/health", nil) req, _ := http.NewRequest("GET", "/health", nil)
router.ServeHTTP(w, req) router.ServeHTTP(w, req)
assert.Equal(t, 200, w.Code) assert.Equal(t, 200, w.Code)
assert.Contains(t, w.Body.String(), "healthy") assert.Contains(t, w.Body.String(), "healthy")
} }
func TestHomeRoute(t *testing.T) { func TestHomeRoute(t *testing.T) {
router := setupRouter() router := setupRouter()
w := httptest.NewRecorder() w := httptest.NewRecorder()
req, _ := http.NewRequest("GET", "/", nil) req, _ := http.NewRequest("GET", "/", nil)
router.ServeHTTP(w, req) router.ServeHTTP(w, req)
assert.Equal(t, 200, w.Code) assert.Equal(t, 200, w.Code)
assert.Contains(t, w.Body.String(), "Welcome to Golang Demo API") assert.Contains(t, w.Body.String(), "Welcome to Golang Demo API")
} }
func TestVersionRoute(t *testing.T) { func TestVersionRoute(t *testing.T) {
router := setupRouter() router := setupRouter()
w := httptest.NewRecorder() w := httptest.NewRecorder()
req, _ := http.NewRequest("GET", "/version", nil) req, _ := http.NewRequest("GET", "/version", nil)
router.ServeHTTP(w, req) router.ServeHTTP(w, req)
assert.Equal(t, 200, w.Code) assert.Equal(t, 200, w.Code)
assert.Contains(t, w.Body.String(), "version") assert.Contains(t, w.Body.String(), "version")
} }
func TestAPIStatusRoute(t *testing.T) { func TestAPIStatusRoute(t *testing.T) {
router := setupRouter() router := setupRouter()
w := httptest.NewRecorder() w := httptest.NewRecorder()
req, _ := http.NewRequest("GET", "/api/v1/status", nil) req, _ := http.NewRequest("GET", "/api/v1/status", nil)
router.ServeHTTP(w, req) router.ServeHTTP(w, req)
assert.Equal(t, 200, w.Code) assert.Equal(t, 200, w.Code)
assert.Contains(t, w.Body.String(), "v1") assert.Contains(t, w.Body.String(), "v1")
} }
func TestEchoRoute(t *testing.T) { func TestEchoRoute(t *testing.T) {
router := setupRouter() router := setupRouter()
body := `{"message": "test message"}` body := `{"message": "test message"}`
w := httptest.NewRecorder() w := httptest.NewRecorder()
req, _ := http.NewRequest("POST", "/api/v1/echo", strings.NewReader(body)) req, _ := http.NewRequest("POST", "/api/v1/echo", strings.NewReader(body))
req.Header.Set("Content-Type", "application/json") req.Header.Set("Content-Type", "application/json")
router.ServeHTTP(w, req) router.ServeHTTP(w, req)
assert.Equal(t, 200, w.Code) assert.Equal(t, 200, w.Code)
assert.Contains(t, w.Body.String(), "test message") assert.Contains(t, w.Body.String(), "test message")
} }