Search in sources :

Example 6 with HubEvent

use of com.amplifyframework.hub.HubEvent in project amplify-android by aws-amplify.

the class AWSS3StorageUploadTest method testUploadFileIsResumable.

/**
 * Tests that file upload operation can be paused and resumed
 * while the transfer hasn't completed yet.
 *
 * @throws Exception if upload is not paused, resumed, and
 *         completed successfully before timeout
 */
@SuppressWarnings("unchecked")
@Ignore("Contains test which either hang themselves, or hang the suite overall.")
public void testUploadFileIsResumable() throws Exception {
    final CountDownLatch completed = new CountDownLatch(1);
    final CountDownLatch resumed = new CountDownLatch(1);
    final AtomicReference<Resumable> opContainer = new AtomicReference<>();
    final AtomicReference<Throwable> errorContainer = new AtomicReference<>();
    // Create a file large enough that transfer won't finish before being paused
    File uploadFile = new RandomTempFile(LARGE_FILE_SIZE);
    // Listen to Hub events to resume when operation has been paused
    SubscriptionToken resumeToken = Amplify.Hub.subscribe(HubChannel.STORAGE, hubEvent -> {
        if (StorageChannelEventName.UPLOAD_STATE.toString().equals(hubEvent.getName())) {
            HubEvent<String> stateEvent = (HubEvent<String>) hubEvent;
            TransferState state = TransferState.getState(stateEvent.getData());
            if (TransferState.PAUSED.equals(state)) {
                opContainer.get().resume();
                resumed.countDown();
            }
        }
    });
    subscriptions.add(resumeToken);
    // Begin uploading a large file
    StorageUploadFileOperation<?> op = storageCategory.uploadFile(uploadFile.getName(), uploadFile, options, progress -> {
        if (progress.getCurrentBytes() > 0 && resumed.getCount() > 0) {
            opContainer.get().pause();
        }
    }, result -> completed.countDown(), errorContainer::set);
    opContainer.set(op);
    // Assert that all the required conditions have been met
    assertTrue(resumed.await(EXTENDED_TIMEOUT_MS, TimeUnit.MILLISECONDS));
    assertTrue(completed.await(EXTENDED_TIMEOUT_MS, TimeUnit.MILLISECONDS));
    assertNull(errorContainer.get());
}
Also used : RandomTempFile(com.amplifyframework.testutils.random.RandomTempFile) HubEvent(com.amplifyframework.hub.HubEvent) AtomicReference(java.util.concurrent.atomic.AtomicReference) CountDownLatch(java.util.concurrent.CountDownLatch) TransferState(com.amazonaws.mobileconnectors.s3.transferutility.TransferState) Resumable(com.amplifyframework.core.async.Resumable) SubscriptionToken(com.amplifyframework.hub.SubscriptionToken) File(java.io.File) RandomTempFile(com.amplifyframework.testutils.random.RandomTempFile) Ignore(org.junit.Ignore)

Example 7 with HubEvent

use of com.amplifyframework.hub.HubEvent in project amplify-android by aws-amplify.

the class AWSS3StorageUploadTest method testUploadFileIsCancelable.

/**
 * Tests that file upload operation can be canceled while the
 * transfer hasn't completed yet.
 *
 * @throws Exception if upload is not canceled successfully
 *         before timeout
 */
@SuppressWarnings("unchecked")
@Ignore("Contains test which either hang themselves, or hang the suite overall.")
public void testUploadFileIsCancelable() throws Exception {
    final CountDownLatch canceled = new CountDownLatch(1);
    final AtomicReference<Cancelable> opContainer = new AtomicReference<>();
    final AtomicReference<Throwable> errorContainer = new AtomicReference<>();
    // Create a file large enough that transfer won't finish before being canceled
    File uploadFile = new RandomTempFile(LARGE_FILE_SIZE);
    // Listen to Hub events for cancel
    SubscriptionToken cancelToken = Amplify.Hub.subscribe(HubChannel.STORAGE, hubEvent -> {
        if (StorageChannelEventName.UPLOAD_STATE.toString().equals(hubEvent.getName())) {
            HubEvent<String> stateEvent = (HubEvent<String>) hubEvent;
            TransferState state = TransferState.getState(stateEvent.getData());
            if (TransferState.CANCELED.equals(state)) {
                canceled.countDown();
            }
        }
    });
    subscriptions.add(cancelToken);
    // Begin uploading a large file
    StorageUploadFileOperation<?> op = storageCategory.uploadFile(uploadFile.getName(), uploadFile, options, progress -> {
        if (progress.getCurrentBytes() > 0) {
            opContainer.get().cancel();
        }
    }, result -> errorContainer.set(new RuntimeException("Upload completed without canceling.")), errorContainer::set);
    opContainer.set(op);
    // Assert that the required conditions have been met
    assertTrue(canceled.await(EXTENDED_TIMEOUT_MS, TimeUnit.MILLISECONDS));
    assertNull(errorContainer.get());
}
Also used : RandomTempFile(com.amplifyframework.testutils.random.RandomTempFile) HubEvent(com.amplifyframework.hub.HubEvent) AtomicReference(java.util.concurrent.atomic.AtomicReference) CountDownLatch(java.util.concurrent.CountDownLatch) TransferState(com.amazonaws.mobileconnectors.s3.transferutility.TransferState) SubscriptionToken(com.amplifyframework.hub.SubscriptionToken) Cancelable(com.amplifyframework.core.async.Cancelable) File(java.io.File) RandomTempFile(com.amplifyframework.testutils.random.RandomTempFile) Ignore(org.junit.Ignore)

Aggregations

HubEvent (com.amplifyframework.hub.HubEvent)7 TransferState (com.amazonaws.mobileconnectors.s3.transferutility.TransferState)4 SubscriptionToken (com.amplifyframework.hub.SubscriptionToken)4 CountDownLatch (java.util.concurrent.CountDownLatch)4 AtomicReference (java.util.concurrent.atomic.AtomicReference)4 Ignore (org.junit.Ignore)4 BlogOwner (com.amplifyframework.testmodels.commentsblog.BlogOwner)3 HubAccumulator (com.amplifyframework.testutils.HubAccumulator)3 Test (org.junit.Test)3 AmplifyException (com.amplifyframework.AmplifyException)2 Cancelable (com.amplifyframework.core.async.Cancelable)2 Resumable (com.amplifyframework.core.async.Resumable)2 ModelWithMetadata (com.amplifyframework.datastore.appsync.ModelWithMetadata)2 HubChannel (com.amplifyframework.hub.HubChannel)2 RandomString (com.amplifyframework.testutils.random.RandomString)2 RandomTempFile (com.amplifyframework.testutils.random.RandomTempFile)2 Observable (io.reactivex.rxjava3.core.Observable)2 File (java.io.File)2 TimeUnit (java.util.concurrent.TimeUnit)2 Assert.assertEquals (org.junit.Assert.assertEquals)2