Golang测试数据准备与清理技巧分享

func TestUserValidate(t *testing.T) {
    user := User{
        Name:  "alice",
        Email: "alice@example.com",
    }
    if !user.Validate() {
        t.Error("expected valid user")
    }
}

func newUserBuilder() *User {
    return &User{
        Name:    "testuser",
        Email:   "test@example.com",
        Active:  true,
        Created: time.Now(),
    }
}

func TestUserCreation(t *testing.T) {
    user := newUserBuilder().SetName("bob")
    // 测试逻辑
}

func setupTestDB() (*sql.DB, func()) {
    db, _ := sql.Open("sqlite3", ":memory:")
    db.Exec(`CREATE TABLE users (id INTEGER PRIMARY KEY, name TEXT)`)
    db.Exec(`INSERT INTO users (name) VALUES ('alice')`)

    return db, func() { db.Close() }
}

func TestListUsers(t *testing.T) {
    db, teardown := setupTestDB()
    defer teardown()

    users, err := ListUsers(db)
    if err != nil || len(users) == 0 {
        t.Fail()
    }
}

func TestReadConfig(t *testing.T) {
    tmpfile, _ := os.CreateTemp("", "config-*.json")
    defer os.Remove(tmpfile.Name()) // 清理
    defer tmpfile.Close()

    tmpfile.Write([]byte(`{"port": 8080}`))
    tmpfile.Seek(0, 0)

    cfg, err := ReadConfig(tmpfile.Name())
    if err != nil || cfg.Port != 8080 {
        t.Error("parse failed")
    }
}

tx, _ := db.Begin()
defer tx.Rollback() // 不提交，自然清理