AssertJ API Test

Author: santhoshshetty

src/test/java/demo/customassert/APIAssert.java

@@ -0,0 +1,34 @@
+package demo.customassert;
+
+import io.restassured.response.Response;
+import org.assertj.core.api.AbstractAssert;
+import org.assertj.core.api.SoftAssertions;
+
+import java.util.function.Predicate;
+
+public class APIAssert extends AbstractAssert<APIAssert, Response> {
+
+    SoftAssertions assertions;
+    private APIAssert(Response response) {
+        super(response, APIAssert.class);
+        assertions=new SoftAssertions();
+    }
+
+    public static APIAssert assertThat(Response response){
+        return new APIAssert(response);
+    }
+
+    public APIAssert statusCodeIs(int statusCode){
+        assertions.assertThat(actual.statusCode()).isEqualTo(statusCode);
+        return this;
+    }
+
+    public APIAssert hasKeyWithValue(Predicate<Response> predicate){
+        assertions.assertThat(predicate).accepts(actual);
+        return this;
+    }
+
+    public void assertAll(){
+        assertions.assertAll();
+    }
+}

src/test/java/demo/customassert/APITest.java

@@ -0,0 +1,22 @@
+package demo.customassert;
+
+import io.restassured.RestAssured;
+import io.restassured.response.Response;
+import org.testng.annotations.Test;
+
+import java.util.function.Predicate;
+
+import static demo.customassert.APIAssert.assertThat;
+
+public class APITest {
+
+    @Test
+    public void apiAssertTest(){
+        Response response=RestAssured.given().get("https://reqres.in/api/users/2");
+        Predicate<Response> keyValue=res -> res.jsonPath().getInt("data.id")==2;
+        assertThat(response)
+                .statusCodeIs(200)
+                .hasKeyWithValue(keyValue)
+                .assertAll();
+    }
+}