How do I use sqlmock to mock a table count query (GORM HasTable) in Go?

Question

I'm using sqlmock to unit test a set of Mysql database query handlers written in Go. This works great for standard SELECT / INSERT queries. For our db healthcheck, I'm using GORM's DB.Migrator().HasTable() to make sure a particular table exists in our db. The functionality seems to be working properly, but I'm having difficulty mocking the query that's happening under the hood of HasTable(). Does anybody have any advice on how to use sqlmock to do this?

HasTable() is executing the following query: SELECT count(*) FROM information_schema.tables WHERE table_schema = '' AND table_name = '[table_name]' AND table_type = 'BASE TABLE'

My helper function for the test looks like this:

    sqlDB, mock, err := sqlmock.New()
    if err != nil {
        log.Panicln(err)
    }

    // This is the block I'm not sure about...
    mock.ExpectBegin()
    mock.ExpectQuery(regexp.QuoteMeta("SELECT count(*) FROM information_schema.tables WHERE table_schema = '' AND table_name = '[table_name]' AND table_type = 'BASE TABLE'")).
        WillReturnRows(sqlmock.NewRows([]string{"count(*)"}).
            AddRow(1))
    mock.ExpectCommit()

    setupSqlMock(sqlDB)
}

I've tried every combination of mock.ExpectExec(), mock.ExpectQuery(), and all of their methods that I can think of. Any ideas?

did you review examples on https://github.com/DATA-DOG/go-sqlmock/blob/master/examples/basic/basic_test.go ? — serge-v, Jan 21 '22 at 16:20
@Orion please share the error that you get while running the test. — Chandan, Apr 07 '22 at 12:29

score -1 · Answer 1 · edited Nov 28 '22 at 03:44

-1

Using sqlmock.QueryMatcherEqual you avoid the * parsing problem

sqlDB, mock, err := sqlmock.New(sqlmock.QueryMatcherOption(sqlmock.QueryMatcherEqual))

edited Nov 28 '22 at 03:44

codetiger

2,650
20
37

answered Nov 23 '22 at 12:41

Marco A. Buarque

1
1

How do I use sqlmock to mock a table count query (GORM HasTable) in Go?

1 Answers1