fix: fixed retries

README.md

@@ -24,6 +24,8 @@ extendable utilities for testing
   * [Run](#run)
   * [Add](#add)
   * [Stdout/Stderr Mocking](#stdoutstderr-mocking)
+  * [Done](#done)
+  * [Retries](#retries)
   * [Chai](#chai)
 - [Chaining](#chaining)
 - [Custom Plugins](#custom-plugins)
@@ -287,8 +289,6 @@ Done
 You can get the mocha `done()` callback by passing in a second argument.
 
 ```js
-import {expect, fancy} from 'fancy-test'
-
 describe('calls done', () => {
   fancy
   .it('expects FOO=bar', (_, done) => {
@@ -297,6 +297,25 @@ describe('calls done', () => {
 })
 ```
 
+Retries
+-------
+
+Retry the test n times.
+
+```js
+let count = 3
+
+describe('test retries', () => {
+  fancy
+  .retries(2)
+  .do(() => {
+    count--
+    if (count > 0) throw new Error('x')
+  })
+  .it('retries 3 times')
+})
+```
+
 Chai
 ----
 

src/base.ts

@@ -21,6 +21,9 @@ const base = <I extends Types.Context>(context: I): Types.Base<I, {}> => {
     }
     if (!arg1) arg1 = context.expectation || 'test'
     async function run(this: mocha.ITestCallbackContext, done?: Types.MochaDone) {
+      // reset error if retrying
+      delete context.error
+      if (context.retries) this.retries(context.retries)
       if (cb) {
         context.chain = [...context.chain, {
           run: async (input: any) => {
@@ -94,12 +97,12 @@ const base = <I extends Types.Context>(context: I): Types.Base<I, {}> => {
         chain: [...context.chain, {finally: (input: any) => cb(input)}]
       })
     },
-    add(key, cb) {
+    add(key, v) {
       return base({
         ...context as any,
         chain: [...context.chain, {
           run: async (ctx: any) => {
-            ctx[key] = await cb(ctx)
+            ctx[key] = await (_.isFunction(v) ? v(ctx) : v)
           }
         }]
       })
@@ -135,3 +138,6 @@ export default base(context)
 .register('only', () => ({
   init: ctx => {ctx.test = it.only}
 }))
+.register('retries', (count: number) => ({
+  init: ctx => ctx.retries = count
+}))

src/types.ts

@@ -8,6 +8,7 @@ export interface Context {
   expectation?: string
   chain: Plugin<any>[]
   error?: Error & {code?: string}
+  retries?: number
 }
 
 export interface Plugin<I> {
@@ -36,7 +37,7 @@ export interface It<I> {
 export type Base<I extends Context, T extends Plugins> = {
   it: It<I>
   end: It<I>
-  add<K extends string, O>(key: K, cb: (context: I) => Promise<O> | O): Base<I & {[P in K]: O}, T>
+  add<K extends string, O>(key: K, cb: ((context: I) => Promise<O> | O) | Promise<O> | O): Base<I & {[P in K]: O}, T>
   do<O>(cb: (context: I & O) => any): Base<O & I, T>
   finally(cb: (context: I) => any): Base<I, T>
   register<K extends string, O, A1, A2, A3, A4>(key: K, plugin: (arg1?: A1, arg2?: A2, arg3?: A3, arg4?: A4) => Plugin<O & I>): Base<I, T & {[P in K]: {output: O, a1: A1, a2: A2, a3: A3, a4: A4}}>

test/retries.test.ts

@@ -0,0 +1,17 @@
+// tslint:disable no-console
+
+import {fancy} from '../src'
+
+let count = 3
+
+describe('retries', () => {
+  // from readme
+  fancy
+  .retries(2)
+  .do(() => {
+    count--
+    if (count > 0) throw new Error('x')
+  })
+  .it('retries 3 times')
+  // from readme
+})