Add test case

Author: wangmengyan95

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

@@ -26,6 +26,7 @@
 import static org.junit.Assert.assertArrayEquals;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertNull;
 import static org.junit.Assert.assertTrue;
 import static org.mockito.Matchers.any;
 import static org.mockito.Matchers.anyString;
@@ -55,30 +56,51 @@ public void testConstructor() throws Exception {
     String name = "name";
     byte[] data = "hello".getBytes();
     String contentType = "content_type";
+    File file = temporaryFolder.newFile("test");
 
-    ParseFile file = new ParseFile(name, data, contentType);
-    assertEquals("name", file.getName());
-    assertEquals("hello", new String(file.getData()));
-    assertEquals("content_type", file.getState().mimeType());
-    assertTrue(file.isDirty());
-
-    file = new ParseFile(data);
-    assertEquals("file", file.getName()); // Default
-    assertEquals("hello", new String(file.getData()));
-    assertEquals(null, file.getState().mimeType());
-    assertTrue(file.isDirty());
-
-    file = new ParseFile(name, data);
-    assertEquals("name", file.getName());
-    assertEquals("hello", new String(file.getData()));
-    assertEquals(null, file.getState().mimeType());
-    assertTrue(file.isDirty());
-
-    file = new ParseFile(data, contentType);
-    assertEquals("file", file.getName()); // Default
-    assertEquals("hello", new String(file.getData()));
-    assertEquals("content_type", file.getState().mimeType());
-    assertTrue(file.isDirty());
+    ParseFile parseFile = new ParseFile(name, data, contentType);
+    assertEquals("name", parseFile.getName());
+    assertEquals("hello", new String(parseFile.getData()));
+    assertEquals("content_type", parseFile.getState().mimeType());
+    assertTrue(parseFile.isDirty());
+
+    parseFile = new ParseFile(data);
+    assertEquals("file", parseFile.getName()); // Default
+    assertEquals("hello", new String(parseFile.getData()));
+    assertEquals(null, parseFile.getState().mimeType());
+    assertTrue(parseFile.isDirty());
+
+    parseFile = new ParseFile(name, data);
+    assertEquals("name", parseFile.getName());
+    assertEquals("hello", new String(parseFile.getData()));
+    assertEquals(null, parseFile.getState().mimeType());
+    assertTrue(parseFile.isDirty());
+
+    parseFile = new ParseFile(data, contentType);
+    assertEquals("file", parseFile.getName()); // Default
+    assertEquals("hello", new String(parseFile.getData()));
+    assertEquals("content_type", parseFile.getState().mimeType());
+    assertTrue(parseFile.isDirty());
+
+    // TODO(mengyan): Test file pointer in ParseFile when we have proper stage strategy
+    parseFile = new ParseFile(name, file, contentType);
+    assertEquals("name", parseFile.getName());
+    assertEquals("content_type", parseFile.getState().mimeType());
+    assertTrue(parseFile.isDirty());
+
+    parseFile = new ParseFile(file);
+    assertEquals("file", parseFile.getName()); // Default
+    assertEquals(null, parseFile.getState().mimeType());
+    assertTrue(parseFile.isDirty());
+
+    parseFile = new ParseFile(name, file);
+    assertEquals("name", parseFile.getName());
+    assertEquals(null, parseFile.getState().mimeType());
+    assertTrue(parseFile.isDirty());
+
+    parseFile = new ParseFile(file, contentType);
+    assertEquals("file", parseFile.getName()); // Default
+    assertEquals("content_type", parseFile.getState().mimeType());
   }
 
   @Test(expected = IllegalArgumentException.class)
@@ -162,6 +184,82 @@ public void testSaveAsyncCancelled() throws Exception {
         Matchers.<Task<Void>>any());
   }
 
