1 files changed, 8 insertions, 2 deletions
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();
         }