From ca5e567366f4ffe00d5664a71023bceab54c0a90 Mon Sep 17 00:00:00 2001
From: Tyler-Larkin <tyllark@amazon.com>
Date: Thu, 14 Nov 2024 03:38:14 -0800
Subject: [PATCH] chore(storage): storage copy integration tests

---
 .../example/integration_test/copy_test.dart   | 84 +++++++++++++++++++
 1 file changed, 84 insertions(+)

diff --git a/packages/storage/amplify_storage_s3/example/integration_test/copy_test.dart b/packages/storage/amplify_storage_s3/example/integration_test/copy_test.dart
index 87afda83b3..66ebdd869d 100644
--- a/packages/storage/amplify_storage_s3/example/integration_test/copy_test.dart
+++ b/packages/storage/amplify_storage_s3/example/integration_test/copy_test.dart
@@ -138,5 +138,89 @@ void main() {
         expect(result.copiedItem.path, destinationPath);
       });
     });
+
+    group('multi bucket', () {
+      final bucketData = 'copy data'.codeUnits;
+      final bucket1 = StorageBucket.fromOutputs(
+        'Storage Integ Test main bucket',
+      );
+      final bucket2 = StorageBucket.fromOutputs(
+        'Storage Integ Test secondary bucket',
+      );
+      final bucket1PathSource = 'public/multi-bucket-get-url-${uuid()}';
+      final bucket2PathSource = 'public/multi-bucket-get-url-${uuid()}';
+      final bucket2PathDestination = 'public/multi-bucket-get-url-${uuid()}';
+      final storageBucket1PathSource =
+          StoragePath.fromString(bucket1PathSource);
+      final storageBucket2PathSource =
+          StoragePath.fromString(bucket2PathSource);
+      final storageBucket2PathDestination =
+          StoragePath.fromString(bucket2PathDestination);
+
+      setUp(() async {
+        addTearDownPath(storageBucket1PathSource);
+        addTearDownPath(storageBucket2PathSource);
+        addTearDownPath(storageBucket2PathDestination);
+        await Amplify.Storage.uploadData(
+          data: StorageDataPayload.bytes(bucketData),
+          path: storageBucket1PathSource,
+          options: StorageUploadDataOptions(
+            bucket: bucket1,
+          ),
+        ).result;
+        await Amplify.Storage.uploadData(
+          data: StorageDataPayload.bytes(bucketData),
+          path: storageBucket2PathSource,
+          options: StorageUploadDataOptions(
+            bucket: bucket2,
+          ),
+        ).result;
+      });
+
+      testWidgets('copy to a different bucket', (_) async {
+        final result = Amplify.Storage.copy(
+          source: storageBucket1PathSource,
+          destination: storageBucket2PathDestination,
+          options: StorageCopyOptions(
+            buckets: CopyBuckets(
+              source: bucket1,
+              destination: bucket2,
+            ),
+          ),
+        ).result;
+        expect(await result, returnsNormally);
+
+        final downloadResult = await Amplify.Storage.downloadData(
+          path: storageBucket2PathDestination,
+          options: StorageDownloadDataOptions(bucket: bucket2),
+        ).result;
+        expect(
+          downloadResult.bytes,
+          bucketData,
+        );
+      });
+
+      testWidgets('copy to the same bucket', (_) async {
+        final result = Amplify.Storage.copy(
+          source: storageBucket2PathSource,
+          destination: storageBucket2PathDestination,
+          options: StorageCopyOptions(
+            buckets: CopyBuckets.sameBucket(
+              bucket2,
+            ),
+          ),
+        ).result;
+        expect(await result, returnsNormally);
+
+        final downloadResult = await Amplify.Storage.downloadData(
+          path: storageBucket2PathDestination,
+          options: StorageDownloadDataOptions(bucket: bucket2),
+        ).result;
+        expect(
+          downloadResult.bytes,
+          bucketData,
+        );
+      });
+    });
   });
 }