From c9cc1e18fb36df25ad28778c26e4b2bd88b6a96d Mon Sep 17 00:00:00 2001
From: 杨宇千 <crupest@outlook.com>
Date: Sun, 20 Oct 2019 16:24:11 +0800
Subject: ...

---
 Timeline.Tests/Controllers/TokenControllerTest.cs | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

(limited to 'Timeline.Tests/Controllers/TokenControllerTest.cs')

diff --git a/Timeline.Tests/Controllers/TokenControllerTest.cs b/Timeline.Tests/Controllers/TokenControllerTest.cs
index 8b1cf071..86a241e5 100644
--- a/Timeline.Tests/Controllers/TokenControllerTest.cs
+++ b/Timeline.Tests/Controllers/TokenControllerTest.cs
@@ -15,7 +15,7 @@ using static Timeline.ErrorCodes.Http.Token;
 
 namespace Timeline.Tests.Controllers
 {
-    public class TokenControllerTest
+    public class TokenControllerTest : IDisposable
     {
         private readonly Mock<IUserService> _mockUserService = new Mock<IUserService>();
         private readonly TestClock _mockClock = new TestClock();
@@ -24,7 +24,14 @@ namespace Timeline.Tests.Controllers
 
         public TokenControllerTest()
         {
-            _controller = new TokenController(_mockUserService.Object, NullLogger<TokenController>.Instance, _mockClock);
+            _controller = new TokenController(_mockUserService.Object,
+                NullLogger<TokenController>.Instance, _mockClock,
+                new MockStringLocalizer<TokenController>());
+        }
+
+        public void Dispose()
+        {
+            _controller.Dispose();
         }
 
         [Theory]
@@ -110,7 +117,5 @@ namespace Timeline.Tests.Controllers
                 .Which.Value.Should().BeAssignableTo<CommonResponse>()
                 .Which.Code.Should().Be(code);
         }
-
-        // TODO! Verify unit tests
     }
 }
-- 
cgit v1.2.3