From 36da221444807742a9e26f2ba636d9c6aef6155f 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 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)
 create mode 100644 Timeline.Tests/Helpers/TestClock.cs

(limited to 'Timeline.Tests/Helpers/TestClock.cs')

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;
+        }
+    }
+}
-- 
cgit v1.2.3