fix: handle no mock args (#6659)

aws-amplify · Feb 17, 2021 · 6b97ffa · 6b97ffa
1 parent f82b069
commit 6b97ffa
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/packages/amplify-util-mock/src/__tests__/func/index.test.ts b/packages/amplify-util-mock/src/__tests__/func/index.test.ts
@@ -136,4 +136,16 @@ describe('function start', () => {
     expect(inquirer_mock.prompt.mock.calls[0][0][0].choices).toStrictEqual(['func1', 'func2', 'func3']);
     expect(getBuilder_mock.mock.calls[0][1]).toBe('func2');
   });
+
+  it('handles no options specified', async () => {
+    const invoker = jest.fn().mockResolvedValue(null);
+    getInvoker_mock.mockResolvedValueOnce(invoker);
+    inquirer_mock.prompt.mockResolvedValueOnce({ eventName: 'event.json' });
+
+    const context_stub_copy = _.merge({}, context_stub);
+    context_stub_copy.input.options = undefined;
+    await start(context_stub_copy);
+    expect(invoker.mock.calls.length).toBe(1);
+    expect(context_stub_copy.print.error.mock.calls.length).toBe(0);
+  });
 });
diff --git a/packages/amplify-util-mock/src/func/index.ts b/packages/amplify-util-mock/src/func/index.ts
@@ -61,7 +61,7 @@ export async function start(context: $TSContext) {
 interface InvokerOptions {
   timeout?: string;
 }
-export const timeConstrainedInvoker = async <T>(promise: Promise<T>, options: InvokerOptions): Promise<T> => {
+export const timeConstrainedInvoker = async <T>(promise: Promise<T>, options?: InvokerOptions): Promise<T> => {
   const { timer, cancel } = getCancellableTimer(options);
   try {
     return await Promise.race([promise, timer]);
@@ -70,7 +70,7 @@ export const timeConstrainedInvoker = async <T>(promise: Promise<T>, options: In
   }
 };
 
-const getCancellableTimer = ({ timeout }: InvokerOptions) => {
+const getCancellableTimer = ({ timeout }: InvokerOptions = {}) => {
   const inputTimeout = Number.parseInt(timeout, 10);
   const lambdaTimeoutSeconds = !!inputTimeout && inputTimeout > 0 ? inputTimeout : DEFAULT_TIMEOUT_SECONDS;
   const timeoutErrorMessage = `Lambda execution timed out after ${lambdaTimeoutSeconds} seconds. Press ctrl + C to exit the process.