fix ut failed.

changmingxie · May 30, 2021 · 826cc1b · 826cc1b
1 parent 0869402
commit 826cc1b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/...nit-test/src/main/java/org/mengyun/tcctransaction/unittest/client/AccountServiceStub.java b/...nit-test/src/main/java/org/mengyun/tcctransaction/unittest/client/AccountServiceStub.java
@@ -1,6 +1,7 @@
 package org.mengyun.tcctransaction.unittest.client;
 
 import org.mengyun.tcctransaction.api.TransactionContext;
+import org.mengyun.tcctransaction.api.TransactionStatus;
 import org.mengyun.tcctransaction.unittest.service.AccountService;
 import org.mengyun.tcctransaction.unittest.service.AccountServiceImpl;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -50,10 +51,12 @@ public void transferToWithTimeoutBeforeBranchTransactionStart(TransactionContext
             @Override
             public void run() {
 
-                try {
-                    Thread.sleep(4000l);
-                } catch (InterruptedException e) {
-                    e.printStackTrace();
+                if (transactionContext.getStatus() == TransactionStatus.TRYING.getId()) {
+                    try {
+                        Thread.sleep(4000l);
+                    } catch (InterruptedException e) {
+                        e.printStackTrace();
+                    }
                 }
                 accountService.transferTo(transactionContext, accountId, amount);
             }

diff --git a/...tion-unit-test/src/test/java/org/mengyun/tcctransaction/unittest/TransferServiceTest.java b/...tion-unit-test/src/test/java/org/mengyun/tcctransaction/unittest/TransferServiceTest.java
@@ -122,7 +122,7 @@ public void testTransferWithTimeoutAndCancelBeforeBranchTransactionStart() {
         Assert.assertEquals(MessageConstants.ACCOUNT_SERVICE_IMPL_TRANSFER_FROM_CANCEL_CALLED, messages.get(3));
 
         try {
-            Thread.sleep(10000l);
+            Thread.sleep(15000l);
         } catch (InterruptedException e) {
             e.printStackTrace();
         }