From f99159e21741e425d0babcec5e058e82cd5b29ec Mon Sep 17 00:00:00 2001
From: yedf2 <120050102@qq.com>
Date: Fri, 5 Nov 2021 16:19:35 +0800
Subject: [PATCH] test msg reviewed

---
 test/msg_grpc_test.go    | 11 ++++++----
 test/msg_options_test.go | 44 ++++++++++++++++++++++++++++++++++++++++
 test/msg_test.go         |  4 ++--
 3 files changed, 53 insertions(+), 6 deletions(-)
 create mode 100644 test/msg_options_test.go

diff --git a/test/msg_grpc_test.go b/test/msg_grpc_test.go
index deffd90..c494fb5 100644
--- a/test/msg_grpc_test.go
+++ b/test/msg_grpc_test.go
@@ -17,23 +17,26 @@ func TestMsgGrpcNormal(t *testing.T) {
 	assert.Nil(t, err)
 	waitTransProcessed(msg.Gid)
 	assert.Equal(t, StatusSucceed, getTransStatus(msg.Gid))
+	assert.Equal(t, []string{StatusSucceed, StatusSucceed}, getBranchesStatus(msg.Gid))
 }
 
-func TestMsgGrpcOngoingSuccess(t *testing.T) {
+func TestMsgGrpcTimeoutSuccess(t *testing.T) {
 	msg := genGrpcMsg(dtmimp.GetFuncName())
 	err := msg.Prepare("")
 	assert.Nil(t, err)
 	examples.MainSwitch.CanSubmitResult.SetOnce(dtmcli.ResultOngoing)
 	cronTransOnceForwardNow(180)
 	assert.Equal(t, StatusPrepared, getTransStatus(msg.Gid))
-	examples.MainSwitch.TransInResult.SetOnce(dtmcli.ResultOngoing)
+	examples.MainSwitch.TransOutResult.SetOnce(dtmcli.ResultOngoing)
 	cronTransOnceForwardNow(180)
 	assert.Equal(t, StatusSubmitted, getTransStatus(msg.Gid))
+	assert.Equal(t, []string{StatusPrepared, StatusPrepared}, getBranchesStatus(msg.Gid))
 	cronTransOnce()
 	assert.Equal(t, StatusSucceed, getTransStatus(msg.Gid))
+	assert.Equal(t, []string{StatusSucceed, StatusSucceed}, getBranchesStatus(msg.Gid))
 }
 
-func TestMsgGrpcOngoingFailed(t *testing.T) {
+func TestMsgGrpcTimeoutFailed(t *testing.T) {
 	msg := genGrpcMsg(dtmimp.GetFuncName())
 	msg.Prepare("")
 	assert.Equal(t, StatusPrepared, getTransStatus(msg.Gid))
@@ -42,8 +45,8 @@ func TestMsgGrpcOngoingFailed(t *testing.T) {
 	assert.Equal(t, StatusPrepared, getTransStatus(msg.Gid))
 	examples.MainSwitch.CanSubmitResult.SetOnce(dtmcli.ResultFailure)
 	cronTransOnceForwardNow(180)
-	assert.Equal(t, []string{StatusPrepared, StatusPrepared}, getBranchesStatus(msg.Gid))
 	assert.Equal(t, StatusFailed, getTransStatus(msg.Gid))
+	assert.Equal(t, []string{StatusPrepared, StatusPrepared}, getBranchesStatus(msg.Gid))
 }
 
 func genGrpcMsg(gid string) *dtmgrpc.MsgGrpc {
diff --git a/test/msg_options_test.go b/test/msg_options_test.go
new file mode 100644
index 0000000..4928459
--- /dev/null
+++ b/test/msg_options_test.go
@@ -0,0 +1,44 @@
+package test
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/yedf/dtm/dtmcli"
+	"github.com/yedf/dtm/dtmcli/dtmimp"
+	"github.com/yedf/dtm/examples"
+)
+
+func TestMsgOptionsTimeout(t *testing.T) {
+	msg := genMsg(dtmimp.GetFuncName())
+	msg.Prepare("")
+	cronTransOnce()
+	assert.Equal(t, StatusPrepared, getTransStatus(msg.Gid))
+	cronTransOnceForwardNow(60)
+	assert.Equal(t, StatusSucceed, getTransStatus(msg.Gid))
+}
+
+func TestMsgOptionsTimeoutCustom(t *testing.T) {
+	msg := genMsg(dtmimp.GetFuncName())
+	msg.TimeoutToFail = 120
+	msg.Prepare("")
+	cronTransOnce()
+	assert.Equal(t, StatusPrepared, getTransStatus(msg.Gid))
+	cronTransOnceForwardNow(60)
+	assert.Equal(t, StatusPrepared, getTransStatus(msg.Gid))
+	cronTransOnceForwardNow(180)
+	assert.Equal(t, StatusSucceed, getTransStatus(msg.Gid))
+}
+
+func TestMsgOptionsTimeoutFailed(t *testing.T) {
+	msg := genMsg(dtmimp.GetFuncName())
+	msg.TimeoutToFail = 120
+	msg.Prepare("")
+	cronTransOnce()
+	assert.Equal(t, StatusPrepared, getTransStatus(msg.Gid))
+	cronTransOnceForwardNow(60)
+	assert.Equal(t, StatusPrepared, getTransStatus(msg.Gid))
+	examples.MainSwitch.CanSubmitResult.SetOnce(dtmcli.ResultFailure)
+	cronTransOnceForwardNow(180)
+	assert.Equal(t, StatusFailed, getTransStatus(msg.Gid))
+}
diff --git a/test/msg_test.go b/test/msg_test.go
index 5397397..1b10b16 100644
--- a/test/msg_test.go
+++ b/test/msg_test.go
@@ -18,7 +18,7 @@ func TestMsgNormal(t *testing.T) {
 	assert.Equal(t, StatusSucceed, getTransStatus(msg.Gid))
 }
 
-func TestMsgOngoingSuccess(t *testing.T) {
+func TestMsgTimeoutSuccess(t *testing.T) {
 	msg := genMsg(dtmimp.GetFuncName())
 	msg.Prepare("")
 	assert.Equal(t, StatusPrepared, getTransStatus(msg.Gid))
@@ -33,7 +33,7 @@ func TestMsgOngoingSuccess(t *testing.T) {
 	assert.Equal(t, StatusSucceed, getTransStatus(msg.Gid))
 }
 
-func TestMsgOngoingFailed(t *testing.T) {
+func TestMsgTimeoutFailed(t *testing.T) {
 	msg := genMsg(dtmimp.GetFuncName())
 	msg.Prepare("")
 	assert.Equal(t, StatusPrepared, getTransStatus(msg.Gid))