1 files changed, 62 insertions, 11 deletions
diff --git a/BackEnd/Timeline.Tests/ServiceTests/User/RegisterCode/RegisterCodeServiceTest.cs b/BackEnd/Timeline.Tests/ServiceTests/User/RegisterCode/RegisterCodeServiceTest.cs
index 6542696f..a0e3d20e 100644
--- a/BackEnd/Timeline.Tests/ServiceTests/User/RegisterCode/RegisterCodeServiceTest.cs
+++ b/BackEnd/Timeline.Tests/ServiceTests/User/RegisterCode/RegisterCodeServiceTest.cs
@@ -1,4 +1,5 @@
-using System.Threading.Tasks;
+using System;
+using System.Threading.Tasks;
 using FluentAssertions;
 using Timeline.Services.User.RegisterCode;
 using Xunit;
@@ -11,7 +12,7 @@ namespace Timeline.Tests.ServiceTests.User.RegisterCode
 
         protected override void OnInitialize()
         {
-            _registerCodeService = new RegisterCodeService(Database);
+            _registerCodeService = new RegisterCodeService(Database, UserService);
         }
 
         protected override void OnDispose()
@@ -20,23 +21,73 @@ namespace Timeline.Tests.ServiceTests.User.RegisterCode
         }
 
         [Fact]
-        public async Task Test()
+        public async Task RegisterCodeTest()
         {
-            var a = await _registerCodeService.GetCurrentCode(AdminId);
+            var a = await _registerCodeService.GetCurrentCodeAsync(AdminId);
             a.Should().BeNull();
-            var b = await _registerCodeService.CreateNewCode(AdminId);
+            var b = await _registerCodeService.CreateNewCodeAsync(AdminId);
             b.Should().NotBeNullOrEmpty();
-            var c = await _registerCodeService.GetCurrentCode(AdminId);
+            var c = await _registerCodeService.GetCurrentCodeAsync(AdminId);
             c.Should().Be(b);
-            var d = await _registerCodeService.CreateNewCode(AdminId);
+            var d = await _registerCodeService.CreateNewCodeAsync(AdminId);
             d.Should().NotBe(b);
-            var e = await _registerCodeService.GetCodeOwner(d);
+            var e = await _registerCodeService.GetCodeOwnerAsync(d);
             e.Should().Be(AdminId);
-            var f = await _registerCodeService.GetCodeOwner(b);
+            var f = await _registerCodeService.GetCodeOwnerAsync(b);
             f.Should().BeNull();
-            var g = await _registerCodeService.GetCodeOwner(b, false);
+            var g = await _registerCodeService.GetCodeOwnerAsync(b, false);
             g.Should().Be(AdminId);
         }
+
+        [Fact]
+        public async Task RegisterInfoTest()
+        {
+            var registerCode = await _registerCodeService.CreateNewCodeAsync(AdminId);
+
+            var a = await _registerCodeService.GetUserRegisterInfoAsync(UserId);
+            a.Should().BeNull();
+
+            var dateTime = DateTime.UtcNow;
+
+            var b = await _registerCodeService.CreateRegisterInfoAsync(UserId, registerCode, dateTime);
+            b.UserId.Should().Be(UserId);
+            b.RegisterCode.Should().Be(registerCode);
+            b.IntroducerId.Should().Be(AdminId);
+            b.RegisterTime.Should().Be(dateTime);
+
+            var c = await _registerCodeService.GetUserRegisterInfoAsync(UserId);
+            c.Should().NotBeNull();
+            c!.UserId.Should().Be(UserId);
+            c.RegisterCode.Should().Be(registerCode);
+            c.IntroducerId.Should().Be(AdminId);
+            c.RegisterTime.Should().Be(dateTime);
+        }
+
+        [Fact]
+        public async Task RegisterInfoTestWithOldCode()
+        {
+            await _registerCodeService.CreateNewCodeAsync(AdminId);
+
+            // Make the old code disabled.
+            var registerCode = await _registerCodeService.CreateNewCodeAsync(AdminId);
+
+            var a = await _registerCodeService.GetUserRegisterInfoAsync(UserId);
+            a.Should().BeNull();
+
+            var dateTime = DateTime.UtcNow;
+
+            var b = await _registerCodeService.CreateRegisterInfoAsync(UserId, registerCode, dateTime);
+            b.UserId.Should().Be(UserId);
+            b.RegisterCode.Should().Be(registerCode);
+            b.IntroducerId.Should().Be(AdminId);
+            b.RegisterTime.Should().Be(dateTime);
+
+            var c = await _registerCodeService.GetUserRegisterInfoAsync(UserId);
+            c.Should().NotBeNull();
+            c!.UserId.Should().Be(UserId);
+            c.RegisterCode.Should().Be(registerCode);
+            c.IntroducerId.Should().Be(AdminId);
+            c.RegisterTime.Should().Be(dateTime);
+        }
     }
 }
-