Tests

Author: natario1

Parse/src/test/java/com/parse/ParseQueryTest.java

@@ -28,6 +28,7 @@
 
 import static org.hamcrest.CoreMatchers.instanceOf;
 import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertNotSame;
 import static org.junit.Assert.assertNull;
 import static org.junit.Assert.assertSame;
@@ -105,73 +106,56 @@ public void testSetUser() throws ParseException {
   }
 
   @Test
-  public void testQueriesOnlyRunOneNetworkConnection() throws ParseException {
-    TestQueryController controller = new TestQueryController();
-    ParseCorePlugins.getInstance().registerQueryController(controller);
+  public void testMultipleQueries() throws ParseException {
+    TestQueryController controller1 = new TestQueryController();
+    TestQueryController controller2 = new TestQueryController();
 
-    TaskCompletionSource<Void> tcs = new TaskCompletionSource();
-    controller.await(tcs.getTask());
+    TaskCompletionSource<Void> tcs1 = new TaskCompletionSource<>();
+    TaskCompletionSource<Void> tcs2 = new TaskCompletionSource<>();
+    controller1.await(tcs1.getTask());
+    controller2.await(tcs2.getTask());
 
     ParseQuery<ParseObject> query = ParseQuery.getQuery("TestObject");
     query.setUser(new ParseUser());
-    Task<Void> task = query.findInBackground().makeVoid();
 
-    try {
-      query.find();
-      fail("Should've thrown an exception.");
-    } catch (RuntimeException e) {
-      // do nothing
-    }
+    ParseCorePlugins.getInstance().registerQueryController(controller1);
+    query.findInBackground();
+    assertTrue(query.isRunning());
 
-    try {
-      query.findInBackground(new FindCallback<ParseObject>() {
-        @Override
-        public void done(List<ParseObject> objects, ParseException e) {
-        }
-      });
-      fail("Should've thrown an exception.");
-    } catch (RuntimeException e) {
-      // do nothing
-    }
+    ParseCorePlugins.getInstance().reset();
+    ParseCorePlugins.getInstance().registerQueryController(controller2);
+    query.countInBackground();
+    assertTrue(query.isRunning());
 
-    try {
-      query.count();
-      fail("Should've thrown an exception.");
-    } catch (RuntimeException e) {
-      // do nothing
-    }
+    // Stop the first operation.
+    tcs1.setResult(null);
+    assertTrue(query.isRunning());
 
-    try {
-      query.countInBackground(new CountCallback() {
-        @Override
-        public void done(int count, ParseException e) {
-        }
-      });
-      fail("Should've thrown an exception.");
-    } catch (RuntimeException e) {
-      // do nothing
-    }
+    // Stop the second.
+    tcs2.setResult(null);
+    assertFalse(query.isRunning());
+  }
 
-    try {
-      query.get("abc");
-      fail("Should've thrown an exception.");
-    } catch (RuntimeException e) {
-      // do nothing
-    }
+  @Test
+  public void testMultipleQueriesWithInflightChanges() throws ParseException {
+    Parse.enableLocalDatastore(null);
+    TestQueryController controller = new TestQueryController();
+    TaskCompletionSource<Void> tcs = new TaskCompletionSource<>();
+    controller.await(tcs.getTask());
 
-    try {
-      query.getInBackground("abc", new GetCallback<ParseObject>() {
-        @Override
-        public void done(ParseObject object, ParseException e) {
-        }
-      });
-      fail("Should've thrown an exception.");
-    } catch (RuntimeException e) {
-      // do nothing
-    }
+    ParseQuery<ParseObject> query = ParseQuery.getQuery("TestObject");
+    query.setUser(new ParseUser());
 
-    tcs.setResult(null);
-    ParseTaskUtils.wait(task);
+    ParseCorePlugins.getInstance().registerQueryController(controller);
+    List<Task<Void>> tasks = Arrays.asList(
+        query.fromNetwork().findInBackground().makeVoid(),
+        query.fromLocalDatastore().findInBackground().makeVoid(),
+        query.setLimit(10).findInBackground().makeVoid(),
+        query.whereEqualTo("key", "value").countInBackground().makeVoid());
+    assertTrue(query.isRunning());
+    tcs.trySetResult(null);
+    ParseTaskUtils.wait(Task.whenAll(tasks));
+    assertFalse(query.isRunning());
   }
 
   @Test
@@ -240,6 +224,29 @@ public void testCountWithCallbackLimit() throws ParseException {
     assertEquals(0, state.getValue().limit());
   }
 
+  @Test
+  public void testIsRunning() throws ParseException {
+    TestQueryController controller = new TestQueryController();
+    ParseCorePlugins.getInstance().registerQueryController(controller);
+    TaskCompletionSource<Void> tcs = new TaskCompletionSource<>();
+    controller.await(tcs.getTask());
+
+    ParseQuery<ParseObject> query = ParseQuery.getQuery("TestObject");
+    query.setUser(new ParseUser());
+    assertFalse(query.isRunning());
+    query.findInBackground();
+    assertTrue(query.isRunning());
+    tcs.setResult(null);
+    assertFalse(query.isRunning());
+    // Run another
+    tcs = new TaskCompletionSource<>();
+    controller.await(tcs.getTask());
+    query.findInBackground();
+    assertTrue(query.isRunning());
+    query.cancel();
+    assertFalse(query.isRunning());
+  }
+
   @Test
   public void testQueryCancellation() throws ParseException {
     TestQueryController controller = new TestQueryController();