support mocks generated by function

Author: johncantrell97

.gitignore

@@ -0,0 +1 @@
+node_modules

changelog.md

@@ -2,6 +2,15 @@
 
 ---
 
+## [0.1.2] 2022-03-28
+
+### Added
+
+- Added support for function generated mocks
+- Added support for accepting cli input for function generated mocks 
+
+
+
 ## [0.1.0] 2022-03-27
 
 ### Added

readme.md

@@ -64,6 +64,9 @@ module.exports = {
     'background-task': {
       'my-first-mock': { /* payload */ },
       'another-mock': { /* payload */ },
+      'function-mock': (id) => {
+        return { /* payload */ }
+      }
     }
   },
   queues: {

src/index.js

@@ -14,6 +14,7 @@ module.exports = {
       let { cwd, preferences } = inv._project
       let jsonMocks = join(cwd, 'sandbox-invoke-mocks.json')
       let jsMocks = join(cwd, 'sandbox-invoke-mocks.js')
+      let inputPromptOpen = false
 
       let pragmas = [ 'events', 'queues', 'scheduled', 'tables-streams' ]
       let prefs = preferences?.sandbox?.invoker
@@ -45,7 +46,7 @@ module.exports = {
             strong: colors.white,
           }
         }
-        if (input === 'i') {
+        if (input === 'i' && !inputPromptOpen) {
           if (Object.keys(events).length === 1) {
             let none = 'No Lambdas found to invoke'
             if (pragmas.length) update.status(none, `Using the following pragmas: @${pragmas.join(', @')}`)
@@ -91,6 +92,21 @@ module.exports = {
             }, options)
             mockName = selection.mock
             userPayload = mocks[pragma][name][mockName] || {}
+
+            if (typeof userPayload === 'function') {
+              inputPromptOpen = true
+              let { mockinput } = await prompt(
+                {
+                  type: 'input',
+                  name: 'mockinput',
+                  message: 'What input to use for mock function?',
+                },
+                options
+              )
+              inputPromptOpen = false
+              let mockInputArgs = mockinput.split(',')
+              userPayload = userPayload(...mockInputArgs)
+            }
           }
 
           let payload