Fix the database bug in unit tests.

1 files changed, 3 insertions, 5 deletions

diff --git a/Timeline.Tests/TokenUnitTest.cs b/Timeline.Tests/TokenUnitTest.cs
index 1fe3cff6..7b83cd13 100644
--- a/Timeline.Tests/TokenUnitTest.cs
+++ b/Timeline.Tests/TokenUnitTest.cs
@@ -6,9 +6,7 @@ using System.Linq;
 using System.Net;

 using System.Net.Http;

 using Timeline.Controllers;

-using Timeline.Entities;

 using Timeline.Entities.Http;

-using Timeline.Models;

 using Timeline.Services;

 using Timeline.Tests.Helpers;

 using Timeline.Tests.Helpers.Authentication;

@@ -17,16 +15,16 @@ using Xunit.Abstractions;
 

 namespace Timeline.Tests

 {

-    public class TokenUnitTest : IClassFixture<WebApplicationFactory<Startup>>

+    public class TokenUnitTest : IClassFixture<MyWebApplicationFactory<Startup>>

     {

         private const string CreateTokenUrl = "token/create";

         private const string VerifyTokenUrl = "token/verify";

 

         private readonly WebApplicationFactory<Startup> _factory;

 

-        public TokenUnitTest(WebApplicationFactory<Startup> factory, ITestOutputHelper outputHelper)

+        public TokenUnitTest(MyWebApplicationFactory<Startup> factory, ITestOutputHelper outputHelper)

         {

-            _factory = factory.WithTestConfig(outputHelper);

+            _factory = factory.WithTestLogging(outputHelper);

         }

 

         [Fact]