chore: fix unit tests

aws-amplify · Aug 29, 2024 · 90a04eb · 90a04eb
1 parent 6a19c26
commit 90a04eb
Show file tree

Hide file tree

Showing 4 changed files with 152 additions and 58 deletions.
diff --git a/packages/storage/__tests__/providers/s3/apis/uploadData/index.test.ts b/packages/storage/__tests__/providers/s3/apis/uploadData/index.test.ts
@@ -1,20 +1,24 @@
 // Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
 // SPDX-License-Identifier: Apache-2.0
 
+import { AmplifyClassV6 } from '@aws-amplify/core';
+
 import { uploadData } from '../../../../../src/providers/s3/apis';
 import { MAX_OBJECT_SIZE } from '../../../../../src/providers/s3/utils/constants';
 import { createUploadTask } from '../../../../../src/providers/s3/utils';
 import {
 	StorageValidationErrorCode,
 	validationErrorMap,
 } from '../../../../../src/errors/types/validation';
-import { putObjectJob } from '../../../../../src/providers/s3/apis/uploadData/putObjectJob';
-import { getMultipartUploadHandlers } from '../../../../../src/providers/s3/apis/uploadData/multipart';
+import { putObjectJob } from '../../../../../src/providers/s3/apis/internal/uploadData/putObjectJob';
+import { getMultipartUploadHandlers } from '../../../../../src/providers/s3/apis/internal/uploadData/multipart';
 import { UploadDataInput, UploadDataWithPathInput } from '../../../../../src';
 
 jest.mock('../../../../../src/providers/s3/utils/');
-jest.mock('../../../../../src/providers/s3/apis/uploadData/putObjectJob');
-jest.mock('../../../../../src/providers/s3/apis/uploadData/multipart');
+jest.mock(
+	'../../../../../src/providers/s3/apis/internal/uploadData/putObjectJob',
+);
+jest.mock('../../../../../src/providers/s3/apis/internal/uploadData/multipart');
 
 const testPath = 'testPath/object';
 const mockCreateUploadTask = createUploadTask as jest.Mock;
@@ -172,6 +176,7 @@ describe('uploadData with path', () => {
 				uploadData(testInput);
 
 				expect(mockPutObjectJob).toHaveBeenCalledWith(
+					expect.any(AmplifyClassV6),
 					testInput,
 					expect.any(AbortSignal),
 					expect.any(Number),
@@ -212,6 +217,7 @@ describe('uploadData with path', () => {
 
 			expect(mockPutObjectJob).not.toHaveBeenCalled();
 			expect(mockGetMultipartUploadHandlers).toHaveBeenCalledWith(
+				expect.any(AmplifyClassV6),
 				testInput,
 				expect.any(Number),
 			);