diff --git a/test/tcc_grpc_cover.go b/test/tcc_grpc_cover.go deleted file mode 100644 index 3cecef2..0000000 --- a/test/tcc_grpc_cover.go +++ /dev/null @@ -1,29 +0,0 @@ -package test - -import ( - "testing" - - "github.com/stretchr/testify/assert" - "github.com/yedf/dtm/dtmcli/dtmimp" - "github.com/yedf/dtm/dtmgrpc" - "github.com/yedf/dtm/examples" -) - -func TestTccGrpcCoverNotConnected(t *testing.T) { - gid := dtmimp.GetFuncName() - err := dtmgrpc.TccGlobalTransaction("localhost:01", gid, func(tcc *dtmgrpc.TccGrpc) error { - return nil - }) - assert.Error(t, err) -} - -func TestTccGrpcCoverPanic(t *testing.T) { - gid := dtmimp.GetFuncName() - err := dtmimp.CatchP(func() { - _ = dtmgrpc.TccGlobalTransaction(examples.DtmHttpServer, gid, func(tcc *dtmgrpc.TccGrpc) error { - panic("user panic") - }) - assert.FailNow(t, "not executed") - }) - assert.Contains(t, err.Error(), "user panic") -} diff --git a/test/tcc_grpc_cover_test.go b/test/tcc_grpc_cover_test.go new file mode 100644 index 0000000..f4c3fef --- /dev/null +++ b/test/tcc_grpc_cover_test.go @@ -0,0 +1,48 @@ +package test + +import ( + "testing" + + "github.com/stretchr/testify/assert" + "github.com/yedf/dtm/dtmcli/dtmimp" + "github.com/yedf/dtm/dtmgrpc" + "github.com/yedf/dtm/examples" + "google.golang.org/protobuf/types/known/emptypb" +) + +func TestTccGrpcCoverNotConnected(t *testing.T) { + gid := dtmimp.GetFuncName() + err := dtmgrpc.TccGlobalTransaction("localhost:01", gid, func(tcc *dtmgrpc.TccGrpc) error { + return nil + }) + assert.Error(t, err) +} + +func TestTccGrpcCoverPanic(t *testing.T) { + gid := dtmimp.GetFuncName() + err := dtmimp.CatchP(func() { + _ = dtmgrpc.TccGlobalTransaction(examples.DtmGrpcServer, gid, func(tcc *dtmgrpc.TccGrpc) error { + panic("user panic") + }) + assert.FailNow(t, "not executed") + }) + assert.Contains(t, err.Error(), "user panic") +} + +func TestTccGrpcCoverCallBranch(t *testing.T) { + req := examples.GenBusiReq(30, false, false) + gid := dtmimp.GetFuncName() + err := dtmgrpc.TccGlobalTransaction(examples.DtmGrpcServer, gid, func(tcc *dtmgrpc.TccGrpc) error { + + r := &emptypb.Empty{} + err := tcc.CallBranch(req, "not_exists://abc", examples.BusiGrpc+"/examples.Busi/TransOutConfirm", examples.BusiGrpc+"/examples.Busi/TransOutRevert", r) + assert.Error(t, err) + + tcc.Dtm = "localhost:01" + err = tcc.CallBranch(req, examples.BusiGrpc+"/examples.Busi/TransOut", examples.BusiGrpc+"/examples.Busi/TransOutConfirm", examples.BusiGrpc+"/examples.Busi/TransOutRevert", r) + assert.Error(t, err) + + return err + }) + assert.Error(t, err) +}