本篇内容介绍了“go单元测试的方法有哪些”的有关知识,在实际案例的操作过程中,不少人都会遇到这样的困境,接下来就让小编带领大家学习一下如何处理这些情况吧!希望大家仔细阅读,能够学有所成!
gomock模拟对象的方式是让用户声明一个接口,然后使用gomock提供的mockgen工具生成mock对象代码。要模拟(mock)被测试代码的依赖对象时候,即可使用mock出来的对象来模拟和记录依赖对象的各种行为:比如最常用的返回值,调用次数等等。文字叙述有点抽象,直接上代码:
dick.go中DickFunc依赖外部对象OutterObj,本示例就是说明如何使用gomock框架控制所依赖的对象。
func DickFunc( outterObj MockInterface,para int)(result int){ fmt.Println("This init DickFunc") fmt.Println("call outter.func:") return outterObj.OutterFunc(para) }
mockgen工具命令是:
mockgen -source {source_file}.go -destination {dest_file}.go
比如,本示例即是:
mockgen -source src_mock.go -destination dst_mock.go
执行完后,可在同目录下找到生成的dst_mock.go文件,可以看到mockgen工具也实现了接口:
在本示例中只简单的更改了返回值,抛砖引玉:
func TestDickFunc(t *testing.T ){ mockCtrl := gomock.NewController(t) //defer mockCtrl.Finish() mockObj := dick.NewMockMockInterface(mockCtrl) mockObj.EXPECT().OutterFunc(3).Return(10) result :=dick.DickFunc(mockObj,3) t.Log("resutl:",result) }
使用go test命令执行这个单测
从结果看:本来应该输出3,最后输出就是10,和其他语言mock框架相似,生产出来的Mock对象不用自己去重定义这么麻烦。
由于go在网络架构上的优秀封装,使得go在很多网络场景被广泛使用,而http协议是其中重要部分,在面对http请求的时候,可以对http的client进行测试,算是mock的特殊应用场景。
看一个简单的示例就轻松的看懂了:
func TestHttp(t *testing.T) { handler := FruitServer() server := httptest.NewServer(handler) defer server.Close() e := httpexpect.New(t, server.URL) e.GET("/fruits"). Expect(). Status(http.StatusOK).JSON().Array().Empty() }
其中还支持对不同方法(包括Header,Post等)的构造以及返回值Json的自定义,更多细节查看其官网
还有一个testify使用起来可以说兼容了《一》中的gocheck和gomock,但是其mock使用稍微有点烦杂,使用继承tetify.Mock(匿名组合)重新实现需要Mock的接口,在这个接口里使用者自己使用Called(反射实现)被Mock的接口。
《单元测试的艺术》中认为stub和mock最大的区别就依赖对象是否和被测对象有交互,而从结果看就是桩对象不会使测试失败,它只是为被测对象提供依赖的对象,并不改变测试结果,而mock则会根据不同的交互测试要求,很可能会更改测试的结果。说了这么多理论,但其实这两种方法都不是割裂的,所以gomock框架除了像其名字一样可以模拟对象以外,还提供了桩对象的功能(stub)。以其实现来说,更像是一个桩对象的注入。但是因为兼容了多个有用的功能,所以其在社区最为火爆。
具体用法可参考其github主页
还有一种比较常见的场景就是和数据库的交互场景,go-sqlmock是sql模拟(Mock)驱动器,主要用于测试数据库的交互,go-sqlmock提供了完整的事务的执行测试框架,最新的版本(16.11.02)还支持prepare参数化提交和执行的Mock方案。
比如有这样的被测函数:
func recordStats(db *sql.DB, userID, productID int64) (err error) { tx, err := db.Begin() if err != nil { return } defer func() { switch err { case nil: err = tx.Commit() default: tx.Rollback() } }() if _, err = tx.Exec("UPDATE products SET views = views + 1"); err != nil { return } if _, err = tx.Exec("INSERT INTO product_viewers (user_id, product_id) VALUES (?, ?)", userID, productID); err != nil { return } return } func main() { db, err := sql.Open("mysql", "root@/root") if err != nil { panic(err) } defer db.Close() if err = recordStats(db, 1 , 5 ); err != nil { panic(err) } }
单测时:
func TestShouldUpdateStats(t *testing.T) { db, mock, err := sqlmock.New() if err != nil { t.Fatalf("mock error: '%s' ", err) } defer db.Close() mock.ExpectBegin() mock.ExpectExec("UPDATE products").WillReturnResult(sqlmock.NewResult(1, 1)) mock.ExpectExec("INSERT INTO product_viewers") .WithArgs(2, 3) .WillReturnResult(sqlmock.NewResult(1, 1)) mock.ExpectCommit() if err = recordStats(db, 2, 3); err != nil { t.Errorf("exe error: %s", err) } if err := mock.ExpectationsWereMet(); err != nil { t.Errorf("not implements: %s", err) } } //测试回滚 func TestShouldRollbackStatUpdatesOnFailure(t *testing.T) { db, mock, err := sqlmock.New() if err != nil { t.Fatalf("mock error: '%s'", err) } defer db.Close() mock.ExpectBegin() mock.ExpectExec("UPDATE products").WillReturnResult(sqlmock.NewResult(1, 1)) mock.ExpectExec("INSERT INTO product_viewers") .WithArgs(2, 3) .WillReturnError(fmt.Errorf("some error")) mock.ExpectRollback() // 执行被测方法,有错 if err = recordStats(db, 2, 3); err == nil { t.Errorf("not error") } // 执行被测方法,mock对象 if err := mock.ExpectationsWereMet(); err != nil { t.Errorf("not implements: %s", err) } }
“go单元测试的方法有哪些”的内容就介绍到这里了,感谢大家的阅读。如果想了解更多行业相关的知识可以关注亿速云网站,小编将为大家输出更多高质量的实用文章!
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。