Retry only on specific exception

[arturmkr]

Is it possible to configure a retry mechanism in tests to run only on specific exceptions?

For example, in the following code, I want the test to retry only if the UnexpectedStatusCodeException is thrown. If any other assertions (like assertEquals) fail, the test should not retry.

code:

public class MyDemoTest {
    @Test
    void sendDataTest() throws UnexpectedStatusCodeException {
        RestAssured.baseURI = "https://petstore.swagger.io/";

        Response response = given()
                .when()
                .get("v2/pet/" + 1);

        checkStatusCode(response, 200);

        String name = response.jsonPath().getString("name");
        assertEquals("dog", name);
    }
    
    static void checkStatusCode(Response response, int expectedStatusCode) throws UnexpectedStatusCodeException {
        int statusCode = response.getStatusCode();
        if (statusCode != expectedStatusCode) {
            throw new UnexpectedStatusCodeException("Unexpected status code: " + statusCode);
        }
    }

    static class UnexpectedStatusCodeException extends Exception {
        public UnexpectedStatusCodeException(String message) {
            super(message);
        }
    }
}