在没有数据库的情况下测试包含对sql.Open连接的调用的Golang函数

package main

import (
    "database/sql"
    "testing"

    "github.com/DATA-DOG/go-sqlmock"
)

func TestMyFunction(t *testing.T) {
    // 创建模拟数据库连接
    db, mock, err := sqlmock.New()
    if err != nil {
        t.Fatalf("Failed to create mock database connection: %v", err)
    }
    defer db.Close()

    // 设置预期的SQL查询和结果
    mock.ExpectQuery("SELECT * FROM users").WillReturnRows(sqlmock.NewRows([]string{"id", "name"}).AddRow(1, "John"))

    // 调用被测试的函数
    result := MyFunction(db)

    // 检查结果是否符合预期
    if result != "John" {
        t.Errorf("Unexpected result. Expected: John, Got: %s", result)
    }

    // 确保所有预期的SQL查询都被执行
    if err := mock.ExpectationsWereMet(); err != nil {
        t.Errorf("Unfulfilled expectations: %v", err)
    }
}

func MyFunction(db *sql.DB) string {
    // 在这里执行包含对sql.Open连接的调用的逻辑
    // 并返回结果
    return "John"
}