fix: takeElementScreenshot

macacajs · Aug 18, 2023 · ba53c2a · ba53c2a
1 parent 4a307f0
commit ba53c2a
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/lib/controllers.js b/lib/controllers.js
@@ -196,7 +196,14 @@ controllers.click = async function(elementId) {
  */
 controllers.takeElementScreenshot = async function(elementId, params = {}) {
   const { file } = params;
-  const image = await this.elements[elementId].screenshot();
+  let image;
+  if (elementId) {
+    image = await this.elements[elementId].screenshot();
+  } else if (this.locator) {
+    image = await this.locator.screenshot();
+  } else {
+    throw new errors.NoSuchElement();
+  }
   const base64 = image.toString('base64');
   if (file) {
     const img = new Buffer(base64, 'base64');
@@ -211,7 +218,7 @@ controllers.takeElementScreenshot = async function(elementId, params = {}) {
  * Search for an element on the page, starting from the document root.
  * @module findElement
  * @param {string} strategy The type
- * @param {string} using The locator strategy to use.
+ * @param selector Selector string
  * @param {string} ctx The search target.
  * @return {Promise.<Element>}
  */