Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

单测: 使用xgo替换部分gomonkey进行mock #271

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

单测: 使用xgo替换部分gomonkey进行mock #271

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions go.mod
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@ require (
github.com/streadway/amqp v0.0.0-20190404075320-75d898a42a94
github.com/syndtr/goleveldb v1.0.0
github.com/xdg/scram v1.0.5
github.com/xhd2015/xgo/runtime v1.0.18
go.mongodb.org/mongo-driver v1.9.1
golang.org/x/sys v0.0.0-20210112080510-489259a85091
gopkg.in/mgo.v2 v2.0.0-20190816093944-a6b53ec6cb22
Expand Down
2 changes: 2 additions & 0 deletions go.sum
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -230,6 +230,8 @@ github.com/xdg/scram v1.0.5 h1:TuS0RFmt5Is5qm9Tm2SoD89OPqe4IRiFtyFY4iwWXsw=
github.com/xdg/scram v1.0.5/go.mod h1:lB8K/P019DLNhemzwFU4jHLhdvlE6uDZjXFejJXr49I=
github.com/xdg/stringprep v1.0.3 h1:cmL5Enob4W83ti/ZHuZLuKD/xqJfus4fVPwE+/BDm+4=
github.com/xdg/stringprep v1.0.3/go.mod h1:Jhud4/sHMO4oL310DaZAKk9ZaJ08SJfe+sJh0HrGL1Y=
github.com/xhd2015/xgo/runtime v1.0.18 h1:aYlGbfgE0RwSfeKWPSQnOMKoky2SuzlMG+I2KUYlkbI=
github.com/xhd2015/xgo/runtime v1.0.18/go.mod h1:9GBQ2SzJCzpD3T+HRN+2C0TUOGv7qIz4s0mad1xJ8Jo=
github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d/go.mod h1:rHwXgn7JulP+udvsHwJoVG1YGAP6VLg4y9I5dyZdqmA=
github.com/youmark/pkcs8 v0.0.0-20201027041543-1326539a0a0a h1:fZHgsYlfvtyqToslyjUt3VOPF4J7aK/3MPcK7xp3PDk=
github.com/youmark/pkcs8 v0.0.0-20201027041543-1326539a0a0a/go.mod h1:ul22v+Nro/R083muKhosV54bj5niojjWZvU8xrevuH4=
Expand Down
53 changes: 21 additions & 32 deletions input/mongo/batch_test.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,76 +4,72 @@ import (
"context"
"errors"
"fmt"
"github.com/agiledragon/gomonkey"
"testing"

inputDriver "github.com/brokercap/Bifrost/input/driver"
outputDriver "github.com/brokercap/Bifrost/plugin/driver"
. "github.com/smartystreets/goconvey/convey"
"github.com/xhd2015/xgo/runtime/mock"
"go.mongodb.org/mongo-driver/mongo"
"reflect"
"testing"
)

func TestMongoInput_BatchStart(t *testing.T) {
Convey("GetBatchTableList error", t, func() {
c := new(MongoInput)
patches := gomonkey.ApplyMethod(reflect.TypeOf(c), "GetBatchTableList", func(c *MongoInput) (dbTableList map[string][]string, err error) {
mock.Patch(c.GetBatchTableList, func() (dbTableList map[string][]string, err error) {
return nil, errors.New("GetBatchTableList error")
})
defer patches.Reset()
err := c.BatchStart()
So(err.Error(), ShouldEqual, "GetBatchTableList error")
})

Convey("CreateMongoClient error", t, func() {
c := new(MongoInput)
patches := gomonkey.ApplyMethod(reflect.TypeOf(c), "GetBatchTableList", func(c *MongoInput) (dbTableList map[string][]string, err error) {
mock.Patch(c.GetBatchTableList, func() (dbTableList map[string][]string, err error) {
dbTableList = make(map[string][]string, 0)
dbTableList["mytest"] = make([]string, 0)
dbTableList["mytest"] = append(dbTableList["mytest"], "tb_1")
return
})
patches.ApplyFunc(CreateMongoClient, func(uri string, ctx context.Context) (*mongo.Client, error) {
mock.Patch(CreateMongoClient, func(uri string, ctx context.Context) (*mongo.Client, error) {
return &mongo.Client{}, fmt.Errorf("mock error")
})
defer patches.Reset()
err := c.BatchStart()
So(err.Error(), ShouldEqual, "mock error")
})

Convey("TableBatchStart error", t, func() {
c := new(MongoInput)
patches := gomonkey.ApplyMethod(reflect.TypeOf(c), "GetBatchTableList", func(c *MongoInput) (dbTableList map[string][]string, err error) {
mock.Patch(c.GetBatchTableList, func() (dbTableList map[string][]string, err error) {
dbTableList = make(map[string][]string, 0)
dbTableList["mytest"] = make([]string, 0)
dbTableList["mytest"] = append(dbTableList["mytest"], "tb_1")
return
})
patches.ApplyFunc(CreateMongoClient, func(uri string, ctx context.Context) (*mongo.Client, error) {
mock.Patch(CreateMongoClient, func(uri string, ctx context.Context) (*mongo.Client, error) {
return &mongo.Client{}, nil
})
patches.ApplyMethod(reflect.TypeOf(c), "TableBatchStart", func(c *MongoInput, collection *mongo.Collection, perBatchLimit int) error {
mock.Patch(c.TableBatchStart, func(collection *mongo.Collection, perBatchLimit int) error {
return errors.New("TableBatchStart error")
})
defer patches.Reset()
err := c.BatchStart()
So(err.Error(), ShouldEqual, "TableBatchStart error")
})

Convey("normal", t, func() {
c := new(MongoInput)
patches := gomonkey.ApplyMethod(reflect.TypeOf(c), "GetBatchTableList", func(c *MongoInput) (dbTableList map[string][]string, err error) {
mock.Patch(c.GetBatchTableList, func() (dbTableList map[string][]string, err error) {
dbTableList = make(map[string][]string, 0)
dbTableList["mytest"] = make([]string, 0)
dbTableList["mytest"] = append(dbTableList["mytest"], "tb_1")
return
})
patches.ApplyFunc(CreateMongoClient, func(uri string, ctx context.Context) (*mongo.Client, error) {
mock.Patch(CreateMongoClient, func(uri string, ctx context.Context) (*mongo.Client, error) {
return &mongo.Client{}, nil
})
patches.ApplyMethod(reflect.TypeOf(c), "TableBatchStart", func(c *MongoInput, collection *mongo.Collection, perBatchLimit int) error {
mock.Patch(c.TableBatchStart, func(collection *mongo.Collection, perBatchLimit int) error {
return nil
})
defer patches.Reset()
err := c.BatchStart()
So(err, ShouldBeNil)
})
Expand All @@ -83,37 +79,35 @@ func TestMongoInput_GetBatchTableList(t *testing.T) {
Convey("* table,get GetSchemaList error", t, func() {
c := new(MongoInput)
c.AddReplicateDoDb0("*", "*")
patches := gomonkey.ApplyMethod(reflect.TypeOf(c), "GetSchemaList", func(c *MongoInput) (data []string, err error) {
mock.Patch(c.GetSchemaList, func() (data []string, err error) {
err = errors.New("GetSchemaList error")
return
})
defer patches.Reset()
_, err := c.GetBatchTableList()
So(err.Error(), ShouldEqual, "GetSchemaList error")
})

Convey("* table,get GetSchemaTableList error", t, func() {
c := new(MongoInput)
c.AddReplicateDoDb0("*", "*")
patches := gomonkey.ApplyMethod(reflect.TypeOf(c), "GetSchemaList", func(c *MongoInput) (data []string, err error) {
mock.Patch(c.GetSchemaList, func() (data []string, err error) {
return []string{"mytest"}, nil
})
patches.ApplyMethod(reflect.TypeOf(c), "GetSchemaTableList", func(c *MongoInput, schema string) (tableList []inputDriver.TableList, err error) {
mock.Patch(c.GetSchemaTableList, func(schema string) (tableList []inputDriver.TableList, err error) {
err = errors.New("GetSchemaTableList error")
return
})
defer patches.Reset()
_, err := c.GetBatchTableList()
So(err.Error(), ShouldEqual, "GetSchemaTableList error")
})

Convey("* table,normal", t, func() {
c := new(MongoInput)
c.AddReplicateDoDb0("*", "*")
patches := gomonkey.ApplyMethod(reflect.TypeOf(c), "GetSchemaList", func(c *MongoInput) (data []string, err error) {
mock.Patch(c.GetSchemaList, func() (data []string, err error) {
return []string{"mytest"}, nil
})
patches.ApplyMethod(reflect.TypeOf(c), "GetSchemaTableList", func(c *MongoInput, schema string) (tableList []inputDriver.TableList, err error) {
mock.Patch(c.GetSchemaTableList, func(schema string) (tableList []inputDriver.TableList, err error) {
tableList = append(tableList, inputDriver.TableList{
TableName: "tb_1",
})
Expand All @@ -125,7 +119,6 @@ func TestMongoInput_GetBatchTableList(t *testing.T) {
})
return
})
defer patches.Reset()
dbTableList, err := c.GetBatchTableList()
So(err, ShouldBeNil)
So(len(dbTableList), ShouldEqual, 1)
Expand All @@ -137,7 +130,7 @@ func TestMongoInput_GetBatchTableList(t *testing.T) {
c.AddReplicateDoDb0("mytest", "tb_1")
c.AddReplicateDoDb0("mytest", "tb_2")
c.AddReplicateDoDb0("mytest_2", "tb_1")
patches := gomonkey.ApplyMethod(reflect.TypeOf(c), "GetSchemaTableList", func(c *MongoInput, schema string) (tableList []inputDriver.TableList, err error) {
mock.Patch(c.GetSchemaTableList, func(schema string) (tableList []inputDriver.TableList, err error) {
tableList = append(tableList, inputDriver.TableList{
TableName: "tb_1",
})
Expand All @@ -149,7 +142,6 @@ func TestMongoInput_GetBatchTableList(t *testing.T) {
})
return
})
defer patches.Reset()
dbTableList, err := c.GetBatchTableList()
So(err, ShouldBeNil)
So(len(dbTableList), ShouldEqual, 2)
Expand All @@ -162,11 +154,10 @@ func TestMongoInput_TableBatchStart(t *testing.T) {
Convey("GetCollectionDataList err", t, func() {
c := new(MongoInput)
client := &mongo.Client{}
patches := gomonkey.ApplyMethod(reflect.TypeOf(c), "GetCollectionDataList", func(c *MongoInput, ctx context.Context, collection *mongo.Collection, minId interface{}, perBatchLimit int) (batchResult []map[string]interface{}, err error) {
mock.Patch(c.GetCollectionDataList, func(ctx context.Context, collection *mongo.Collection, minId interface{}, perBatchLimit int) (batchResult []map[string]interface{}, err error) {
err = errors.New("GetCollectionDataList error")
return
})
defer patches.Reset()
collection := c.GetCollection(client, "mytest", "tb_1")
err := c.TableBatchStart(collection, 1000)
So(err, ShouldNotBeNil)
Expand All @@ -178,7 +169,7 @@ func TestMongoInput_TableBatchStart(t *testing.T) {
c := new(MongoInput)
client := &mongo.Client{}
var GetCollectionDataListMockI = 0
patches := gomonkey.ApplyMethod(reflect.TypeOf(c), "GetCollectionDataList", func(c *MongoInput, ctx context.Context, collection *mongo.Collection, minId interface{}, perBatchLimit int) (batchResult []map[string]interface{}, err error) {
mock.Patch(c.GetCollectionDataList, func(ctx context.Context, collection *mongo.Collection, minId interface{}, perBatchLimit int) (batchResult []map[string]interface{}, err error) {
switch GetCollectionDataListMockI {
case 0:
for i := 0; i < perBatchLimit; i++ {
Expand All @@ -204,7 +195,6 @@ func TestMongoInput_TableBatchStart(t *testing.T) {
GetCollectionDataListMockI++
return
})
defer patches.Reset()

var callbackDataList []*outputDriver.PluginDataType
var callbackFun = func(data *outputDriver.PluginDataType) {
Expand All @@ -222,7 +212,7 @@ func TestMongoInput_TableBatchStart(t *testing.T) {
c := new(MongoInput)
client := &mongo.Client{}
var GetCollectionDataListMockI = 0
patches := gomonkey.ApplyMethod(reflect.TypeOf(c), "GetCollectionDataList", func(c *MongoInput, ctx context.Context, collection *mongo.Collection, minId interface{}, perBatchLimit int) (batchResult []map[string]interface{}, err error) {
mock.Patch(c.GetCollectionDataList, func(ctx context.Context, collection *mongo.Collection, minId interface{}, perBatchLimit int) (batchResult []map[string]interface{}, err error) {
switch GetCollectionDataListMockI {
case 0:
for i := 0; i < perBatchLimit; i++ {
Expand All @@ -241,7 +231,6 @@ func TestMongoInput_TableBatchStart(t *testing.T) {
GetCollectionDataListMockI++
return
})
defer patches.Reset()

var callbackDataList []*outputDriver.PluginDataType
var callbackFun = func(data *outputDriver.PluginDataType) {
Expand Down