Go 单元测试 - 在使用 go-sqlmock 模拟 gorm 时无法将实际 sql 与预期的正则表达式匹配？

Question

在单元测试时，我使用 sqlmock 模拟 gorm。 但是当运行所有初始设置时都很好，但是当实际查询和测试查询匹配时会发生错误。 我在下面给出了所有代码。

用户去

func (r *users) GetUserByID(userID uint) (*domain.User, *errors.RestErr) {
    var resp domain.User

    res := r.DB.Model(&domain.User{}).Where("id = ?", userID).First(&resp)

    if res.RowsAffected == 0 {
        logger.Error("error occurred when getting user by user id", res.Error)
        return nil, errors.NewNotFoundError(errors.ErrRecordNotFound)
    }

    return &resp, nil
}

嘲笑

user_test.go

type Suite struct {
    suite.Suite
    DB         *gorm.DB
    mock       sqlmock.Sqlmock
    db         *sql.DB
    err        error
    repository repository.IUsers
}

func (s *Suite) SetupSuite() {
    s.db, s.mock, s.err = sqlmock.New(sqlmock.QueryMatcherOption(sqlmock.QueryMatcherEqual))

    require.NoError(s.T(), s.err)

    dialector := mysql.New(mysql.Config{
        DSN:                       "sqlmock_db_0",
        DriverName:                "mysql",
        Conn:                      s.db,
        SkipInitializeWithVersion: true,
    })

    s.DB, s.err = gorm.Open(dialector, &gorm.Config{})

    require.NoError(s.T(), s.err)
    s.repository = impl.NewMySqlUsersRepository(s.DB)
}

func (s *Suite) AfterTest(_, _ string) {
    require.NoError(s.T(), s.mock.ExpectationsWereMet())
}

func TestInit(t *testing.T) {
    suite.Run(t, new(Suite))
}

// .................Start Testing....................

func (s *Suite) Test_repository_Get() {
    var (
        id        = uint(1)
        user_name = "user-name"
        fast_name = "fast-name"
    )

    s.mock.ExpectQuery(regexp.QuoteMeta(`SELECT * FROM "users" WHERE (id = $1) ORDER BY "users"."id" LIMIT 1`)).
        WithArgs(uint(id)).
        WillReturnRows(sqlmock.NewRows([]string{"id", "user_name", "fast_name"}).
            AddRow(uint(id), user_name, fast_name))

    res, _ := s.repository.GetUserByID(uint(id))
    require.Nil(s.T(), deep.Equal(&domain.User{ID: uint(id), UserName: user_name, LastName: fast_name}, res))
}

当我执行res, _ := s.repository.GetUserByID(uint(id))这一行时，它转到 user.go 文件。 这里在 sql 查询行中出现错误如下所示。

actual sql: "SELECT * FROM `users` WHERE id = ? ORDER BY `users`.`id` LIMIT 1" 
does not equal to expected "SELECT \* FROM "users" WHERE \(id = \$1\) ORDER BY "users"\."id" LIMIT 1"


=> expecting Query, QueryContext or QueryRow which:
     - matches sql: 'SELECT \* FROM "users" WHERE \(id = \$1\) ORDER BY "users"\."id" LIMIT 1'
     - is with arguments:
                0 - 1
     - should return rows:
                row 0 - [1 user-name fast-name]

这里有什么问题？

Answer 1

这里的主要问题是 sql 查询。 对于 Get-Item，这里不需要完整查询。 Gorm 自动生成这个。 只需要通过

"SELECT(.*)"

我是从 Medium 上的一篇博文中发现的。 在这里你可以找到详细信息

波纹管代码工作正常

func (s *Suite) Test_repository_GetUserByID() {
var (
    id         = uint(1)
    user_name  = "user-name"
    first_name = "first-name"
)
s.mock.ExpectQuery("SELECT(.*)").
    WithArgs(id).
    WillReturnRows(sqlmock.NewRows([]string{"id", "user_name", "first_name"}).
        AddRow(uint(id), user_name, first_name))

 res, _ := s.repository.GetUserByID(uint(id))
 require.Nil(s.T(), deep.Equal(&domain.User{ID: uint(id), UserName: user_name, FirstName: first_name}, res))
 }