Go语言中mgo.Database的单元测试策略：使用接口实现依赖倒置

package main

import (
    "fmt"
    "gopkg.in/mgo.v2"
    "gopkg.in/mgo.v2/bson"
)

// MyData 示例数据结构
type MyData struct {
    ID   bson.ObjectId `bson:"_id,omitempty"`
    Name string        `bson:"name"`
    Age  int           `bson:"age"`
}

// myFunc 接收 *mgo.Database 参数，并执行一些数据库操作
func myFunc(db *mgo.Database, data MyData) error {
    c := db.C("mycollection") // 获取集合
    // 假设我们只关心插入操作
    err := c.Insert(data)
    if err != nil {
        return fmt.Errorf("failed to insert data: %w", err)
    }
    fmt.Printf("Data inserted successfully: %+v\n", data)
    return nil
}

package main

import (
    "gopkg.in/mgo.v2/bson"
)

// Collectioner 是一个接口，代表 mgo.Collection 的 Insert 方法
type Collectioner interface {
    Insert(docs ...interface{}) error
}

// DataBaser 是一个接口，代表 mgo.Database 的 C 方法
type DataBaser interface {
    C(name string) Collectioner // C 方法返回一个 Collectioner 接口
}

package main

import (
    "fmt"
    // ... 其他导入保持不变
)

// myFunc 接收 DataBaser 接口参数
func myFunc(db DataBaser, data MyData) error {
    c := db.C("mycollection") // 获取 Collectioner 接口
    err := c.Insert(data)
    if err != nil {
        return fmt.Errorf("failed to insert data: %w", err)
    }
    fmt.Printf("Data inserted successfully: %+v\n", data)
    return nil
}

package main

import (
    "errors"
    // ... 其他导入保持不变
)

// MockCollection 是 Collectioner 接口的模拟实现
type MockCollection struct {
    InsertFunc func(docs ...interface{}) error
}

func (mc *MockCollection) Insert(docs ...interface{}) error {
    if mc.InsertFunc != nil {
        return mc.InsertFunc(docs...)
    }
    return nil // 默认成功
}

// MockDatabase 是 DataBaser 接口的模拟实现
type MockDatabase struct {
    CFunc func(name string) Collectioner
}

func (md *MockDatabase) C(name string) Collectioner {
    if md.CFunc != nil {
        return md.CFunc(name)
    }
    return &MockCollection{} // 默认返回一个默认成功的集合模拟
}

package main

import (
    "testing"
    // ... 其他导入保持不变
)

func TestMyFunc(t *testing.T) {
    testData := MyData{
        ID:   bson.NewObjectId(),
        Name: "Test User",
        Age:  30,
    }

    t.Run("successful insert", func(t *testing.T) {
        // 创建一个模拟集合，其 Insert 方法总是成功
        mockCol := &MockCollection{
            InsertFunc: func(docs ...interface{}) error {
                t.Log("Mock Insert called successfully")
                return nil
            },
        }
        // 创建一个模拟数据库，其 C 方法返回上述模拟集合
        mockDB := &MockDatabase{
            CFunc: func(name string) Collectioner {
                if name != "mycollection" {
                    t.Errorf("Expected collection 'mycollection', got %s", name)
                }
                return mockCol
            },
        }

        err := myFunc(mockDB, testData)
        if err != nil {
            t.Errorf("myFunc returned an error for successful insert: %v", err)
        }
    })

    t.Run("failed insert", func(t *testing.T) {
        // 创建一个模拟集合，其 Insert 方法返回一个错误
        mockCol := &MockCollection{
            InsertFunc: func(docs ...interface{}) error {
                t.Log("Mock Insert called, returning error")
                return errors.New("mock insert error")
            },
        }
        // 创建一个模拟数据库，其 C 方法返回上述模拟集合
        mockDB := &MockDatabase{
            CFunc: func(name string) Collectioner {
                return mockCol
            },
        }

        err := myFunc(mockDB, testData)
        if err == nil {
            t.Errorf("myFunc did not return an error for failed insert")
        }
        expectedErr := "failed to insert data: mock insert error"
        if err.Error() != expectedErr {
            t.Errorf("Expected error message '%s', got '%s'", expectedErr, err.Error())
        }
    })
}

package main

import (
    "log"
    "time"
    // ... 其他导入保持不变
)

func main() {
    // 假设你已经有了一个 mgo.Session
    session, err := mgo.DialWithTimeout("mongodb://localhost:27017", 5*time.Second)
    if err != nil {
        log.Fatalf("Failed to connect to MongoDB: %v", err)
    }
    defer session.Close()

    // 获取真实的 mgo.Database 实例
    realDB := session.DB("mydatabase")

    // 真实数据
    prodData := MyData{
        ID:   bson.NewObjectId(),
        Name: "Production User",
        Age:  40,
    }

    // 将真实的 mgo.Database 实例传入 myFunc，它会自动满足 DataBaser 接口
    err = myFunc(realDB, prodData)
    if err != nil {
        log.Printf("Error in production call: %v", err)
    } else {
        log.Println("Production data processed successfully.")
    }
}