如何使用Sqlmock对GORM应用进行单元测试

小伙伴们有没有觉得学习数据库很有意思？有意思就对了！今天就给大家带来《如何使用Sqlmock对GORM应用进行单元测试》，以下内容将会涉及到MySQL、gorm、单元测试、Mock，若是在学习中对其中部分知识点有疑问，或许看了本文就能帮到你！

概述

对DB交互代码进行单元测试并不容易，当涉及到诸如GORM之类的ORM库时，这将变得更加困难。

从理论上讲，我们可以使用强大的模拟工具GoMock)来模拟

// modle.go
import "github.com/lib/pq"
...
type Blog struct {
    ID        uint
    Title     string
    Content   string
    Tags      pq.StringArray // string array for tags
    CreatedAt time.Time
}


// repository.go
import "github.com/jinzhu/gorm"
...

type Repository struct {
    db *gorm.DB
}

func (p *Repository) ListAll() ([]*Blog, error) {
    var l []*Blog
    err := p.db.Find(&l).Error
    return l, err
}

func (p *Repository) Load(id uint) (*Blog, error) {
    blog := &Blog{}
    err := p.db.Where(`id = ?`, id).First(blog).Error
    return blog, err
}

...

Tips: 注意

import (
    ...
  
    . "github.com/onsi/ginkgo"
    . "github.com/onsi/gomega"
    "github.com/DATA-DOG/go-sqlmock"
    "github.com/jinzhu/gorm"
)

var _ = Describe("Repository", func() {
    var repository *Repository
    var mock sqlmock.Sqlmock

    BeforeEach(func() {
        var db *sql.DB
        var err error

        db, mock, err = sqlmock.New() // mock sql.DB
        Expect(err).ShouldNot(HaveOccurred())

        gdb, err := gorm.Open("postgres", db) // open gorm db
        Expect(err).ShouldNot(HaveOccurred())

        repository = &Repository{db: gdb}
    })
    AfterEach(func() {
        err := mock.ExpectationsWereMet() // make sure all expectations were met
        Expect(err).ShouldNot(HaveOccurred())
    })
  
    It("test something", func(){
        ...
    })
})

要将Sqlmock与GORM一起使用，我们需要在

// repository.go
...
func (p *Repository) ListAll() ([]*Blog, error) {
    var l []*Blog
    err := p.db.Find(&l).Error
    return l, err
}
...



// repository_test.go
...
Context("list all", func() {
    It("empty", func() {
        
        const sqlSelectAll = `SELECT * FROM "blogs"`
        
        mock.ExpectQuery(sqlSelectAll).
            WillReturnRows(sqlmock.NewRows(nil))

        l, err := repository.ListAll()
        Expect(err).ShouldNot(HaveOccurred())
        Expect(l).Should(BeEmpty())
    })
})
...

如上面的代码片段所示，

➜ ginkgo     
Running Suite: Pg Suite
=======================
Random Seed: 1585542357
Will run 8 of 8 specs


(/Users/dche423/dbtest/pg/repository.go:24) 
[2020-03-30 12:26:01]  Query: could not match actual sql: "SELECT * FROM "blogs"" with expected regexp "SELECT * FROM "blogs"" 
• Failure [0.001 seconds]
Repository
/Users/dche423/dbtest/pg/repository_test.go:16
  list all
  /Users/dche423/dbtest/pg/repository_test.go:37
    empty [It]
    /Users/dche423/dbtest/pg/repository_test.go:38

...
Test Suite Failed
➜  

您可能会对这个简单的测试用例失败感到惊讶。但是控制台日志为我们提供了线索：“could not match actual sql with expected regexp.(翻译过来就是：无法将实际的sql与预期的regexp相匹配。)”

事实证明Sqlmock使用

// repository.go
func (p *Repository) Load(id uint) (*Blog, error) {
    blog := &Blog{}
    err := p.db.Where(`id = ?`, id).First(blog).Error
    return blog, err
}
...


