Reorganize unit tests.

1 files changed, 25 insertions, 0 deletions

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;

+        }

+    }

+}