From 2c6b812382e04956793d90ba4148dd4aa7da3b70 Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Thu, 12 Nov 2020 20:03:31 +0800
Subject: test: Write tests for user permission service.

---
 BackEnd/Timeline.Tests/Services/DatabaseBasedTest.cs | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

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

diff --git a/BackEnd/Timeline.Tests/Services/DatabaseBasedTest.cs b/BackEnd/Timeline.Tests/Services/DatabaseBasedTest.cs
index 838787e9..7c97158c 100644
--- a/BackEnd/Timeline.Tests/Services/DatabaseBasedTest.cs
+++ b/BackEnd/Timeline.Tests/Services/DatabaseBasedTest.cs
@@ -1,7 +1,4 @@
-using System;
-using System.Collections.Generic;
-using System.Linq;
-using System.Threading.Tasks;
+using System.Threading.Tasks;
 using Timeline.Entities;
 using Timeline.Tests.Helpers;
 using Xunit;
@@ -10,9 +7,14 @@ namespace Timeline.Tests.Services
 {
     public abstract class DatabaseBasedTest : IAsyncLifetime
     {
-        protected TestDatabase TestDatabase { get; } = new TestDatabase();
+        protected TestDatabase TestDatabase { get; }
         protected DatabaseContext Database { get; private set; }
 
+        protected DatabaseBasedTest(bool databaseCreateUsers = true)
+        {
+            TestDatabase = new TestDatabase(databaseCreateUsers);
+        }
+
         public async Task InitializeAsync()
         {
             await TestDatabase.InitializeAsync();
-- 
cgit v1.2.3