From 9f189100c93c43372459c25984cb56ed00165f39 Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Mon, 18 Apr 2022 18:38:32 +0800
Subject: ...

---
 .../User/RegisterCode/RegisterCodeServiceTest.cs       | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

(limited to 'BackEnd/Timeline.Tests')

diff --git a/BackEnd/Timeline.Tests/ServiceTests/User/RegisterCode/RegisterCodeServiceTest.cs b/BackEnd/Timeline.Tests/ServiceTests/User/RegisterCode/RegisterCodeServiceTest.cs
index a0e3d20e..1afc94a1 100644
--- a/BackEnd/Timeline.Tests/ServiceTests/User/RegisterCode/RegisterCodeServiceTest.cs
+++ b/BackEnd/Timeline.Tests/ServiceTests/User/RegisterCode/RegisterCodeServiceTest.cs
@@ -1,6 +1,7 @@
 using System;
 using System.Threading.Tasks;
 using FluentAssertions;
+using Timeline.Services.User;
 using Timeline.Services.User.RegisterCode;
 using Xunit;
 
@@ -12,7 +13,7 @@ namespace Timeline.Tests.ServiceTests.User.RegisterCode
 
         protected override void OnInitialize()
         {
-            _registerCodeService = new RegisterCodeService(Database, UserService);
+            _registerCodeService = new RegisterCodeService(Database, UserService, Clock);
         }
 
         protected override void OnDispose()
@@ -89,5 +90,20 @@ namespace Timeline.Tests.ServiceTests.User.RegisterCode
             c.IntroducerId.Should().Be(AdminId);
             c.RegisterTime.Should().Be(dateTime);
         }
+
+        [Fact]
+        public async Task RegisterUserTest()
+        {
+            var registerCode = await _registerCodeService.CreateNewCodeAsync(AdminId);
+
+            var a = await _registerCodeService.RegisterUserWithCode(new CreateUserParams("user2", "user2pw"), registerCode);
+            a.Should().NotBeNull();
+
+            await _registerCodeService.CreateNewCodeAsync(AdminId);
+
+            await _registerCodeService.Awaiting(s => s.RegisterUserWithCode(new CreateUserParams("user3", "user3pw"), registerCode))
+                .Should().ThrowAsync<InvalidRegisterCodeException>();
+
+        }
     }
 }
-- 
cgit v1.2.3