diff --git a/test/tcc_barrier_test.go b/test/tcc_barrier_test.go index 61d0a4f..4f511c0 100644 --- a/test/tcc_barrier_test.go +++ b/test/tcc_barrier_test.go @@ -28,6 +28,7 @@ func TestTccBarrierNormal(t *testing.T) { assert.Nil(t, err) waitTransProcessed(gid) assert.Equal(t, StatusSucceed, getTransStatus(gid)) + assert.Equal(t, []string{StatusPrepared, StatusSucceed, StatusPrepared, StatusPrepared, StatusSucceed, StatusPrepared}, getBranchesStatus(gid)) } func TestTccBarrierRollback(t *testing.T) { @@ -41,6 +42,7 @@ func TestTccBarrierRollback(t *testing.T) { assert.Error(t, err) waitTransProcessed(gid) assert.Equal(t, StatusFailed, getTransStatus(gid)) + assert.Equal(t, []string{StatusSucceed, StatusPrepared, StatusPrepared, StatusSucceed, StatusPrepared, StatusPrepared}, getBranchesStatus(gid)) } func TestTccBarrierDisorder(t *testing.T) { diff --git a/test/tcc_test.go b/test/tcc_test.go index f7bd709..2d19b07 100644 --- a/test/tcc_test.go +++ b/test/tcc_test.go @@ -21,6 +21,7 @@ func TestTccNormal(t *testing.T) { assert.Nil(t, err) waitTransProcessed(gid) assert.Equal(t, StatusSucceed, getTransStatus(gid)) + assert.Equal(t, []string{StatusPrepared, StatusSucceed, StatusPrepared, StatusPrepared, StatusSucceed, StatusPrepared}, getBranchesStatus(gid)) } func TestTccRollback(t *testing.T) { @@ -37,4 +38,5 @@ func TestTccRollback(t *testing.T) { assert.Equal(t, StatusAborting, getTransStatus(gid)) cronTransOnce() assert.Equal(t, StatusFailed, getTransStatus(gid)) + assert.Equal(t, []string{StatusSucceed, StatusPrepared, StatusPrepared, StatusSucceed, StatusPrepared, StatusPrepared}, getBranchesStatus(gid)) }