+  @Test
+  public void testSaveAsyncSuccessWithData() throws Exception {
+    String name = "name";
+    byte[] data = "hello".getBytes();
+    String contentType = "content_type";
+    String url = "url";
+    ParseFile.State state = new ParseFile.State.Builder()
+        .url(url)
+        .build();
+    ParseFileController controller = mock(ParseFileController.class);
+    when(controller.saveAsync(
+        any(ParseFile.State.class),
+        any(byte[].class),
+        any(String.class),
+        any(ProgressCallback.class),
+        Matchers.<Task<Void>>any())).thenReturn(Task.forResult(state));
+    ParseCorePlugins.getInstance().registerFileController(controller);
+
+    ParseFile parseFile = new ParseFile(name, data, contentType);
+    ParseTaskUtils.wait(parseFile.saveAsync(null, null, null));
+
+    // Verify controller get the correct data
+    ArgumentCaptor<ParseFile.State> stateCaptor = ArgumentCaptor.forClass(ParseFile.State.class);
+    ArgumentCaptor<byte[]> dataCaptor = ArgumentCaptor.forClass(byte[].class);
+    verify(controller, times(1)).saveAsync(
+        stateCaptor.capture(),
+        dataCaptor.capture(),
+        any(String.class),
+        any(ProgressCallback.class),
+        Matchers.<Task<Void>>any());
+    assertNull(stateCaptor.getValue().url());
+    assertEquals(name, stateCaptor.getValue().name());
+    assertEquals(contentType, stateCaptor.getValue().mimeType());
+    assertArrayEquals(data, dataCaptor.getValue());
+    // Verify the state of ParseFile has been updated
+    assertEquals(url, parseFile.getUrl());
+  }
+
+  @Test
+  public void testSaveAsyncSuccessWithFile() throws Exception {
+    String name = "name";
+    File file = temporaryFolder.newFile("test");
+    String contentType = "content_type";
+    String url = "url";
+    ParseFile.State state = new ParseFile.State.Builder()
+        .url(url)
+        .build();
+    ParseFileController controller = mock(ParseFileController.class);
+    when(controller.saveAsync(
+        any(ParseFile.State.class),
+        any(File.class),
+        any(String.class),
+        any(ProgressCallback.class),
+        Matchers.<Task<Void>>any())).thenReturn(Task.forResult(state));
+    ParseCorePlugins.getInstance().registerFileController(controller);
+
+    ParseFile parseFile = new ParseFile(name, file, contentType);
+    ParseTaskUtils.wait(parseFile.saveAsync(null, null, null));
+
+    // Verify controller get the correct data
+    ArgumentCaptor<ParseFile.State> stateCaptor = ArgumentCaptor.forClass(ParseFile.State.class);
+    ArgumentCaptor<File> fileCaptor = ArgumentCaptor.forClass(File.class);
+    verify(controller, times(1)).saveAsync(
+        stateCaptor.capture(),
+        fileCaptor.capture(),
+        any(String.class),
+        any(ProgressCallback.class),
+        Matchers.<Task<Void>>any());
+    assertNull(stateCaptor.getValue().url());
+    assertEquals(name, stateCaptor.getValue().name());
+    assertEquals(contentType, stateCaptor.getValue().mimeType());
+    assertEquals(file, fileCaptor.getValue());
+    // Verify the state of ParseFile has been updated
+    assertEquals(url, parseFile.getUrl());
+  }
+
   // TODO(grantland): testSaveAsyncNotDirtyAfterQueueAwait
   // TODO(grantland): testSaveAsyncSuccess
   // TODO(grantland): testSaveAsyncFailure
@@ -287,6 +385,12 @@ public void testTaskQueuedMethods() throws Exception {
         any(String.class),
         any(ProgressCallback.class),
         Matchers.<Task<Void>>any())).thenReturn(Task.forResult(state));
+    when(controller.saveAsync(
+        any(ParseFile.State.class),
+        any(File.class),
+        any(String.class),
+        any(ProgressCallback.class),
+        Matchers.<Task<Void>>any())).thenReturn(Task.forResult(state));
     when(controller.fetchAsync(
         any(ParseFile.State.class),
         any(String.class),