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

---
 .../Helpers/Authentication/AuthenticationExtensions.cs      | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

(limited to 'Timeline.Tests/Helpers/Authentication/AuthenticationExtensions.cs')

diff --git a/Timeline.Tests/Helpers/Authentication/AuthenticationExtensions.cs b/Timeline.Tests/Helpers/Authentication/AuthenticationExtensions.cs
index de88fd35..e31bd51c 100644
--- a/Timeline.Tests/Helpers/Authentication/AuthenticationExtensions.cs
+++ b/Timeline.Tests/Helpers/Authentication/AuthenticationExtensions.cs
@@ -3,6 +3,7 @@ using Newtonsoft.Json;
 using System.Net.Http;
 using System.Threading.Tasks;
 using Timeline.Models.Http;
+using Timeline.Tests.Mock.Data;
 
 namespace Timeline.Tests.Helpers.Authentication
 {
@@ -17,12 +18,22 @@ namespace Timeline.Tests.Helpers.Authentication
             return result;
         }
 
-        public static async Task<HttpClient> CreateClientWithUser<T>(this WebApplicationFactory<T> factory, string username, string password) where T : class
+        public static async Task<HttpClient> CreateClientWithCredential<T>(this WebApplicationFactory<T> factory, string username, string password) where T : class
         {
             var client = factory.CreateDefaultClient();
             var token = (await client.CreateUserTokenAsync(username, password)).Token;
             client.DefaultRequestHeaders.Add("Authorization", "Bearer " + token);
             return client;
         }
+
+        public static Task<HttpClient> CreateClientAsUser<T>(this WebApplicationFactory<T> factory) where T : class
+        {
+            return factory.CreateClientWithCredential(MockUsers.UserUsername, MockUsers.UserPassword);
+        }
+
+        public static Task<HttpClient> CreateClientAsAdmin<T>(this WebApplicationFactory<T> factory) where T : class
+        {
+            return factory.CreateClientWithCredential(MockUsers.AdminUsername, MockUsers.AdminPassword);
+        }
     }
 }
-- 
cgit v1.2.3