Add waitForEvent() method (#972)

* Add waitForEvent This commit adds a `waitForEvent()` method to the `WaitsForElements` trait, making it available to the `Browser` class. The `waitForEvent()` method pauses the test execution until a given event `type` occurs on a `target` element or object. It also adds a corresponding test case to `WaitsForElementsTest`. * Fix syntax in exception message * formatting Co-authored-by: Taylor Otwell <[email protected]>
laravel · Apr 11, 2022 · 98901d4 · 98901d4
1 parent ea9a455
commit 98901d4
Show file tree

Hide file tree

Showing 2 changed files with 52 additions and 0 deletions.
diff --git a/src/Concerns/WaitsForElements.php b/src/Concerns/WaitsForElements.php
@@ -6,6 +6,7 @@
 use Closure;
 use Exception;
 use Facebook\WebDriver\Exception\NoSuchElementException;
+use Facebook\WebDriver\Exception\ScriptTimeoutException;
 use Facebook\WebDriver\Exception\TimeOutException;
 use Facebook\WebDriver\WebDriverExpectedCondition;
 use Illuminate\Support\Arr;
@@ -343,6 +344,38 @@ public function clickAndWaitForReload($selector = null)
         });
     }
 
+    /**
+     * Wait for the given event type to occur on a target.
+     *
+     * @param  string  $type
+     * @param  string|null  $target
+     * @param  int|null  $seconds
+     * @return $this
+     *
+     * @throws \Facebook\WebDriver\Exception\TimeOutException
+     */
+    public function waitForEvent($type, $target = null, $seconds = null)
+    {
+        $seconds = is_null($seconds) ? static::$waitSeconds : $seconds;
+
+        if ($target !== 'document' && $target !== 'window') {
+            $target = $this->resolver->findOrFail($target ?? '');
+        }
+
+        $this->driver->manage()->timeouts()->setScriptTimeout($seconds);
+
+        try {
+            $this->driver->executeAsyncScript(
+                'eval(arguments[0]).addEventListener(arguments[1], () => arguments[2](), { once: true });',
+                [$target, $type]
+            );
+        } catch (ScriptTimeoutException $e) {
+            throw new TimeoutException("Waited {$seconds} seconds for event [{$type}].");
+        }
+
+        return $this;
+    }
+
     /**
      * Wait for the given callback to be true.
      *

diff --git a/tests/WaitsForElementsTest.php b/tests/WaitsForElementsTest.php
@@ -324,4 +324,23 @@ public function test_wait_for_reload()
 
         $browser->waitForReload();
     }
+
+    public function test_wait_for_event()
+    {
+        $driver = m::mock(stdClass::class);
+        $driver->shouldReceive('manage->timeouts->setScriptTimeout')->with(3);
+        $driver->shouldReceive('executeAsyncScript')->with(
+            'eval(arguments[0]).addEventListener(arguments[1], () => arguments[2](), { once: true });',
+            ['body form', 'submit']
+        );
+
+        $resolver = m::mock(stdClass::class);
+        $resolver->shouldReceive('findOrFail')
+            ->with('form')
+            ->andReturn('body form');
+
+        $browser = new Browser($driver, $resolver);
+
+        $browser->waitForEvent('submit', 'form', 3);
+    }
 }