chore: query selector on target element (#165)

Fix #164

---------

Co-authored-by: Baruch Odem (Rothkoff) <[email protected]>

src/utils/utils.js

@@ -1,26 +1,28 @@
 /**
  * Waits for element to be rendered in DOM.
  * @param {string} selector A valid css selector string
- * @param {Node} target A DOM Node (which may be an Element) within the DOM tree to watch for changes, or to be the root of a subtree of nodes to be watched.
+ * @param {Node} [target=document] A DOM Node (which may be an Element) within the DOM tree to watch for changes, or to be the root of a subtree of nodes to be watched.
  *
  */
 import { SECOND } from '../globals';
 
 const waitElementTimeOot = 10 * SECOND;
 
-const waitForElement = (selector, target) => {
+const waitForElement = (selector, target = document) => {
   return new Promise((resolve, reject) => {
     setTimeout(() => {
       reject(new Error('Time out to wait for the element ' + selector));
     }, waitElementTimeOot);
 
-    if (document.querySelector(selector)) {
-      return resolve(document.querySelector(selector));
+    const wishedElement = target.querySelector(selector);
+    if (wishedElement) {
+      return resolve(wishedElement);
     }
 
     const observer = new MutationObserver(() => {
-      if (document.querySelector(selector)) {
-        resolve(document.querySelector(selector));
+      const wishedElement = target.querySelector(selector);
+      if (wishedElement) {
+        resolve(wishedElement);
         observer.disconnect();
       }
     });