Merge pull request #18 from catcherwong/tcctest

tests: add ut for tcc

Author: catcherwong

src/Dtmcli.Tests/TccTests.cs

@@ -0,0 +1,98 @@
+using System.Collections.Generic;
+using System.Net.Http;
+using System.Threading;
+using System.Threading.Tasks;
+using Microsoft.Extensions.Logging.Abstractions;
+using Moq;
+using Xunit;
+
+namespace Dtmcli.Tests
+{
+    public class TccTests
+    {
+        [Fact]
+        public async void Execute_Should_Submit()
+        {
+            var dtmClient = new Mock<IDtmClient>();
+            MockTransCallDtm(dtmClient, Constant.Request.OPERATION_PREPARE, true);
+            MockTransRegisterBranch(dtmClient, Constant.Request.OPERATION_REGISTERBRANCH, true);
+            MockTransRequestBranch(dtmClient, System.Net.HttpStatusCode.OK);
+
+            var gid = "tcc_gid";
+            var globalTrans = new TccGlobalTransaction(dtmClient.Object, NullLoggerFactory.Instance);
+            var res = await globalTrans.Excecute(gid, async (tcc) =>
+            {
+                var res1 = await tcc.CallBranch(new { }, "http://localhost:9999/TransOutTry", "http://localhost:9999/TransOutConfirm", "http://localhost:9999/TransOutCancel", default);
+                var res2 = await tcc.CallBranch(new { }, "http://localhost:9999/TransInTry", "http://localhost:9999/TransInConfirm", "http://localhost:9999/TransInCancel", default);
+            });
+
+            Assert.Equal(gid, res);
+        }
+
+        [Fact]
+        public async void Execute_Should_Abort_When_CallBranch_With_Old_Ver_Exception()
+        {
+            var dtmClient = new Mock<IDtmClient>();
+            MockTransCallDtm(dtmClient, Constant.Request.OPERATION_PREPARE, true);
+            MockTransCallDtm(dtmClient, Constant.Request.OPERATION_ABORT, true);
+            MockTransRegisterBranch(dtmClient, Constant.Request.OPERATION_REGISTERBRANCH, true);
+            MockTransRequestBranch(dtmClient, System.Net.HttpStatusCode.OK, "FAILURE");
+
+            var gid = "tcc_gid";
+            var globalTrans = new TccGlobalTransaction(dtmClient.Object, NullLoggerFactory.Instance);
+            var res = await globalTrans.Excecute(gid, async (tcc) =>
+            {
+                var res1 = await tcc.CallBranch(new { }, "http://localhost:9999/TransOutTry", "http://localhost:9999/TransOutConfirm", "http://localhost:9999/TransOutCancel", default);
+                var res2 = await tcc.CallBranch(new { }, "http://localhost:9999/TransInTry", "http://localhost:9999/TransInConfirm", "http://localhost:9999/TransInCancel", default);
+            });
+
+            Assert.Empty(res);
+            dtmClient.Verify(x => x.TransCallDtm(It.IsAny<DtmImp.TransBase>(), It.IsAny<object>(), Constant.Request.OPERATION_ABORT, It.IsAny<CancellationToken>()), Times.Once);
+        }
+
+        [Fact]
+        public async void Execute_Should_Abort_When_CallBranch_With_New_Ver_Exception()
+        {
+            var dtmClient = new Mock<IDtmClient>();
+            MockTransCallDtm(dtmClient, Constant.Request.OPERATION_PREPARE, true);
+            MockTransCallDtm(dtmClient, Constant.Request.OPERATION_ABORT, true);
+            MockTransRegisterBranch(dtmClient, Constant.Request.OPERATION_REGISTERBRANCH, true);
+            MockTransRequestBranch(dtmClient, System.Net.HttpStatusCode.BadRequest);
+
+            var gid = "tcc_gid";
+            var globalTrans = new TccGlobalTransaction(dtmClient.Object, NullLoggerFactory.Instance);
+            var res = await globalTrans.Excecute(gid, async (tcc) =>
+            {
+                var res1 = await tcc.CallBranch(new { }, "http://localhost:9999/TransOutTry", "http://localhost:9999/TransOutConfirm", "http://localhost:9999/TransOutCancel", default);
+                var res2 = await tcc.CallBranch(new { }, "http://localhost:9999/TransInTry", "http://localhost:9999/TransInConfirm", "http://localhost:9999/TransInCancel", default);
+            });
+
+            Assert.Empty(res);
+            dtmClient.Verify(x => x.TransCallDtm(It.IsAny<DtmImp.TransBase>(), It.IsAny<object>(), Constant.Request.OPERATION_ABORT, It.IsAny<CancellationToken>()), Times.Once);
+        }
+
+        private void MockTransCallDtm(Mock<IDtmClient> mock, string op, bool result)
+        {
+            mock
+                .Setup(x => x.TransCallDtm(It.IsAny<DtmImp.TransBase>(), It.IsAny<object>(), op, It.IsAny<CancellationToken>()))
+                .Returns(Task.FromResult(result));
+        }
+
+        private void MockTransRegisterBranch(Mock<IDtmClient> mock, string op, bool result)
+        {
+            mock
+                .Setup(x => x.TransRegisterBranch(It.IsAny<DtmImp.TransBase>(), It.IsAny<Dictionary<string, string>>(), op, It.IsAny<CancellationToken>()))
+                .Returns(Task.FromResult(result));
+        }
+
+        private void MockTransRequestBranch(Mock<IDtmClient> mock, System.Net.HttpStatusCode statusCode, string content = "content")
+        {
+            var httpRspMsg = new HttpResponseMessage(statusCode);
+            httpRspMsg.Content = new StringContent(content);
+
+            mock
+                .Setup(x => x.TransRequestBranch(It.IsAny<DtmImp.TransBase>(), It.IsAny<HttpMethod>(), It.IsAny<object>(), It.IsAny<string>(), It.IsAny<string>(), It.IsAny<string>(), It.IsAny<CancellationToken>()))
+                .Returns(Task.FromResult(httpRspMsg));
+        }
+    }
+}

src/Dtmcli/ServiceCollectionExtensions.cs

@@ -5,9 +5,6 @@ namespace Dtmcli
 {
     public static class ServiceCollectionExtensions
     {
-        internal static IServiceCollection ServiceCollection;
-
-
         public static IServiceCollection AddDtmcli(this IServiceCollection services, Action<DtmOptions> setupAction)
         {
             if(setupAction == null)
@@ -16,7 +13,7 @@ public static IServiceCollection AddDtmcli(this IServiceCollection services, Act
             }
             var options = new DtmOptions();
             setupAction(options);
-            ServiceCollection = services;
+
             services.AddHttpClient("dtmClient", client =>
             {
                 client.BaseAddress = new Uri(options.DtmUrl);
@@ -26,7 +23,6 @@ public static IServiceCollection AddDtmcli(this IServiceCollection services, Act
 
             services.AddSingleton<TccGlobalTransaction>();
 
-            // 子事务屏障
             services.AddSingleton<IBranchBarrierFactory, DefaultBranchBarrierFactory>();
             return services;
         }