Golang集成测试与单元测试如何区分 分享测试金字塔实践心得

package main

import (
    "net/http"
    "net/http/httptest"
    "strings"
    "testing"

    "github.com/DATA-DOG/go-sqlmock"
    "gorm.io/driver/mysql"
    "gorm.io/gorm"
)

func setupMockDB() (*gorm.DB, sqlmock.Sqlmock, error) {
    db, mock, err := sqlmock.New()
    if err != nil {
        return nil, nil, err
    }

    gormDB, err := gorm.Open(mysql.New(mysql.Config{
        Conn:                      db,
        SkipInitializeWithVersion: true,
    }), &gorm.Config{})

    if err != nil {
        return nil, nil, err
    }

    return gormDB, mock, nil
}

func TestMyHandler(t *testing.T) {
    db, mock, err := setupMockDB()
    if err != nil {
        t.Fatalf("Failed to setup mock DB: %v", err)
    }
    defer func() {
        sqlDB, _ := db.DB()
        sqlDB.Close()
    }()

    // 假设handler需要查询数据库获取用户信息
    mock.ExpectQuery("SELECT \* FROM users WHERE id = ?").
        WithArgs(1).
        WillReturnRows(sqlmock.NewRows([]string{"id", "name"}).AddRow(1, "testuser"))

    req, err := http.NewRequest("GET", "/users/1", nil)
    if err != nil {
        t.Fatal(err)
    }

    rr := httptest.NewRecorder()
    handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 模拟从数据库获取用户
        var user struct {
            ID   int
            Name string
        }
        db.First(&user, 1)
        w.WriteHeader(http.StatusOK)
        w.Write([]byte(user.Name))
    })

    handler.ServeHTTP(rr, req)

    if status := rr.Code; status != http.StatusOK {
        t.Errorf("handler returned wrong status code: got %v want %v",
            status, http.StatusOK)
    }

    expected := "testuser"
    if strings.TrimSpace(rr.Body.String()) != expected {
        t.Errorf("handler returned unexpected body: got %v want %v",
            rr.Body.String(), expected)
    }

    if err := mock.ExpectationsWereMet(); err != nil {
        t.Errorf("there were unfulfilled expectations: %s", err)
    }
}

package calculator

func Add(a, b int) int {
    return a + b
}

package calculator

import "testing"

func TestAdd(t *testing.T) {
    result := Add(2, 3)
    if result != 5 {
        t.Errorf("Add(2, 3) returned %d, expected 5", result)
    }
}