Message.java

public class Message extends HtmlWrapper {
 
    public enum Severity {
        INFO("info"),
        WARNING("warn"),
        ERROR("error");
 
        Severity(String severity) {
            this.severity = severity;
        }
 
        private final String severity;
 
        public String getSeverity() {
            return severity;
        }
    }
 
    protected Message(WebElement element) {
        super(element);
    }
 
    public static Message create(WebElement element) {
        assertNotNull(element);
        return new Message(element);
    }
 
    public boolean isAnyMessageExist(Severity severity) {
        List<WebElement> messages = findElements(
      By.cssSelector(".ui-messages .ui-messages-" + severity.getSeverity()));
        return messages.size() > 0;
    }
 
    public boolean isAnyMessageExist() {
        for (Severity severity: Severity.values()) {
            List<WebElement> messages = findElements(
       By.cssSelector(".ui-messages .ui-messages-" + severity.getSeverity()));
            if (messages.size() > 0) {
                return true;
            }
        }
 
        return false;
    }
 
    public List<String> getMessages(Severity severity) {
        List<WebElement> messages = findElements(
      By.cssSelector(".ui-messages .ui-messages-" + severity.getSeverity() + "-summary"));
        if (messages.isEmpty()) {
            return null;
        }
 
        List<String> text = new ArrayList<> ();
        for (WebElement element: messages) {
            text.add(element.getText());
        }
 
        return text;
    }
}