Golang中的测试环境管理方法

package main

import (
    "testing"
)

func TestAddition(t *testing.T) {
    a := 1
    b := 2
    expected := 3

    result := a + b

    if result != expected {
        t.Errorf("Addition test failed: got %d, expected %d", result, expected)
    }
}

func TestSubtraction(t *testing.T) {
    a := 3
    b := 2
    expected := 1

    result := a - b

    if result != expected {
        t.Errorf("Subtraction test failed: got %d, expected %d", result, expected)
    }
}

package main

import (
    "testing"
)

func TestAddition(t *testing.T) {
    tests := []struct {
        a        int
        b        int
        expected int
    }{
        {1, 2, 3},
        {3, 4, 7},
        {5, 6, 11},
    }

    for _, test := range tests {
        result := test.a + test.b

        if result != test.expected {
            t.Errorf("Addition test failed: got %d, expected %d", result, test.expected)
        }
    }
}

package main

import (
    "fmt"
    "testing"
)

// 定义一个接口
type Database interface {
    GetData() string
}

// 定义一个Mock对象
type MockDatabase struct{}

// 实现接口方法
func (m *MockDatabase) GetData() string {
    return "mock data"
}

func TestGetData(t *testing.T) {
    // 创建一个Mock对象
    mockDB := &MockDatabase{}
    // 使用Mock对象替代真实对象
    data := getDataFromDatabase(mockDB)
    expected := "mock data"

    if data != expected {
        t.Errorf("GetData test failed: got %s, expected %s", data, expected)
    }
}

// 在被测试的代码中使用接口
func getDataFromDatabase(db Database) string {
    return db.GetData()
}

func main() {
    fmt.Println(getDataFromDatabase(&MockDatabase{}))
}

package main

import (
    "database/sql"
    "testing"

    _ "github.com/mattn/go-sqlite3"
)

func TestGetData(t *testing.T) {
    // 创建一个临时数据库
    db, err := sql.Open("sqlite3", ":memory:")
    if err != nil {
        t.Fatalf("Failed to create temporary database: %v", err)
    }

    // 创建相关表和数据
    _, err = db.Exec("CREATE TABLE users (id INTEGER PRIMARY KEY, name TEXT)")
    if err != nil {
        t.Fatalf("Failed to create table: %v", err)
    }
    _, err = db.Exec("INSERT INTO users (id, name) VALUES (1, 'user_1')")
    if err != nil {
        t.Fatalf("Failed to insert data: %v", err)
    }

    // 查询数据
    rows, err := db.Query("SELECT * FROM users")
    if err != nil {
        t.Fatalf("Failed to query data: %v", err)
    }
    defer rows.Close()

    var id int
    var name string
    for rows.Next() {
        err = rows.Scan(&id, &name)
        if err != nil {
            t.Fatalf("Failed to scan data: %v", err)
        }
        // 断言
        if id != 1 || name != "user_1" {
            t.Errorf("GetData test failed: got id=%d name=%s, expected id=1 name=user_1", id, name)
        }
    }
}

func main() {}