From b1eebd980490fcefcec14b51425f5d708221aa04 Mon Sep 17 00:00:00 2001
From: witzig <git.witzig@icloud.com>
Date: Sat, 27 May 2017 14:29:16 +0200
Subject: [PATCH] Fixed waitUntilVisible should wait for selector, if present.
 This is used by waitForFlash.

---
 test/e2e/lib/page.js | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/test/e2e/lib/page.js b/test/e2e/lib/page.js
index 3026f671..d455d895 100644
--- a/test/e2e/lib/page.js
+++ b/test/e2e/lib/page.js
@@ -35,6 +35,10 @@ module.exports = (...extras) => Object.assign({
     async waitUntilVisible(selector) {
         await driver.wait(until.elementLocated(By.css('body')), waitTimeout);
 
+        if (selector) {
+            await driver.wait(until.elementLocated(By.css(selector)), waitTimeout);
+        }
+        
         for (const elem of (this.elementsToWaitFor || [])) {
             const sel = this.elements[elem];
             if (!sel) {