예제 #1
0
  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);
    }
  }
예제 #2
0
  /**
   * Executes Put Operation immediately in current thread.
   *
   * <p>Notice: This is blocking I/O operation that should not be executed on the Main Thread, it
   * can cause ANR (Activity Not Responding dialog), block the UI and drop animations frames. So
   * please, call this method on some background thread. See {@link WorkerThread}.
   *
   * @return non-null result of Put Operation.
   */
  @SuppressWarnings("unchecked")
  @WorkerThread
  @NonNull
  @Override
  public PutResult executeAsBlocking() {
    try {
      final StorIOSQLite.Internal internal = storIOSQLite.internal();

      final PutResolver<T> putResolver;

      if (explicitPutResolver != null) {
        putResolver = explicitPutResolver;
      } else {
        final SQLiteTypeMapping<T> typeMapping = internal.typeMapping((Class<T>) object.getClass());

        if (typeMapping == null) {
          throw new IllegalStateException(
              "Object does not have type mapping: "
                  + "object = "
                  + object
                  + ", object.class = "
                  + object.getClass()
                  + ", "
                  + "db was not affected by this operation, please add type mapping for this type");
        }

        putResolver = typeMapping.putResolver();
      }

      final PutResult putResult = putResolver.performPut(storIOSQLite, object);

      if (putResult.wasInserted() || putResult.wasUpdated()) {
        internal.notifyAboutChanges(Changes.newInstance(putResult.affectedTables()));
      }

      return putResult;
    } catch (Exception exception) {
      throw new StorIOException(exception);
    }
  }