public List<WebElement> getUpdatedMsgState(MessageType messageType) throws Exception { WebElement messagesTable = service.findElement(tableLocator); if (messageType.equals(MessageType.AGENT)) { return messagesTable.findElements(agentLineLocator); } else if (messageType.equals(MessageType.VISITOR)) { return messagesTable.findElements(visitorLineLocator); } return null; }
private void initMessage() { apply.setText(constants.Apply()); if (messageType.equals(MessageType.ERROR)) { imageMessage.setResource(images.error()); } else if (messageType.equals(MessageType.INFO)) { imageMessage.setResource(images.information()); } else if (messageType.equals(MessageType.WARNING)) { imageMessage.setResource(images.warning()); } else { imageMessage.setResource(images.information()); } }