1 files changed, 9 insertions, 17 deletions
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);
     }
 }