From 0bda82609a08965f7bfefd20af3ab5112ce98dd2 Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Thu, 17 Dec 2020 20:08:33 +0800
Subject: ...

---
 BackEnd/Timeline.Tests/Services/DatabaseBasedTest.cs | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 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 3bb6ebb5..90fb6463 100644
--- a/BackEnd/Timeline.Tests/Services/DatabaseBasedTest.cs
+++ b/BackEnd/Timeline.Tests/Services/DatabaseBasedTest.cs
@@ -2,6 +2,7 @@
 using Timeline.Entities;
 using Timeline.Tests.Helpers;
 using Xunit;
+using Xunit.Abstractions;
 
 namespace Timeline.Tests.Services
 {
@@ -10,15 +11,20 @@ namespace Timeline.Tests.Services
         protected TestDatabase TestDatabase { get; }
         protected DatabaseContext Database { get; private set; } = default!;
 
-        protected DatabaseBasedTest(bool databaseCreateUsers = true)
+        private readonly ITestOutputHelper? _testOutputHelper;
+
+        protected DatabaseBasedTest(bool databaseCreateUsers = true, ITestOutputHelper? testOutputHelper = null)
         {
+            _testOutputHelper = testOutputHelper;
             TestDatabase = new TestDatabase(databaseCreateUsers);
         }
 
+        protected DatabaseBasedTest(ITestOutputHelper? testOutputHelper) : this(true, testOutputHelper) { }
+
         public async Task InitializeAsync()
         {
             await TestDatabase.InitializeAsync();
-            Database = TestDatabase.CreateContext();
+            Database = TestDatabase.CreateContext(_testOutputHelper);
             await OnDatabaseCreatedAsync();
             OnDatabaseCreated();
         }
-- 
cgit v1.2.3