test: fix new func test

aws-amplify · Feb 1, 2021 · e106da1 · e106da1
1 parent 19cd69d
commit e106da1
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/...es/amplify-category-function/src/__tests__/provider-utils/awscloudformation/index.test.ts b/...es/amplify-category-function/src/__tests__/provider-utils/awscloudformation/index.test.ts
@@ -3,7 +3,7 @@ import { ServiceName } from '../../../provider-utils/awscloudformation/utils/con
 import open from 'open';
 import { $TSContext, stateManager } from 'amplify-cli-core';
 import { buildFunction } from '../../../provider-utils/awscloudformation/utils/buildFunction';
-import { getBuilder } from '../../../../lib';
+import { getBuilder } from '../../..';
 import { BuildType } from 'amplify-function-plugin-interface';
 
 jest.mock('amplify-cli-core');
@@ -18,7 +18,13 @@ stateManager_mock.getMeta.mockReturnValue({
 });
 jest.mock('open');
 
-jest.mock('../../../provider-utils/awscloudformation/utils/buildFunction');
+jest.mock('../../../provider-utils/awscloudformation/utils/buildFunction', () => ({
+  buildFunction: jest.fn(),
+  buildTypeKeyMap: {
+    PROD: 'lastBuildTimeStamp',
+    DEV: 'lastDevBuildTimeStamp',
+  },
+}));
 const buildFunction_mock = buildFunction as jest.MockedFunction<typeof buildFunction>;
 
 describe('awscloudformation function provider', () => {