...

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();

         }