在进行软件开发时，经常会遇到需要模拟数据库连接的情况，以便进行测试。然而，有时候我们可能没有sqlmock这样的工具来帮助我们实现这个目标。那么，在没有sqlmock的情况下，我们应该如何模拟数据库的ping操作呢？在本文中，php小编柚子将为您介绍一种简单而有效的方法来实现这个目标，让您能够顺利进行数据库连接的模拟测试。
问题内容
我想了解如何使用 mock.mock 重构我的代码以模拟 db.ping() 。我想先在没有框架的情况下理解这个概念。
这是我要测试的代码：
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 登录后复制
我尝试了这篇文章如何模拟 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 登录后复制
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)
}
登录后复制
以上就是如何在没有 sqlmock 的情况下模拟 db ping的详细内容，更多请关注每日运维网(www.mryunwei.com)其它相关文章！
