1 files changed, 76 insertions, 0 deletions
diff --git a/BackEnd/Timeline.Tests/Helpers/TestDatabase.cs b/BackEnd/Timeline.Tests/Helpers/TestDatabase.cs
new file mode 100644
index 00000000..f0c26180
--- /dev/null
+++ b/BackEnd/Timeline.Tests/Helpers/TestDatabase.cs
@@ -0,0 +1,76 @@
+using Microsoft.Data.Sqlite;
+using Microsoft.EntityFrameworkCore;
+using Microsoft.Extensions.Logging.Abstractions;
+using System.Threading.Tasks;
+using Timeline.Entities;
+using Timeline.Migrations;
+using Timeline.Models;
+using Timeline.Services;
+using Xunit;
+
+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();
+
+                if (_createUser)
+                {
+                    var passwordService = new PasswordService();
+                    var userService = new UserService(NullLogger<UserService>.Instance, context, passwordService, new Clock());
+
+                    await userService.CreateUser(new User
+                    {
+                        Username = "admin",
+                        Password = "adminpw",
+                        Administrator = true,
+                        Nickname = "administrator"
+                    });
+
+                    await userService.CreateUser(new User
+                    {
+                        Username = "user",
+                        Password = "userpw",
+                        Administrator = false,
+                        Nickname = "imuser"
+                    });
+                }
+            }
+        }
+
+        public async Task DisposeAsync()
+        {
+            await Connection.CloseAsync();
+            await Connection.DisposeAsync();
+        }
+
+        public SqliteConnection Connection { get; }
+
+        public DatabaseContext CreateContext()
+        {
+            var options = new DbContextOptionsBuilder<DatabaseContext>()
+                .UseSqlite(Connection).Options;
+
+            return new DatabaseContext(options);
+        }
+    }
+}