Merge pull request #1 from JWaldecker/testAddAssertionFunctions

added a few assertion methods to TestExchange

Author: JanHolger

src/main/java/org/javawebstack/httpserver/test/TestExchange.java

@@ -6,6 +6,9 @@
 import org.javawebstack.httpserver.helper.MimeType;
 import org.junit.jupiter.api.Assertions;
 
+import java.util.ArrayList;
+import java.util.List;
+
 public class TestExchange extends Exchange {
     private MockHttpServletRequest mockReq;
     private MockHttpServletResponse mockRes;
@@ -74,8 +77,40 @@ public TestExchange assertCreated(String message) {
         Assertions.assertEquals(201, mockRes.getStatus(), message);
         return this;
     }
-
-    public TestExchange assertNotFound() {
+    
+    public TestExchange assertNoContent() {
+        Assertions.assertEquals(204, mockRes.getStatus());
+        return this;
+    }
+    
+    public TestExchange assertNoContent(String message) {
+        Assertions.assertEquals(204, mockRes.getStatus(), message);
+        return this;
+    }
+    
+    public TestExchange assertRedirect() {
+        Assertions.assertTrue(isRedirect());
+        return this;
+    }
+    
+    public TestExchange assertRedirect(String message) {
+        Assertions.assertTrue(isRedirect(), message);
+        return this;
+    }
+    
+    public TestExchange assertRedirectTo(String url) {
+        assertRedirect();
+        Assertions.assertEquals(url, mockRes.getHeader("Location"));
+        return this;
+    }
+    
+    public TestExchange assertRedirectTo(String url, String message) {
+        assertRedirect(message);
+        Assertions.assertEquals(url, mockRes.getHeader("Location"), message);
+        return this;
+    }
+    
+    public TestExchange assertNotFound(){
         Assertions.assertEquals(404, mockRes.getStatus());
         return this;
     }
@@ -236,4 +271,14 @@ private boolean checkGraph(AbstractElement element, Object value) {
         }
         return false;
     }
+    private boolean isRedirect() {
+        List<Integer> redirectCodes = new ArrayList<>();
+        redirectCodes.add(301);
+        redirectCodes.add(302);
+        redirectCodes.add(303);
+        redirectCodes.add(307);
+        redirectCodes.add(308);
+
+        return redirectCodes.contains(mockRes.getStatus());
+    }
 }