mirror of https://github.com/dtm-labs/dtm.git
csharpjavadistributed-transactionsdtmgogolangmicroservicenodejsphpdatabasesagaseatatcctransactiontransactionsxapythondistributed
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
3.0 KiB
83 lines
3.0 KiB
package test
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/dtm-labs/dtm/client/dtmcli"
|
|
"github.com/dtm-labs/dtm/client/dtmcli/dtmimp"
|
|
"github.com/dtm-labs/dtm/client/dtmgrpc"
|
|
"github.com/dtm-labs/dtm/test/busi"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestMsgGrpcRedisDo(t *testing.T) {
|
|
before := getBeforeBalances("redis")
|
|
gid := dtmimp.GetFuncName()
|
|
req := busi.GenReqGrpc(30, false, false)
|
|
msg := dtmgrpc.NewMsgGrpc(DtmGrpcServer, gid).
|
|
Add(busi.BusiGrpc+"/busi.Busi/TransInRedis", req)
|
|
err := msg.DoAndSubmit(busi.BusiGrpc+"/busi.Busi/QueryPreparedRedis", func(bb *dtmcli.BranchBarrier) error {
|
|
return bb.RedisCheckAdjustAmount(busi.RedisGet(), busi.GetRedisAccountKey(busi.TransOutUID), -30, 86400)
|
|
})
|
|
assert.Nil(t, err)
|
|
waitTransProcessed(msg.Gid)
|
|
assert.Equal(t, []string{StatusSucceed}, getBranchesStatus(msg.Gid))
|
|
assert.Equal(t, StatusSucceed, getTransStatus(msg.Gid))
|
|
assertNotSameBalance(t, before, "redis")
|
|
}
|
|
|
|
func TestMsgGrpcRedisDoBusiFailed(t *testing.T) {
|
|
before := getBeforeBalances("redis")
|
|
gid := dtmimp.GetFuncName()
|
|
req := busi.GenReqGrpc(30, false, false)
|
|
msg := dtmgrpc.NewMsgGrpc(DtmGrpcServer, gid).
|
|
Add(busi.BusiGrpc+"/busi.Busi/TransInRedis", req)
|
|
err := msg.DoAndSubmit(busi.BusiGrpc+"/busi.Busi/QueryPreparedRedis", func(bb *dtmcli.BranchBarrier) error {
|
|
return errors.New("an error")
|
|
})
|
|
assert.Error(t, err)
|
|
assertSameBalance(t, before, "redis")
|
|
}
|
|
|
|
func TestMsgGrpcRedisDoPrepareFailed(t *testing.T) {
|
|
before := getBeforeBalances("redis")
|
|
gid := dtmimp.GetFuncName()
|
|
req := busi.GenReqGrpc(30, false, false)
|
|
msg := dtmgrpc.NewMsgGrpc(DtmGrpcServer+"not-exists", gid).
|
|
Add(busi.BusiGrpc+"/busi.Busi/TransInRedis", req)
|
|
err := msg.DoAndSubmit(busi.BusiGrpc+"/busi.Busi/QueryPreparedRedis", func(bb *dtmcli.BranchBarrier) error {
|
|
return bb.RedisCheckAdjustAmount(busi.RedisGet(), busi.GetRedisAccountKey(busi.TransOutUID), -30, 86400)
|
|
})
|
|
assert.Error(t, err)
|
|
assertSameBalance(t, before, "redis")
|
|
}
|
|
|
|
func TestMsgGrpcRedisDoCommitFailed(t *testing.T) {
|
|
before := getBeforeBalances("redis")
|
|
gid := dtmimp.GetFuncName()
|
|
req := busi.GenReqGrpc(30, false, false)
|
|
msg := dtmgrpc.NewMsgGrpc(DtmGrpcServer, gid).
|
|
Add(busi.BusiGrpc+"/busi.Busi/TransInRedis", req)
|
|
err := msg.DoAndSubmit(busi.BusiGrpc+"/busi.Busi/QueryPreparedRedis", func(bb *dtmcli.BranchBarrier) error {
|
|
return errors.New("after commit error")
|
|
})
|
|
assert.Error(t, err)
|
|
assertSameBalance(t, before, "redis")
|
|
}
|
|
|
|
func TestMsgGrpcRedisDoCommitAfterFailed(t *testing.T) {
|
|
before := getBeforeBalances("redis")
|
|
gid := dtmimp.GetFuncName()
|
|
req := busi.GenReqGrpc(30, false, false)
|
|
msg := dtmgrpc.NewMsgGrpc(DtmGrpcServer, gid).
|
|
Add(busi.BusiGrpc+"/busi.Busi/TransInRedis", req)
|
|
err := msg.DoAndSubmit(busi.BusiGrpc+"/busi.Busi/QueryPreparedRedis", func(bb *dtmcli.BranchBarrier) error {
|
|
err := bb.RedisCheckAdjustAmount(busi.RedisGet(), busi.GetRedisAccountKey(busi.TransOutUID), -30, 86400)
|
|
dtmimp.E2P(err)
|
|
return errors.New("an error")
|
|
})
|
|
assert.Error(t, err)
|
|
waitTransProcessed(gid)
|
|
assertNotSameBalance(t, before, "redis")
|
|
}
|
|
|