如何在没有 sqlmock 的情况下模拟 db ping

在进行软件开发时，经常会遇到需要模拟数据库连接的情况，以便进行测试。然而，有时候我们可能没有sqlmock这样的工具来帮助我们实现这个目标。那么，在没有sqlmock的情况下，我们应该如何模拟数据库的ping操作呢？在本文中，php小编柚子将为您介绍一种简单而有效的方法来实现这个目标，让您能够顺利进行数据库连接的模拟测试。

问题内容

我想了解如何使用 mock.mock 重构我的代码以模拟 db.ping() 。我想先在没有框架的情况下理解这个概念。

这是我要测试的代码：

v0.dev

Vercel推出的AI生成式UI工具，通过文本描述生成UI组件代码

下载

func Connect() (*sql.DB, error) {

    db, err := sql.Open("mysql", "root:secret@tcp(s-maria-db)/s_db")

    if err != nil {
        return nil, err
    }

    for i := 0; i < 60; i++ {
        // I am hoping to mock this portion
        if err := db.Ping(); err == nil {
            break
        }
        time.Sleep(time.Second)
    }

    return db, nil
}

我尝试了这篇文章如何模拟 ping 命令，但这实际上并没有回答问题。如果是这样，我不理解该解决方案，因为它适用于我的代码。

解决方法

您需要使用单独的函数进行 ping，并且它需要接受一个接口，以便您可以有条件地传入真实的 *sql.db 或模拟。这是因为您无法重写结构上的方法。

// connect opens a connection to the database.
func connect() (*sql.db, error) {
    return sql.open("mysql", "root:secret@tcp(s-maria-db)/s_db")
}

// pinger defines an interface for pinging.
type pinger interface {
    ping() error
}

// ping attempts to ping the database, trying several times before failing.
func ping(p pinger) error {
    const maxattempts = 60
    var err error
    for i := 0; i < maxattempts; i++ {
        if err = p.ping(); err == nil {
            return nil
        }
        if i < maxattempts - 1 {
            time.sleep(time.second)
        }
    }
    return err
}

func main() {
    if err := run(); err != nil {
        log.fatal(err)
    }
}

func run() error {
    db, err := connect()
    if err != nil {
        return fmt.errorf("connecting to db: %w", err)
    }
    defer db.close()

    if err = ping(db); err != nil {
        return fmt.errorf("pinging db: %w", err)
    }
    
    ...
}

type mockDB struct {
    mock.Mock
}

func (m *mockDB) Ping() error {
    args := m.Called()
    return args.Error(0)
}

func TestPing(t *testing.T) {
    db := &mockDB{}
    db.On("Ping").Return(...)

    err := Ping(db)

    ...
    db.AssertExpectations(t)
}