From 0af4e8c9e788a3bbf4d6879a7f42660cb47ddedb Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Thu, 12 Nov 2020 18:31:41 +0800
Subject: feat: Add user permission service.

TODO: Add unit tests.
---
 .../Timeline.Tests/Services/DatabaseBasedTest.cs   | 47 ++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 BackEnd/Timeline.Tests/Services/DatabaseBasedTest.cs

(limited to 'BackEnd/Timeline.Tests/Services/DatabaseBasedTest.cs')

diff --git a/BackEnd/Timeline.Tests/Services/DatabaseBasedTest.cs b/BackEnd/Timeline.Tests/Services/DatabaseBasedTest.cs
new file mode 100644
index 00000000..838787e9
--- /dev/null
+++ b/BackEnd/Timeline.Tests/Services/DatabaseBasedTest.cs
@@ -0,0 +1,47 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+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; } = new TestDatabase();
+        protected DatabaseContext Database { get; private set; }
+
+        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;
+        }
+    }
+}
-- 
cgit v1.2.3