chore: update tests

aws-amplify · Mar 25, 2024 · 434a57a · 434a57a
1 parent b9d778f
commit 434a57a
Show file tree

Hide file tree

Showing 5 changed files with 23 additions and 8 deletions.
diff --git a/...plify_storage_s3/example/integration_test/content_type_infer/content_type_infer_html.dart b/...plify_storage_s3/example/integration_test/content_type_infer/content_type_infer_html.dart
@@ -39,7 +39,11 @@ void testContentTypeInferTest({
     tearDownAll(() async {
       await Amplify.Storage.removeMany(
         paths: testUploadKeys
-            .map((key) => StoragePath.fromString('private/$key'))
+            .map(
+              (key) => StoragePath.withIdentityId(
+                (identityId) => 'private/$identityId/$key',
+              ),
+            )
             .toList(),
       ).result;
     });
@@ -53,7 +57,9 @@ void testContentTypeInferTest({
         final result = await s3Plugin
             .uploadFile(
               localFile: file,
-              path: StoragePath.fromString('private/${testUploadKeys[index]}'),
+              path: StoragePath.withIdentityId(
+                (identityId) => 'private/$identityId/${testUploadKeys[index]}',
+              ),
               options: const StorageUploadFileOptions(
                 pluginOptions: S3UploadFilePluginOptions(
                   getProperties: true,

diff --git a/...amplify_storage_s3/example/integration_test/content_type_infer/content_type_infer_io.dart b/...amplify_storage_s3/example/integration_test/content_type_infer/content_type_infer_io.dart
@@ -35,7 +35,11 @@ void testContentTypeInferTest({
     tearDownAll(() async {
       await Amplify.Storage.removeMany(
         paths: testUploadKeys
-            .map((key) => StoragePath.fromString('private/$key'))
+            .map(
+              (key) => StoragePath.withIdentityId(
+                (identityId) => 'private/$identityId/$key',
+              ),
+            )
             .toList(),
       ).result;
     });
@@ -50,8 +54,9 @@ void testContentTypeInferTest({
           final result = await s3Plugin
               .uploadFile(
                 localFile: file,
-                path: StoragePath.fromString(
-                  '/private/${testUploadKeys[index]}',
+                path: StoragePath.withIdentityId(
+                  (identityId) =>
+                      'private/$identityId/${testUploadKeys[index]}',
                 ),
                 options: const StorageUploadFileOptions(
                   pluginOptions: S3UploadFilePluginOptions(

diff --git a/packages/storage/amplify_storage_s3/example/integration_test/use_case_test.dart b/packages/storage/amplify_storage_s3/example/integration_test/use_case_test.dart
@@ -766,7 +766,11 @@ void main() {
             // Clean up files from this test.
             await Amplify.Storage.removeMany(
               paths: uploadedKeys
-                  .map((key) => StoragePath.fromString('private/$key'))
+                  .map(
+                    (key) => StoragePath.withIdentityId(
+                      (identityId) => 'private/$identityId/$key',
+                    ),
+                  )
                   .toList(),
             ).result;
           });

diff --git a/packages/storage/amplify_storage_s3_dart/test/amplify_storage_s3_dart_test.dart b/packages/storage/amplify_storage_s3_dart/test/amplify_storage_s3_dart_test.dart
@@ -456,7 +456,7 @@ void main() {
         registerFallbackValue(const StoragePath.fromString('public/$testKey'));
         registerFallbackValue(
           StoragePathWithIdentityId(
-            (identityId) => '/private/$identityId/$testKey',
+            (identityId) => 'private/$identityId/$testKey',
           ),
         );
       });

diff --git a/packages/storage/amplify_storage_s3_dart/test/platform_impl/download_file_io_test.dart b/packages/storage/amplify_storage_s3_dart/test/platform_impl/download_file_io_test.dart
@@ -49,7 +49,7 @@ void main() {
       registerFallbackValue(const StoragePath.fromString('public/$testKey'));
       registerFallbackValue(
         StoragePathWithIdentityId(
-          (identityId) => '/private/$identityId/$testKey',
+          (identityId) => 'private/$identityId/$testKey',
         ),
       );