use of com.amplifyframework.core.Consumer in project amplify-android by aws-amplify.
the class SubscriptionProcessorTest method arrangeDataEmittingSubscription.
@SuppressWarnings("SameParameterValue")
private static <T extends Model> void arrangeDataEmittingSubscription(AppSync appSync, ModelSchema modelSchema, SubscriptionType subscriptionType, GraphQLResponse<ModelWithMetadata<T>> response) throws DataStoreException {
Answer<Cancelable> answer = invocation -> {
final int startConsumerIndex = 1;
Consumer<String> onStart = invocation.getArgument(startConsumerIndex);
onStart.accept(RandomString.string());
final int dataConsumerIndex = 2;
Consumer<GraphQLResponse<ModelWithMetadata<T>>> onData = invocation.getArgument(dataConsumerIndex);
onData.accept(response);
return new NoOpCancelable();
};
arrangeSubscription(appSync, answer, modelSchema, subscriptionType);
}
use of com.amplifyframework.core.Consumer in project amplify-android by aws-amplify.
the class SyncStatusTest method syncStatusGetReturnNotSyncedStatus.
/**
* Get returns false for sync status.
*/
@Test
public void syncStatusGetReturnNotSyncedStatus() {
final LastSyncMetadata lastSyncMetadata = LastSyncMetadata.baseSyncedAt(BlogOwner.class.getName(), Time.now());
List<LastSyncMetadata> resultList = new ArrayList<>();
resultList.add(lastSyncMetadata);
Consumer<DataStoreException> onObservationError = value -> {
};
SqlQueryProcessor mockSqlQueryProcessor = mock(SqlQueryProcessor.class);
when(mockSqlQueryProcessor.queryOfflineData(eq(LastSyncMetadata.class), any(), any())).thenReturn(resultList);
DataStoreConfiguration mockDataStoreConfig = mock(DataStoreConfiguration.class);
when(mockDataStoreConfig.getSyncIntervalInMinutes()).thenReturn(0L);
SyncStatus subject = new SyncStatus(mockSqlQueryProcessor, mockDataStoreConfig);
boolean result = subject.get(LastSyncMetadata.class.getName(), onObservationError);
Assert.assertFalse(result);
}
use of com.amplifyframework.core.Consumer in project amplify-android by aws-amplify.
the class SQLiteStorageAdapterObserveQueryTest method querySavedDataWithSingleItem.
/**
* Test querying the saved item in the SQLite database with observeQuery.
*
* @throws DataStoreException On unexpected failure manipulating items in/out of DataStore
* @throws InterruptedException On unexpected failure manipulating items in/out of DataStore
*/
@Test
public void querySavedDataWithSingleItem() throws DataStoreException, InterruptedException {
CountDownLatch latch = new CountDownLatch(1);
final BlogOwner blogOwner = BlogOwner.builder().name("Alan Turing").build();
adapter.save(blogOwner);
Consumer<Cancelable> observationStarted = NoOpConsumer.create();
Consumer<DataStoreQuerySnapshot<BlogOwner>> onQuerySnapshot = value -> {
assertTrue(value.getItems().contains(blogOwner));
latch.countDown();
};
Consumer<DataStoreException> onObservationError = NoOpConsumer.create();
Action onObservationComplete = NoOpAction.create();
adapter.observeQuery(BlogOwner.class, new ObserveQueryOptions(null, null), observationStarted, onQuerySnapshot, onObservationError, onObservationComplete);
assertTrue(latch.await(1, TimeUnit.SECONDS));
}
use of com.amplifyframework.core.Consumer in project amplify-android by aws-amplify.
the class SQLiteStorageAdapterObserveQueryTest method observeQueryWithOrderBy.
/**
* Test observe query with order by.
* Validate that a list of BlogOwners can be sorted first by name in descending order,
* then by wea in ascending order.
*
* @throws DataStoreException On failure to arrange items into store, or from the
* query action itself
* @throws InterruptedException interruptedException.
*/
@Test
public void observeQueryWithOrderBy() throws DataStoreException, InterruptedException {
// Expect
List<String> names = Arrays.asList("Joe", "Joe", "Joe", "Bob", "Bob", "Bob", "Dan", "Dan", "Dan");
List<String> weas = Arrays.asList("pon", "lth", "ver", "kly", "ken", "sel", "ner", "rer", "ned");
List<BlogOwner> owners = new ArrayList<>();
CountDownLatch latch = new CountDownLatch(1);
for (int i = 0; i < names.size(); i++) {
BlogOwner owner = BlogOwner.builder().name(names.get(i)).wea(weas.get(i)).build();
adapter.save(owner);
owners.add(owner);
}
List<BlogOwner> sorted = new ArrayList<>(owners);
Collections.sort(sorted, Comparator.comparing(BlogOwner::getName).reversed().thenComparing(BlogOwner::getWea));
Consumer<Cancelable> observationStarted = NoOpConsumer.create();
Consumer<DataStoreQuerySnapshot<BlogOwner>> onQuerySnapshot = value -> {
assertEquals(sorted, value.getItems());
latch.countDown();
};
Consumer<DataStoreException> onObservationError = NoOpConsumer.create();
Action onObservationComplete = NoOpAction.create();
List<QuerySortBy> sortBy = new ArrayList<>();
sortBy.add(BlogOwner.NAME.descending());
sortBy.add(BlogOwner.WEA.ascending());
// Act
adapter.observeQuery(BlogOwner.class, new ObserveQueryOptions(null, sortBy), observationStarted, onQuerySnapshot, onObservationError, onObservationComplete);
assertTrue(latch.await(1, TimeUnit.SECONDS));
}
use of com.amplifyframework.core.Consumer in project amplify-android by aws-amplify.
the class SQLiteStorageAdapterObserveQueryTest method queryWithMaliciousPredicates.
/**
* Test query with SQL injection.
*
* @throws DataStoreException On unexpected failure manipulating items in/out of DataStore
* @throws InterruptedException On unexpected failure manipulating items in/out of DataStore
*/
@Test
public void queryWithMaliciousPredicates() throws DataStoreException, InterruptedException {
final BlogOwner jane = BlogOwner.builder().name("Jane Doe").build();
adapter.save(jane);
QueryPredicate predicate = BlogOwner.NAME.eq("Jane; DROP TABLE Person; --");
CountDownLatch latch = new CountDownLatch(2);
Consumer<Cancelable> observationStarted = NoOpConsumer.create();
Consumer<DataStoreQuerySnapshot<BlogOwner>> onMaliciousQuerySnapshot = resultOfMaliciousQuery -> {
assertTrue(resultOfMaliciousQuery.getItems().isEmpty());
latch.countDown();
};
Consumer<DataStoreException> onObservationError = value -> {
};
Action onObservationComplete = NoOpAction.create();
adapter.observeQuery(BlogOwner.class, new ObserveQueryOptions(predicate, null), observationStarted, onMaliciousQuerySnapshot, onObservationError, onObservationComplete);
Consumer<DataStoreQuerySnapshot<BlogOwner>> onAfterMaliciousQuery = resultAfterMaliciousQuery -> {
assertTrue(resultAfterMaliciousQuery.getItems().contains(jane));
latch.countDown();
};
adapter.observeQuery(BlogOwner.class, new ObserveQueryOptions(null, null), observationStarted, onAfterMaliciousQuery, onObservationError, onObservationComplete);
assertTrue(latch.await(1, TimeUnit.SECONDS));
}
Aggregations