// repository_test.go
Context("load", func() {
        It("found", func() {
                blog := &Blog{
                        ID:        1,
                        Title:     "post",
                        ...
                }

                rows := sqlmock.
                        NewRows([]string{"id", "title", "content", "tags", "created_at"}).
                        AddRow(blog.ID, blog.Title, blog.Content, blog.Tags, blog.CreatedAt)

                const sqlSelectOne = `SELECT * FROM "blogs" WHERE (id = $1) ORDER BY "blogs"."id" ASC LIMIT 1`

                mock.ExpectQuery(regexp.QuoteMeta(sqlSelectOne)).WithArgs(blog.ID).WillReturnRows(rows)

                dbBlog, err := repository.Load(blog.ID)
                Expect(err).ShouldNot(HaveOccurred())
                Expect(dbBlog).Should(Equal(blog))
        })

        It("not found", func() {
                // ignore sql match
                mock.ExpectQuery(`.+`).WillReturnRows(sqlmock.NewRows(nil))
                _, err := repository.Load(1)
                Expect(err).Should(Equal(gorm.ErrRecordNotFound))
        })
})
...

// repository.go
...
func (p *Repository) Save(blog *Blog) error {
    return p.db.Save(blog).Error
}


// repository_test.go
...
Context("save", func() {
        var blog *Blog
        BeforeEach(func() {
                blog = &Blog{
                        Title:     "post",
                        Content:   "hello",
                        Tags:      pq.StringArray{"a", "b"},
                        CreatedAt: time.Now(),
                }
        })

        It("insert", func() {
                // gorm use query instead of exec
                // https://github.com/DATA-DOG/go-sqlmock/issues/118
                const sqlInsert = `
                                INSERT INTO "blogs" ("title","content","tags","created_at") 
                                        VALUES ($1,$2,$3,$4) RETURNING "blogs"."id"`
                const newId = 1
                mock.ExpectBegin() // begin transaction
                mock.ExpectQuery(regexp.QuoteMeta(sqlInsert)).
                        WithArgs(blog.Title, blog.Content, blog.Tags, blog.CreatedAt).
                        WillReturnRows(sqlmock.NewRows([]string{"id"}).AddRow(newId))
                mock.ExpectCommit() // commit transaction

                Expect(blog.ID).Should(BeZero())

                err := repository.Save(blog)
                Expect(err).ShouldNot(HaveOccurred())

                Expect(blog.ID).Should(BeEquivalentTo(newId))
        })
    
    It("update", func() {
        ...        
    })
        

})

当数据Model具有主键时，

Save

上面的代码段显示了后一种情况。

我们创建一个新的博客实例，而不设置其主键。然后，使用

mock.ExpectQuery

通常，非查询SQL期望值（例如，插入/更新）应由

mock.ExpectExec

QueryRow

Exec

postgres

最后，我们使用

Expect(blog.ID).Should(BeEquivalentTo(*newId*))

blog.ID

Save

提示：如果您使用的是PostgreSQL，请对GORM模型插入使用

mock.ExpectQuery

您可能建议不必对简单的“插入/更新”操作进行单元测试。实际上，是的，没有必要。我们要向您展示的是，GORM可能会执行一些您之前没有注意到的隐式操作。

结论

Sqlmock是对DB交互式代码进行单元测试的好工具，但是在使用GORM和PostgreSQL时有一些陷阱。

在本文中，我们构建了一个简单的博客应用程序，并使用Sqlmock对它进行了单元测试。我相信您可以在此示例的帮助下开始单元测试。

有关完整的源代码，请访问 这个仓库。

文章来源: https://1024casts.com/topics/R9re7QDaq8MnJoaXRZxdljbNA5BwoK

今天关于《如何使用Sqlmock对GORM应用进行单元测试》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于mysql的内容请关注golang学习网公众号！