API测试 #

package main

import (
    "bytes"
    "encoding/json"
    "net/http/httptest"
    "testing"
    
    "github.com/gofiber/fiber/v2"
    "github.com/stretchr/testify/assert"
)

func TestUserAPI(t *testing.T) {
    app := fiber.New()
    
    var users []map[string]interface{}
    
    app.Get("/users", func(c *fiber.Ctx) error {
        return c.JSON(users)
    })
    
    app.Post("/users", func(c *fiber.Ctx) error {
        var user map[string]interface{}
        c.BodyParser(&user)
        users = append(users, user)
        return c.Status(201).JSON(user)
    })
    
    app.Get("/users/:id", func(c *fiber.Ctx) error {
        id := c.Params("id")
        for _, u := range users {
            if u["id"] == id {
                return c.JSON(u)
            }
        }
        return c.Status(404).SendString("Not found")
    })
    
    t.Run("ListUsers", func(t *testing.T) {
        req := httptest.NewRequest("GET", "/users", nil)
        resp, _ := app.Test(req)
        assert.Equal(t, 200, resp.StatusCode)
    })
    
    t.Run("CreateUser", func(t *testing.T) {
        body, _ := json.Marshal(map[string]string{"id": "1", "name": "John"})
        req := httptest.NewRequest("POST", "/users", bytes.NewReader(body))
        req.Header.Set("Content-Type", "application/json")
        resp, _ := app.Test(req)
        assert.Equal(t, 201, resp.StatusCode)
    })
    
    t.Run("GetUser", func(t *testing.T) {
        req := httptest.NewRequest("GET", "/users/1", nil)
        resp, _ := app.Test(req)
        assert.Equal(t, 200, resp.StatusCode)
    })
}

func TestIntegration(t *testing.T) {
    // 设置测试数据库
    db := setupTestDB()
    defer db.Close()
    
    // 创建应用
    app := setupApp(db)
    
    // 测试完整流程
    t.Run("RegisterAndLogin", func(t *testing.T) {
        // 注册
        body := `{"username":"test","password":"password"}`
        req := httptest.NewRequest("POST", "/register", strings.NewReader(body))
        req.Header.Set("Content-Type", "application/json")
        resp, _ := app.Test(req)
        assert.Equal(t, 201, resp.StatusCode)
        
        // 登录
        req = httptest.NewRequest("POST", "/login", strings.NewReader(body))
        req.Header.Set("Content-Type", "application/json")
        resp, _ = app.Test(req)
        assert.Equal(t, 200, resp.StatusCode)
    })
}

func BenchmarkGetUsers(b *testing.B) {
    app := fiber.New()
    app.Get("/users", func(c *fiber.Ctx) error {
        return c.JSON([]string{"user1", "user2"})
    })
    
    b.ResetTimer()
    
    for i := 0; i < b.N; i++ {
        req := httptest.NewRequest("GET", "/users", nil)
        app.Test(req)
    }
}