From 6c50630c4d6576446fc2338714feaefbf28c4ed1 Mon Sep 17 00:00:00 2001
From: 杨宇千 <crupest@outlook.com>
Date: Thu, 8 Aug 2019 17:41:59 +0800
Subject: Reorganize unit tests.

---
 Timeline.Tests/Mock/Services/TestClock.cs | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)
 create mode 100644 Timeline.Tests/Mock/Services/TestClock.cs

(limited to 'Timeline.Tests/Mock/Services/TestClock.cs')

diff --git a/Timeline.Tests/Mock/Services/TestClock.cs b/Timeline.Tests/Mock/Services/TestClock.cs
new file mode 100644
index 00000000..0082171e
--- /dev/null
+++ b/Timeline.Tests/Mock/Services/TestClock.cs
@@ -0,0 +1,25 @@
+using Microsoft.AspNetCore.Mvc.Testing;
+using Microsoft.Extensions.DependencyInjection;
+using System;
+using Timeline.Services;
+
+namespace Timeline.Tests.Mock.Services
+{
+    public class TestClock : IClock
+    {
+        public 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