From f5ccd0d9855f82e14c6b765eee6a04b22c50dc8a Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Sun, 15 Nov 2020 01:44:48 +0800
Subject: ...

---
 .../HttpClientTimelineExtensions.cs                | 26 ++++++++--------------
 1 file changed, 9 insertions(+), 17 deletions(-)

(limited to 'BackEnd/Timeline.Tests/IntegratedTests/HttpClientTimelineExtensions.cs')

diff --git a/BackEnd/Timeline.Tests/IntegratedTests/HttpClientTimelineExtensions.cs b/BackEnd/Timeline.Tests/IntegratedTests/HttpClientTimelineExtensions.cs
index 125435f9..992889e3 100644
--- a/BackEnd/Timeline.Tests/IntegratedTests/HttpClientTimelineExtensions.cs
+++ b/BackEnd/Timeline.Tests/IntegratedTests/HttpClientTimelineExtensions.cs
@@ -7,24 +7,16 @@ namespace Timeline.Tests.IntegratedTests
 {
     public static class HttpClientTimelineExtensions
     {
-        public static async Task<TimelineInfo> GetTimelineAsync(this HttpClient client, string timelineName)
-        {
-            var res = await client.GetAsync($"timelines/{timelineName}");
-            res.Should().HaveStatusCode(HttpStatusCode.OK);
-            return await res.Should().HaveAndGetJsonBodyAsync<TimelineInfo>();
-        }
+        public static Task<TimelineInfo> GetTimelineAsync(this HttpClient client, string timelineName)
+            => client.TestGetAsync<TimelineInfo>($"timelines/{timelineName}");
 
-        public static async Task<TimelineInfo> PatchTimelineAsync(this HttpClient client, string timelineName, TimelinePatchRequest body)
-        {
-            var res = await client.PatchAsJsonAsync($"timelines/{timelineName}", body);
-            res.Should().HaveStatusCode(HttpStatusCode.OK);
-            return await res.Should().HaveAndGetJsonBodyAsync<TimelineInfo>();
-        }
+        public static Task<TimelineInfo> PatchTimelineAsync(this HttpClient client, string timelineName, TimelinePatchRequest body)
+            => client.TestPatchAsync<TimelineInfo>($"timelines/{timelineName}", body);
 
-        public static async Task PutTimelineMemberAsync(this HttpClient client, string timelineName, string memberUsername)
-        {
-            var res = await client.PutAsync($"timelines/{timelineName}/members/{memberUsername}");
-            res.Should().HaveStatusCode(HttpStatusCode.OK);
-        }
+        public static Task PutTimelineMemberAsync(this HttpClient client, string timelineName, string memberUsername)
+            => client.TestPutAsync($"timelines/{timelineName}/members/{memberUsername}");
+
+        public static Task DeleteTimelineMemberAsync(this HttpClient client, string timelineName, string memberUsername, bool? delete)
+            => client.TestDeleteAsync($"timelines/{timelineName}/members/{memberUsername}", delete);
     }
 }
-- 
cgit v1.2.3