Extract addParameter to helpers

Author: aslakhellesoy

src/support_code_library/builder.js

@@ -2,7 +2,6 @@ import util from 'util'
 import _ from 'lodash'
 import arity from 'util-arity'
 import isGenerator from 'is-generator'
-import {Parameter} from 'cucumber-expressions'
 import path from 'path'
 import TransformLookupBuilder from './transform_lookup_builder'
 import * as helpers from './helpers'
@@ -20,19 +19,9 @@ function build({cwd, fns}) {
     }
   }
   let definitionFunctionWrapper = null
-  function addParameter({captureGroupRegexps, transformer, typeName}) {
-    const parameter = new Parameter(
-      typeName,
-      function() {},
-      captureGroupRegexps,
-      transformer
-    )
-    options.parameterRegistry.addParameter(parameter)
-  }
-  const addTransform = util.deprecate(addParameter, 'addTransform is deprecated and will be removed in a future version. Please use addParameter instead.')
   const fnArgument = {
-    addParameter,
-    addTransform,
+    addParameter: helpers.addParameter(options.parameterRegistry),
+    addTransform: util.deprecate(helpers.addParameter(options.parameterRegistry), 'addTransform is deprecated and will be removed in a future version. Please use addParameter instead.'),
     After: helpers.defineHook(cwd, options.afterHookDefinitions),
     Before: helpers.defineHook(cwd, options.beforeHookDefinitions),
     defineStep: helpers.defineStep(cwd, options.stepDefinitions),

src/support_code_library/helpers.js

@@ -1,4 +1,5 @@
 import _ from 'lodash'
+import {Parameter} from 'cucumber-expressions'
 import {formatLocation} from '../formatter/utils'
 import HookDefinition from '../models/hook_definition'
 import path from 'path'
@@ -70,3 +71,15 @@ export function registerHandler(cwd, collection) {
     collection.push(listener)
   }
 }
+
+export function addParameter(parameterRegistry) {
+  return ({captureGroupRegexps, transformer, typeName}) => {
+    const parameter = new Parameter(
+      typeName,
+      function() {},
+      captureGroupRegexps,
+      transformer
+    )
+    parameterRegistry.addParameter(parameter)
+  }
+}