From 7f7fdd4d3d72980d575f7e27cea4a2d11c8482a8 Mon Sep 17 00:00:00 2001
From: 杨宇千 <crupest@outlook.com>
Date: Sun, 4 Aug 2019 15:46:31 +0800
Subject: Continue to add unit tests for token. Fix a bug thanks to unit test.

---
 Timeline.Tests/Helpers/TestClock.cs                | 25 ++++++++++++++++++++++
 .../Helpers/WebApplicationFactoryExtensions.cs     |  6 ++++++
 2 files changed, 31 insertions(+)
 create mode 100644 Timeline.Tests/Helpers/TestClock.cs

(limited to 'Timeline.Tests/Helpers')

diff --git a/Timeline.Tests/Helpers/TestClock.cs b/Timeline.Tests/Helpers/TestClock.cs
new file mode 100644
index 00000000..fc200be9
--- /dev/null
+++ b/Timeline.Tests/Helpers/TestClock.cs
@@ -0,0 +1,25 @@
+using Microsoft.AspNetCore.Mvc.Testing;
+using Microsoft.Extensions.DependencyInjection;
+using System;
+using Timeline.Services;
+
+namespace Timeline.Tests.Helpers
+{
+    public class TestClock : IClock
+    {
+        DateTime? MockCurrentTime { get; set; } = null;
+
+        public DateTime GetCurrentTime()
+        {
+            return MockCurrentTime.GetValueOrDefault(DateTime.Now);
+        }
+    }
+
+    public static class TestClockWebApplicationFactoryExtensions
+    {
+        public static TestClock GetTestClock<T>(this WebApplicationFactory<T> factory) where T : class
+        {
+            return factory.Server.Host.Services.GetRequiredService<IClock>() as TestClock;
+        }
+    }
+}
diff --git a/Timeline.Tests/Helpers/WebApplicationFactoryExtensions.cs b/Timeline.Tests/Helpers/WebApplicationFactoryExtensions.cs
index a7616b41..aa005ba3 100644
--- a/Timeline.Tests/Helpers/WebApplicationFactoryExtensions.cs
+++ b/Timeline.Tests/Helpers/WebApplicationFactoryExtensions.cs
@@ -1,9 +1,11 @@
 using Microsoft.AspNetCore.Hosting;
 using Microsoft.AspNetCore.Mvc.Testing;
+using Microsoft.AspNetCore.TestHost;
 using Microsoft.EntityFrameworkCore;
 using Microsoft.Extensions.DependencyInjection;
 using Microsoft.Extensions.Logging;
 using Timeline.Models;
+using Timeline.Services;
 using Xunit.Abstractions;
 
 namespace Timeline.Tests.Helpers
@@ -46,6 +48,10 @@ namespace Timeline.Tests.Helpers
                             db.Users.AddRange(TestMockUsers.MockUsers);
                             db.SaveChanges();
                         }
+                    })
+                    .ConfigureTestServices(services =>
+                    {
+                        services.AddSingleton<IClock, TestClock>();
                     });
             });
         }
-- 
cgit v1.2.3