fix tests

Author: tylerjroach

aws-storage-s3/src/test/java/com/amplifyframework/storage/s3/AWSS3StoragePluginTest.kt

@@ -29,11 +29,14 @@ import io.mockk.every
 import io.mockk.mockk
 import io.mockk.verify
 import org.junit.Test
+import org.junit.runner.RunWith
+import org.robolectric.RobolectricTestRunner
 
+@RunWith(RobolectricTestRunner::class)
 class AWSS3StoragePluginTest {
 
     private val storageServiceFactory = mockk<AWSS3StorageService.Factory> {
-        every { create(any(), any(), any(), any(), mockk()) } returns mockk<AWSS3StorageService>()
+        every { create(any(), any(), any(), any(), any()) } returns mockk<AWSS3StorageService>()
     }
 
     private val plugin = AWSS3StoragePlugin(
@@ -51,10 +54,10 @@ class AWSS3StoragePluginTest {
             }
         }
 
-        plugin.configure(data, mockk())
+        plugin.configure(data, mockk(relaxed = true))
 
         verify {
-            storageServiceFactory.create(any(), "test-region", "test-bucket", any(), mockk())
+            storageServiceFactory.create(any(), "test-region", "test-bucket", any(), any())
         }
     }
 
@@ -65,7 +68,7 @@ class AWSS3StoragePluginTest {
         }
 
         shouldThrow<StorageException> {
-            plugin.configure(data, mockk())
+            plugin.configure(data, mockk(relaxed = true))
         }
     }
 
@@ -83,7 +86,7 @@ class AWSS3StoragePluginTest {
             }
         }
 
-        plugin.configure(data, mockk())
+        plugin.configure(data, mockk(relaxed = true))
         val service = plugin.getStorageService(null)
         service shouldNotBe null
     }
@@ -102,7 +105,7 @@ class AWSS3StoragePluginTest {
             }
         }
 
-        plugin.configure(data, mockk())
+        plugin.configure(data, mockk(relaxed = true))
         val bucketInfo = BucketInfo("test-bucket", "test-region")
         val bucket = StorageBucket.fromBucketInfo(bucketInfo)
         val service = plugin.getStorageService(bucket)
@@ -123,7 +126,7 @@ class AWSS3StoragePluginTest {
             }
         }
 
-        plugin.configure(data, mockk())
+        plugin.configure(data, mockk(relaxed = true))
         val bucket = StorageBucket.fromOutputs("test=name")
         val service = plugin.getStorageService(bucket)
         service shouldNotBe null
@@ -143,7 +146,7 @@ class AWSS3StoragePluginTest {
             }
         }
 
-        plugin.configure(data, mockk())
+        plugin.configure(data, mockk(relaxed = true))
         val bucket = StorageBucket.fromOutputs("myBucket")
         val exception = shouldThrow<StorageException> {
             plugin.getStorageService(bucket)
@@ -165,7 +168,7 @@ class AWSS3StoragePluginTest {
             }
         }
 
-        plugin.configure(data, mockk())
+        plugin.configure(data, mockk(relaxed = true))
         val bucket = StorageBucket.fromOutputs("test-name")
         val result = plugin.getStorageServiceResult(bucket)
         val service = result.storageService
@@ -188,7 +191,7 @@ class AWSS3StoragePluginTest {
             }
         }
 
-        plugin.configure(data, mockk())
+        plugin.configure(data, mockk(relaxed = true))
         val bucket = StorageBucket.fromOutputs("myBucket")
         val exception = plugin.getStorageServiceResult(bucket).storageException
         exception shouldNotBe null

aws-storage-s3/src/test/java/com/amplifyframework/storage/s3/StorageComponentTest.java

@@ -35,6 +35,7 @@
 import com.amplifyframework.storage.s3.transfer.StorageTransferClientProvider;
 import com.amplifyframework.storage.s3.transfer.TransferListener;
 import com.amplifyframework.storage.s3.transfer.TransferObserver;
+import com.amplifyframework.storage.s3.transfer.TransferStatusUpdater;
 import com.amplifyframework.testutils.Await;
 import com.amplifyframework.testutils.random.RandomBytes;
 import com.amplifyframework.testutils.random.RandomString;
@@ -91,7 +92,7 @@ public void setup() throws AmplifyException {
         this.storage = new StorageCategory();
         this.storageService = mock(AWSS3StorageService.class);
         AWSS3StorageService.Factory storageServiceFactory
-                = (context, region, bucket, clientProvider) -> (AWSS3StorageService) storageService;
+                = (context, region, bucket, clientProvider, transferStatusUpdater) -> (AWSS3StorageService) storageService;
         AuthCredentialsProvider cognitoAuthProvider = mock(AuthCredentialsProvider.class);
         doReturn(RandomString.string()).when(cognitoAuthProvider).getIdentityId(null);
         this.storage.addPlugin(new AWSS3StoragePlugin(storageServiceFactory,