BackEnd/Timeline.Tests/Services/DatabaseBasedTest.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

using System.Threading.Tasks;
using Timeline.Entities;
using Timeline.Tests.Helpers;
using Xunit;

namespace Timeline.Tests.Services
{
    public abstract class DatabaseBasedTest : IAsyncLifetime
    {
        protected TestDatabase TestDatabase { get; }
        protected DatabaseContext Database { get; private set; } = default!;

        protected DatabaseBasedTest(bool databaseCreateUsers = true)
        {
            TestDatabase = new TestDatabase(databaseCreateUsers);
        }

        public async Task InitializeAsync()
        {
            await TestDatabase.InitializeAsync();
            Database = TestDatabase.CreateContext();
            await OnDatabaseCreatedAsync();
            OnDatabaseCreated();
        }

        public async Task DisposeAsync()
        {
            BeforeDatabaseDestroy();
            await BeforeDatabaseDestroyAsync();
            await Database.DisposeAsync();
            await TestDatabase.DisposeAsync();
        }


        protected virtual void OnDatabaseCreated() { }
        protected virtual void BeforeDatabaseDestroy() { }


        protected virtual Task OnDatabaseCreatedAsync()
        {
            return Task.CompletedTask;
        }

        protected virtual Task BeforeDatabaseDestroyAsync()
        {
            return Task.CompletedTask;
        }
    }
}