From 3e471ac783d91fcc61a90b759fecefe3b80014ba Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Thu, 18 Jun 2020 16:21:39 +0800
Subject: Add last modified info to timeline.

---
 Timeline.Tests/Helpers/TestDatabase.cs | 76 ++++++++++++++++++++++++++++++++++
 1 file changed, 76 insertions(+)
 create mode 100644 Timeline.Tests/Helpers/TestDatabase.cs

(limited to 'Timeline.Tests/Helpers/TestDatabase.cs')

diff --git a/Timeline.Tests/Helpers/TestDatabase.cs b/Timeline.Tests/Helpers/TestDatabase.cs
new file mode 100644
index 00000000..40f8f2d4
--- /dev/null
+++ b/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);
+
+                    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);
+        }
+    }
+}
-- 
cgit v1.2.3