added hook config

Author: DavertMik

lib/mocha/hooks.js

@@ -4,7 +4,7 @@ class Hook {
   constructor(context, error) {
     this.suite = context.suite
     this.test = context.test
-    this.runnable = context.ctx.test
+    this.runnable = context?.ctx?.test
     this.ctx = context.ctx
     this.error = error
   }
@@ -17,6 +17,10 @@ class Hook {
     return this.toString() + '()'
   }
 
+  retry(n) {
+    // must be implemented for each hook
+  }
+
   get title() {
     return this.ctx?.test?.title || this.name
   }
@@ -26,13 +30,29 @@ class Hook {
   }
 }
 
-class BeforeHook extends Hook {}
+class BeforeHook extends Hook {
+  retry(n) {
+    this.suite.opts['retryBefore'] = n
+  }
+}
 
-class AfterHook extends Hook {}
+class AfterHook extends Hook {
+  retry(n) {
+    this.suite.opts['retryAfter'] = n
+  }
+}
 
-class BeforeSuiteHook extends Hook {}
+class BeforeSuiteHook extends Hook {
+  retry(n) {
+    this.suite.opts['retryBeforeSuite'] = n
+  }
+}
 
-class AfterSuiteHook extends Hook {}
+class AfterSuiteHook extends Hook {
+  retry(n) {
+    this.suite.opts['retryAfterSuite'] = n
+  }
+}
 
 function fireHook(eventType, suite, error) {
   const hook = suite.ctx?.test?.title?.match(/"([^"]*)"/)[1]
@@ -54,10 +74,22 @@ function fireHook(eventType, suite, error) {
   }
 }
 
+class HookConfig {
+  constructor(hook) {
+    this.hook = hook
+  }
+
+  retry(n) {
+    this.hook.retry(n)
+    return this
+  }
+}
+
 module.exports = {
   BeforeHook,
   AfterHook,
   BeforeSuiteHook,
   AfterSuiteHook,
   fireHook,
+  HookConfig,
 }

lib/mocha/scenarioConfig.js

@@ -81,10 +81,9 @@ class ScenarioConfig {
     if (typeof obj === 'function') {
       if (isAsyncFunction(obj)) {
         obj(this.test).then(res => (this.test.config[helper] = res))
-      } else {
-        obj = obj(this.test)
+        return this
       }
-      return this
+      obj = obj(this.test)
     }
 
     this.test.config[helper] = obj

lib/mocha/ui.js

@@ -5,6 +5,7 @@ const FeatureConfig = require('./featureConfig')
 const addDataContext = require('../data/context')
 const { createTest } = require('./test')
 const { createSuite } = require('./suite')
+const { HookConfig, AfterSuiteHook, AfterHook, BeforeSuiteHook, BeforeHook } = require('./hooks')
 
 const setContextTranslation = context => {
   const container = require('../container')
@@ -118,18 +119,22 @@ module.exports = function (suite) {
 
     context.BeforeSuite = function (fn) {
       suites[0].beforeAll('BeforeSuite', scenario.injected(fn, suites[0], 'beforeSuite'))
+      return new HookConfig(new BeforeSuiteHook({ suite: suites[0] }))
     }
 
     context.AfterSuite = function (fn) {
       afterAllHooks.unshift(['AfterSuite', scenario.injected(fn, suites[0], 'afterSuite')])
+      return new HookConfig(new AfterSuiteHook({ suite: suites[0] }))
     }
 
     context.Background = context.Before = function (fn) {
       suites[0].beforeEach('Before', scenario.injected(fn, suites[0], 'before'))
+      return new HookConfig(new BeforeHook({ suite: suites[0] }))
     }
 
     context.After = function (fn) {
       afterEachHooks.unshift(['After', scenario.injected(fn, suites[0], 'after')])
+      return new HookConfig(new AfterHook({ suite: suites[0] }))
     }
 
     /**