1 files changed, 4 insertions, 8 deletions
diff --git a/Timeline.Tests/Helpers/InvalidModelTestHelpers.cs b/Timeline.Tests/Helpers/InvalidModelTestHelpers.cs
index 1c079d0e..af432095 100644
--- a/Timeline.Tests/Helpers/InvalidModelTestHelpers.cs
+++ b/Timeline.Tests/Helpers/InvalidModelTestHelpers.cs
@@ -1,8 +1,6 @@
-using System.Net;
 using System.Net.Http;
 using System.Threading.Tasks;
 using Timeline.Models.Http;
-using Xunit;
 
 namespace Timeline.Tests.Helpers
 {
@@ -11,17 +9,15 @@ namespace Timeline.Tests.Helpers
         public static async Task TestPostInvalidModel<T>(HttpClient client, string url, T body)
         {
             var response = await client.PostAsJsonAsync(url, body);
-            Assert.Equal(HttpStatusCode.BadRequest, response.StatusCode);
-            var responseBody = await response.ReadBodyAsJson<CommonResponse>();
-            Assert.Equal(CommonResponse.ErrorCodes.InvalidModel, responseBody.Code);
+            response.Should().HaveStatusCodeBadRequest()
+                .And.Should().HaveBodyAsCommonResponseWithCode(CommonResponse.ErrorCodes.InvalidModel);
         }
 
         public static async Task TestPutInvalidModel<T>(HttpClient client, string url, T body)
         {
             var response = await client.PutAsJsonAsync(url, body);
-            Assert.Equal(HttpStatusCode.BadRequest, response.StatusCode);
-            var responseBody = await response.ReadBodyAsJson<CommonResponse>();
-            Assert.Equal(CommonResponse.ErrorCodes.InvalidModel, responseBody.Code);
+            response.Should().HaveStatusCodeBadRequest()
+                .And.Should().HaveBodyAsCommonResponseWithCode(CommonResponse.ErrorCodes.InvalidModel);
         }
     }
 }