From 58986da4a5bfe519af44e5834edfbe8d4651b51c Mon Sep 17 00:00:00 2001
From: 杨宇千 <crupest@outlook.com>
Date: Fri, 9 Aug 2019 15:39:58 +0800
Subject: Add UserController unit tests.

---
 Timeline.Tests/Helpers/ResponseExtensions.cs | 47 ++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)

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

diff --git a/Timeline.Tests/Helpers/ResponseExtensions.cs b/Timeline.Tests/Helpers/ResponseExtensions.cs
index 155836fb..46c9e81d 100644
--- a/Timeline.Tests/Helpers/ResponseExtensions.cs
+++ b/Timeline.Tests/Helpers/ResponseExtensions.cs
@@ -1,11 +1,58 @@
 using Newtonsoft.Json;
+using System.Net;
 using System.Net.Http;
 using System.Threading.Tasks;
+using Timeline.Models.Http;
+using Xunit;
 
 namespace Timeline.Tests.Helpers
 {
     public static class ResponseExtensions
     {
+        public static void AssertOk(this HttpResponseMessage response)
+        {
+            Assert.Equal(HttpStatusCode.OK, response.StatusCode);
+        }
+
+        public static void AssertNotFound(this HttpResponseMessage response)
+        {
+            Assert.Equal(HttpStatusCode.NotFound, response.StatusCode);
+        }
+
+        public static void AssertBadRequest(this HttpResponseMessage response)
+        {
+            Assert.Equal(HttpStatusCode.BadRequest, response.StatusCode);
+        }
+
+        public static async Task AssertIsPutCreated(this HttpResponseMessage response)
+        {
+            Assert.Equal(HttpStatusCode.Created, response.StatusCode);
+            var body = await response.ReadBodyAsJson<CommonResponse>();
+            Assert.Equal(CommonPutResponse.CreatedCode, body.Code);
+        }
+
+        public static async Task AssertIsPutModified(this HttpResponseMessage response)
+        {
+            response.AssertOk();
+            var body = await response.ReadBodyAsJson<CommonResponse>();
+            Assert.Equal(CommonPutResponse.ModifiedCode, body.Code);
+        }
+
+
+        public static async Task AssertIsDeleteDeleted(this HttpResponseMessage response)
+        {
+            response.AssertOk();
+            var body = await response.ReadBodyAsJson<CommonResponse>();
+            Assert.Equal(CommonDeleteResponse.DeletedCode, body.Code);
+        }
+
+        public static async Task AssertIsDeleteNotExist(this HttpResponseMessage response)
+        {
+            response.AssertOk();
+            var body = await response.ReadBodyAsJson<CommonResponse>();
+            Assert.Equal(CommonDeleteResponse.NotExistsCode, body.Code);
+        }
+
         public static async Task<T> ReadBodyAsJson<T>(this HttpResponseMessage response)
         {
             return JsonConvert.DeserializeObject<T>(await response.Content.ReadAsStringAsync());
-- 
cgit v1.2.3