#855 Add asRxCompletable to PreparedExecuteSQL

pushtorefresh · Nov 21, 2018 · 2f561af · 2f561af
1 parent 9f467b5
commit 2f561af
Show file tree

Hide file tree

Showing 2 changed files with 107 additions and 1 deletion.
diff --git a/...src/main/java/com/pushtorefresh/storio3/sqlite/operations/execute/PreparedExecuteSQL.java b/...src/main/java/com/pushtorefresh/storio3/sqlite/operations/execute/PreparedExecuteSQL.java
@@ -5,6 +5,7 @@
 import android.support.annotation.WorkerThread;
 
 import com.pushtorefresh.storio3.StorIOException;
+import com.pushtorefresh.storio3.operations.PreparedCompletableOperation;
 import com.pushtorefresh.storio3.operations.PreparedOperation;
 import com.pushtorefresh.storio3.sqlite.Changes;
 import com.pushtorefresh.storio3.Interceptor;
@@ -15,6 +16,7 @@
 import java.util.Set;
 
 import io.reactivex.BackpressureStrategy;
+import io.reactivex.Completable;
 import io.reactivex.Flowable;
 import io.reactivex.Single;
 
@@ -24,7 +26,7 @@
 /**
  * Prepared Execute SQL Operation for {@link StorIOSQLite}.
  */
-public class PreparedExecuteSQL implements PreparedOperation<Object, Object, RawQuery> {
+public class PreparedExecuteSQL implements PreparedCompletableOperation<Object, RawQuery> {
 
  @NonNull
  private final StorIOSQLite storIOSQLite;
@@ -98,6 +100,21 @@ public Single<Object> asRxSingle() {
  return RxJavaUtils.createSingle(storIOSQLite, this);
  }
 
+ /**
+ * Creates {@link Completable} which will perform Execute SQL Operation lazily when somebody subscribes to it.
+ * <dl>
+ * <dt><b>Scheduler:</b></dt>
+ * <dd>Operates on {@link StorIOSQLite#defaultRxScheduler()} if not {@code null}.</dd>
+ * </dl>
+ *
+ * @return non-null {@link Completable} which will perform Execute SQL Operation.
+ */
+ @NonNull
+ @Override
+ public Completable asRxCompletable() {
+ return RxJavaUtils.createCompletable(storIOSQLite, this);
+ }
+
  private class RealCallInterceptor implements Interceptor {
  @NonNull
  @Override

diff --git a/...test/java/com/pushtorefresh/storio3/sqlite/operations/execute/PreparedExecuteSQLTest.java b/...test/java/com/pushtorefresh/storio3/sqlite/operations/execute/PreparedExecuteSQLTest.java
@@ -17,6 +17,7 @@
 import java.util.HashSet;
 import java.util.List;
 
+import io.reactivex.Completable;
 import io.reactivex.Flowable;
 import io.reactivex.Single;
 import io.reactivex.functions.Consumer;
@@ -136,6 +137,34 @@ public void executeSQLSingleWithoutNotifications() {
  stub.verifyBehavior(single);
  }
 
+ @Test
+ public void executeSQLCompletableWithNotification() {
+ final Stub stub = Stub.newInstanceApplyNotEmptyAffectedTablesAndTags();
+
+ final Completable completable = stub.storIOSQLite
+ .executeSQL()
+ .withQuery(stub.rawQuery)
+ .prepare()
+ .asRxCompletable();
+
+ verify(stub.storIOSQLite).defaultRxScheduler();
+ stub.verifyBehavior(completable);
+ }
+
+ @Test
+ public void executeSQLCompletableWithoutNotifications() {
+ final Stub stub = Stub.newInstanceDoNotApplyAffectedTablesAndTags();
+
+ final Completable completable = stub.storIOSQLite
+ .executeSQL()
+ .withQuery(stub.rawQuery)
+ .prepare()
+ .asRxCompletable();
+
+ verify(stub.storIOSQLite).defaultRxScheduler();
+ stub.verifyBehavior(completable);
+ }
+
  @Test
  public void executeSQLFlowableWithNotification() {
  final Stub stub = Stub.newInstanceApplyNotEmptyAffectedTablesAndTags();
@@ -176,6 +205,19 @@ public void executeSQLSingleExecutesOnSpecifiedScheduler() {
  schedulerChecker.checkAsSingle(operation);
  }
 
+ @Test
+ public void executeSQLCompletableExecutesOnSpecifiedScheduler() {
+ final Stub stub = Stub.newInstanceApplyNotEmptyAffectedTablesAndTags();
+ final SchedulerChecker schedulerChecker = SchedulerChecker.create(stub.storIOSQLite);
+
+ final PreparedExecuteSQL operation = stub.storIOSQLite
+ .executeSQL()
+ .withQuery(stub.rawQuery)
+ .prepare();
+
+ schedulerChecker.checkAsCompletable(operation);
+ }
+
  @Test
  public void shouldWrapExceptionIntoStorIOExceptionBlocking() {
  final Stub stub = Stub.newInstanceApplyNotEmptyAffectedTablesAndTags();
@@ -264,6 +306,40 @@ public void shouldWrapExceptionIntoStorIOExceptionSingle() {
  verifyNoMoreInteractions(stub.storIOSQLite, stub.lowLevel);
  }
 
+ @Test
+ public void shouldWrapExceptionIntoStorIOExceptionCompletable() {
+ final Stub stub = Stub.newInstanceApplyNotEmptyAffectedTablesAndTags();
+
+ IllegalStateException testException = new IllegalStateException("test exception");
+ doThrow(testException).when(stub.lowLevel).executeSQL(stub.rawQuery);
+
+ final TestObserver<Object> testObserver = new TestObserver<Object>();
+
+ stub.storIOSQLite
+ .executeSQL()
+ .withQuery(stub.rawQuery)
+ .prepare()
+ .asRxCompletable()
+ .subscribe(testObserver);
+
+ testObserver.awaitTerminalEvent();
+ testObserver.assertNoValues();
+ testObserver.assertError(StorIOException.class);
+
+ //noinspection ThrowableResultOfMethodCallIgnored
+ StorIOException expected = (StorIOException) testObserver.errors().get(0);
+
+ IllegalStateException cause = (IllegalStateException) expected.getCause();
+ assertThat(cause).hasMessage("test exception");
+
+ verify(stub.storIOSQLite).executeSQL();
+ verify(stub.storIOSQLite).defaultRxScheduler();
+ verify(stub.storIOSQLite).lowLevel();
+ verify(stub.lowLevel).executeSQL(stub.rawQuery);
+ verify(stub.storIOSQLite).interceptors();
+ verifyNoMoreInteractions(stub.storIOSQLite, stub.lowLevel);
+ }
+
  static class Stub {
 
  private final StorIOSQLite storIOSQLite;
@@ -375,5 +451,18 @@ public void accept(Object anObject) {
  })
  .checkBehaviorOfFlowable();
  }
+
+ void verifyBehavior(@NonNull Completable completable) {
+ new FlowableBehaviorChecker<Object>()
+ .flowable(completable.toFlowable())
+ .expectedNumberOfEmissions(1)
+ .testAction(new Consumer<Object>() {
+ @Override
+ public void accept(Object anObject) {
+ verifyBehavior();
+ }
+ })
+ .checkBehaviorOfFlowable();
+ }
  }
 }