From a46fb537f25fa773f8a0cb16e7310eee4b36c3d3 Mon Sep 17 00:00:00 2001
From: Sean Flanigan <sflaniga@redhat.com>
Date: Wed, 18 Feb 2015 15:59:06 +1000
Subject: [PATCH] Wait for at least one notification message

---
 functional-test/src/main/java/org/zanata/page/CorePage.java | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/functional-test/src/main/java/org/zanata/page/CorePage.java b/functional-test/src/main/java/org/zanata/page/CorePage.java
index 8fdb5373f0..a621b1f2f0 100644
--- a/functional-test/src/main/java/org/zanata/page/CorePage.java
+++ b/functional-test/src/main/java/org/zanata/page/CorePage.java
@@ -140,9 +140,9 @@ public List<String> expectErrors() {
 
     public String getNotificationMessage() {
         log.info("Query notification message");
-        List<WebElement>messages = (List<WebElement>) getExecutor()
-                .executeScript("return $('#messages li')");
-        return messages.size() > 0 ? messages.get(0).getText() : "";
+        WebElement message = waitForElementExists(
+                By.cssSelector("#messages li"));
+        return message.getText();
     }
 
     public boolean expectNotification(final String notification) {