2 files changed, 109 insertions, 0 deletions
diff --git a/BackEnd/Timeline.Tests/ServiceTests/ServiceTestBase.cs b/BackEnd/Timeline.Tests/ServiceTests/ServiceTestBase.cs
new file mode 100644
index 00000000..0019de02
--- /dev/null
+++ b/BackEnd/Timeline.Tests/ServiceTests/ServiceTestBase.cs
@@ -0,0 +1,67 @@
+using Microsoft.Extensions.Logging.Abstractions;
+using System.Threading.Tasks;
+using Timeline.Entities;
+using Timeline.Services.User;
+using Timeline.Tests.Helpers;
+using Xunit;
+using Xunit.Abstractions;
+
+namespace Timeline.Tests.ServiceTests
+{
+    public abstract class ServiceTestBase : IAsyncLifetime
+    {
+        private readonly ITestOutputHelper? _testOutputHelper;
+
+        protected TestDatabase TestDatabase { get; }
+        protected DatabaseContext Database { get; private set; } = default!;
+
+        protected TestClock Clock { get; } = new TestClock();
+        protected UserService UserService { get; private set; } = default!;
+
+        protected long AdminId { get; private set; }
+        protected long UserId { get; private set; }
+
+        protected ServiceTestBase(ITestOutputHelper? testOutputHelper = null)
+        {
+            _testOutputHelper = testOutputHelper;
+            TestDatabase = new TestDatabase();
+        }
+
+        public async Task InitializeAsync()
+        {
+            await TestDatabase.InitializeAsync();
+            Database = TestDatabase.CreateContext(_testOutputHelper);
+
+            UserService = new UserService(NullLogger<UserService>.Instance, Database, new PasswordService(), Clock);
+
+            AdminId = await UserService.GetUserIdByUsernameAsync("admin");
+            UserId = await UserService.GetUserIdByUsernameAsync("user");
+
+            await OnInitializeAsync();
+            OnInitialize();
+        }
+
+        public async Task DisposeAsync()
+        {
+            OnDispose();
+            await OnDisposeAsync();
+            await Database.DisposeAsync();
+            await TestDatabase.DisposeAsync();
+        }
+
+
+        protected virtual void OnInitialize() { }
+        protected virtual void OnDispose() { }
+
+
+        protected virtual Task OnInitializeAsync()
+        {
+            return Task.CompletedTask;
+        }
+
+        protected virtual Task OnDisposeAsync()
+        {
+            return Task.CompletedTask;
+        }
+    }
+}
diff --git a/BackEnd/Timeline.Tests/ServiceTests/User/RegisterCode/RegisterCodeServiceTest.cs b/BackEnd/Timeline.Tests/ServiceTests/User/RegisterCode/RegisterCodeServiceTest.cs
new file mode 100644
index 00000000..6542696f
--- /dev/null
+++ b/BackEnd/Timeline.Tests/ServiceTests/User/RegisterCode/RegisterCodeServiceTest.cs
@@ -0,0 +1,42 @@
+using System.Threading.Tasks;
+using FluentAssertions;
+using Timeline.Services.User.RegisterCode;
+using Xunit;
+
+namespace Timeline.Tests.ServiceTests.User.RegisterCode
+{
+    public class RegisterCodeServiceTest : ServiceTestBase
+    {
+        private RegisterCodeService _registerCodeService = default!;
+
+        protected override void OnInitialize()
+        {
+            _registerCodeService = new RegisterCodeService(Database);
+        }
+
+        protected override void OnDispose()
+        {
+            _registerCodeService.Dispose();
+        }
+
+        [Fact]
+        public async Task Test()
+        {
+            var a = await _registerCodeService.GetCurrentCode(AdminId);
+            a.Should().BeNull();
+            var b = await _registerCodeService.CreateNewCode(AdminId);
+            b.Should().NotBeNullOrEmpty();
+            var c = await _registerCodeService.GetCurrentCode(AdminId);
+            c.Should().Be(b);
+            var d = await _registerCodeService.CreateNewCode(AdminId);
+            d.Should().NotBe(b);
+            var e = await _registerCodeService.GetCodeOwner(d);
+            e.Should().Be(AdminId);
+            var f = await _registerCodeService.GetCodeOwner(b);
+            f.Should().BeNull();
+            var g = await _registerCodeService.GetCodeOwner(b, false);
+            g.Should().Be(AdminId);
+        }
+    }
+}
+