private GetObjectStub(boolean withTypeMapping) {
this.withTypeMapping = withTypeMapping;
storIOSQLite = mock(StorIOSQLite.class);
internal = mock(StorIOSQLite.Internal.class);
when(storIOSQLite.internal()).thenReturn(internal);
query = Query.builder().table("test_table").build();
rawQuery =
RawQuery.builder().query("select * from who_cares").observesTables("test_table").build();
//noinspection unchecked
getResolver = mock(GetResolver.class);
cursor = mock(Cursor.class);
item = new TestItem();
when(cursor.getCount()).thenReturn(1);
when(cursor.moveToNext())
.thenAnswer(
new Answer<Boolean>() {
int invocationsCount = 0;
@Override
public Boolean answer(InvocationOnMock invocation) throws Throwable {
return invocationsCount++ < 1;
}
});
when(storIOSQLite.get()).thenReturn(new PreparedGet.Builder(storIOSQLite));
when(storIOSQLite.observeChangesInTables(eq(singleton(query.table()))))
.thenReturn(Observable.<Changes>empty());
assertThat(rawQuery.observesTables()).isNotNull();
when(storIOSQLite.observeChangesInTables(rawQuery.observesTables()))
.thenReturn(Observable.<Changes>empty());
when(getResolver.performGet(storIOSQLite, query)).thenReturn(cursor);
when(getResolver.performGet(storIOSQLite, rawQuery)).thenReturn(cursor);
when(getResolver.mapFromCursor(cursor)).thenReturn(item);
//noinspection unchecked
typeMapping = mock(SQLiteTypeMapping.class);
if (withTypeMapping) {
when(internal.typeMapping(TestItem.class)).thenReturn(typeMapping);
when(typeMapping.getResolver()).thenReturn(getResolver);
}
}
