Custom screenshot command doesn't wait for image to be taken

closes reportportal#76
pashidlos · Mar 22, 2021 · 8f5fa40 · 8f5fa40
1 parent 36f66c5
commit 8f5fa40
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/commands/reportPortalCommands.js b/lib/commands/reportPortalCommands.js
@@ -163,11 +163,11 @@ Cypress.Commands.add('setTestCaseId', (testCaseId, suiteTitle) => {
  */
 Cypress.Commands.overwrite('screenshot', (originalFn, subject, name, option) => {
   if (name) {
-    cy.task('rp_screenshot', {
+    return cy.task('rp_screenshot', {
       fileName: name,
-    });
+    }).then(() => originalFn(subject, name, option));
   }
-  originalFn(subject, name, option);
+  return originalFn(subject, name, option);
 });
 
 /**