blob: c51a94baa85b2c0119d40c479e1e81a393cad0ca (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
using Microsoft.Data.Sqlite;
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;
namespace Timeline.Tests.Helpers
{
public class TestDatabase : IAsyncLifetime
{
private readonly bool _createUser;
public TestDatabase(bool createUser = true)
{
_createUser = createUser;
Connection = new SqliteConnection("Data Source=:memory:;");
}
public async Task InitializeAsync()
{
await Connection.OpenAsync();
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());
var admin = await userService.CreateUser("admin", "adminpw");
await userService.ModifyUser(admin.Id, new ModifyUserParams() { Nickname = "administrator" });
await context.SaveChangesAsync();
if (_createUser)
{
var user = await userService.CreateUser("user", "userpw");
await userService.ModifyUser(user.Id, new ModifyUserParams() { Nickname = "imuser" });
}
}
}
public async Task DisposeAsync()
{
await Connection.CloseAsync();
await Connection.DisposeAsync();
}
public SqliteConnection Connection { get; }
public DatabaseContext CreateContext(ITestOutputHelper? testOutputHelper = null)
{
var optionsBuilder = new DbContextOptionsBuilder<DatabaseContext>()
.UseSqlite(Connection);
if (testOutputHelper != null) optionsBuilder.LogTo(testOutputHelper.WriteLine).EnableDetailedErrors().EnableSensitiveDataLogging();
return new DatabaseContext(optionsBuilder.Options);
}
}
}
|
