codeceptjs 3.3.7 → 3.3.8-beta.1

package/lib/scenario.js

@@ -1,3 +1,4 @@
+const promiseRetry = require('promise-retry');
 const event = require('./event');
 const recorder = require('./recorder');
 const assertThrown = require('./assert/throws');
@@ -106,7 +107,7 @@ module.exports.test = (test) => {
 /**
  * Injects arguments to function from controller
  */
-module.exports.injected = function (fn, suite, hookName) {
+module.exports.injected = function (fn, suite, hookName) {  
   return function (done) {
     const errHandler = (err) => {
       recorder.session.start('teardown');
@@ -133,35 +134,26 @@ module.exports.injected = function (fn, suite, hookName) {
 
     this.test.body = fn.toString();
 
-    if (isAsyncFunction(fn)) {
-      fn.call(this, getInjectedArguments(fn)).then(() => {
-        recorder.add('fire hook.passed', () => event.emit(event.hook.passed, suite));
-        recorder.add(`finish ${hookName} hook`, () => done());
-        recorder.catch();
-      }).catch((e) => {
-        recorder.throw(e);
-        recorder.catch((e) => {
-          const err = (recorder.getAsyncErr() === null) ? e : recorder.getAsyncErr();
-          errHandler(err);
-        });
-        recorder.add('fire hook.failed', () => event.emit(event.hook.failed, suite, e));
-      });
-      return;
-    }
-
-    try {
-      fn.call(this, getInjectedArguments(fn));
+    recorder.retry(suite.retries());
+    promiseRetry(async (retry) => {
+      try {
+        await fn.call(this, getInjectedArguments(fn))
+      } catch (err) {
+        retry(err);
+      }
+    }, { retries: suite.retries() })
+    .then(() => {
       recorder.add('fire hook.passed', () => event.emit(event.hook.passed, suite));
       recorder.add(`finish ${hookName} hook`, () => done());
       recorder.catch();
-    } catch (err) {
-      recorder.throw(err);
+    }).catch((e) => {
+      recorder.throw(e);
       recorder.catch((e) => {
         const err = (recorder.getAsyncErr() === null) ? e : recorder.getAsyncErr();
         errHandler(err);
       });
-      recorder.add('fire hook.failed', () => event.emit(event.hook.failed, suite, err));
-    }
+      recorder.add('fire hook.failed', () => event.emit(event.hook.failed, suite, e));
+    });
   };
 };
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "codeceptjs",
-  "version": "3.3.7",
+  "version": "3.3.8-beta.1",
   "description": "Supercharged End 2 End Testing Framework for NodeJS",
   "keywords": [
     "acceptance",