BackEnd/Timeline.Tests/Helpers/TestDatabase.cs


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

using Microsoft.Data.Sqlite;
using Microsoft.EntityFrameworkCore;
using Microsoft.Extensions.Logging.Abstractions;
using System.Threading.Tasks;
using Timeline.Entities;
using Timeline.Services;
using Xunit;
using Xunit.Abstractions;

namespace Timeline.Tests.Helpers
{
    public class TestDatabase : IAsyncLifetime
    {
        public TestDatabase()
        {
            Connection = new SqliteConnection("Data Source=:memory:;");
        }

        public async Task InitializeAsync()
        {
            await Connection.OpenAsync();

            using var context = CreateContext();
            await context.Database.MigrateAsync();

            var userService = new UserService(NullLogger<UserService>.Instance, context, new PasswordService(), new Clock());

            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" });
        }

        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);
        }
    }
}