From 6e067a28d8527726a2a17045bef0f0e3d3430ed5 Mon Sep 17 00:00:00 2001 From: crupest Date: Fri, 23 Apr 2021 17:28:18 +0800 Subject: refactor: Refactor a lot. --- BackEnd/Timeline.Tests/Helpers/TestDatabase.cs | 17 ++++++++--------- 1 file changed, 8 insertions(+), 9 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 c51a94ba..d37aace4 100644 --- a/BackEnd/Timeline.Tests/Helpers/TestDatabase.cs +++ b/BackEnd/Timeline.Tests/Helpers/TestDatabase.cs @@ -12,11 +12,11 @@ namespace Timeline.Tests.Helpers { public class TestDatabase : IAsyncLifetime { - private readonly bool _createUser; + private readonly bool _init; - public TestDatabase(bool createUser = true) + public TestDatabase(bool init = true) { - _createUser = createUser; + _init = init; Connection = new SqliteConnection("Data Source=:memory:;"); } @@ -24,9 +24,11 @@ namespace Timeline.Tests.Helpers { await Connection.OpenAsync(); - using (var context = CreateContext()) + if (_init) { + using var context = CreateContext(); await context.Database.EnsureCreatedAsync(); + context.JwtToken.Add(new JwtTokenEntity { Key = JwtTokenGenerateHelper.GenerateKey() @@ -42,11 +44,8 @@ namespace Timeline.Tests.Helpers await context.SaveChangesAsync(); - if (_createUser) - { - 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" }); } } -- cgit v1.2.3