From b0b95bd1c7cf97bfa9fc2886f10a871a7c252b6a Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Fri, 23 Apr 2021 17:44:02 +0800
Subject: test: Test database now use migrator.

---
 BackEnd/Timeline.Tests/Helpers/TestDatabase.cs | 34 +++++++-------------------
 1 file changed, 9 insertions(+), 25 deletions(-)

(limited to 'BackEnd/Timeline.Tests/Helpers/TestDatabase.cs')

diff --git a/BackEnd/Timeline.Tests/Helpers/TestDatabase.cs b/BackEnd/Timeline.Tests/Helpers/TestDatabase.cs
index d37aace4..c3b61919 100644
--- a/BackEnd/Timeline.Tests/Helpers/TestDatabase.cs
+++ b/BackEnd/Timeline.Tests/Helpers/TestDatabase.cs
@@ -3,7 +3,6 @@ using Microsoft.EntityFrameworkCore;
 using Microsoft.Extensions.Logging.Abstractions;
 using System.Threading.Tasks;
 using Timeline.Entities;
-using Timeline.Migrations;
 using Timeline.Services;
 using Xunit;
 using Xunit.Abstractions;
@@ -12,11 +11,8 @@ namespace Timeline.Tests.Helpers
 {
     public class TestDatabase : IAsyncLifetime
     {
-        private readonly bool _init;
-
-        public TestDatabase(bool init = true)
+        public TestDatabase()
         {
-            _init = init;
             Connection = new SqliteConnection("Data Source=:memory:;");
         }
 
@@ -24,29 +20,17 @@ namespace Timeline.Tests.Helpers
         {
             await Connection.OpenAsync();
 
-            if (_init)
-            {
-                using var context = CreateContext();
-                await context.Database.EnsureCreatedAsync();
-
-                context.JwtToken.Add(new JwtTokenEntity
-                {
-                    Key = JwtTokenGenerateHelper.GenerateKey()
-                });
-
-                await context.SaveChangesAsync();
-
-                var passwordService = new PasswordService();
-                var userService = new UserService(NullLogger<UserService>.Instance, context, passwordService, new Clock());
+            using var context = CreateContext();
+            await context.Database.MigrateAsync();
 
-                var admin = await userService.CreateUser("admin", "adminpw");
-                await userService.ModifyUser(admin.Id, new ModifyUserParams() { Nickname = "administrator" });
+            var userService = new UserService(NullLogger<UserService>.Instance, context, new PasswordService(), new Clock());
 
-                await context.SaveChangesAsync();
+            await userService.ModifyUser(
+                await userService.GetUserIdByUsername("administrator"),
+                new ModifyUserParams() { Username = "admin", Password = "adminpw", Nickname = "administrator" });
 
-                var user = await userService.CreateUser("user", "userpw");
-                await userService.ModifyUser(user.Id, new ModifyUserParams() { Nickname = "imuser" });
-            }
+            var user = await userService.CreateUser("user", "userpw");
+            await userService.ModifyUser(user.Id, new ModifyUserParams() { Nickname = "imuser" });
         }
 
         public async Task DisposeAsync()
-- 
cgit v1.2.3