- arranges unit test to upload file before testing

Author: baywet

src/test/java/com/microsoft/graph/functional/OneDriveTests.java

@@ -3,14 +3,17 @@
 import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertTrue;
 
+import java.io.ByteArrayInputStream;
 import java.io.IOException;
 import java.io.InputStream;
+import java.nio.charset.StandardCharsets;
 
 import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Ignore;
 import org.junit.Test;
 
+import com.google.gson.JsonPrimitive;
 import com.microsoft.graph.concurrency.ChunkedUploadProvider;
 import com.microsoft.graph.concurrency.IProgressCallback;
 import com.microsoft.graph.core.ClientException;
@@ -27,7 +30,25 @@ public class OneDriveTests {
     public void setUp() {
        testBase = new TestBase();
     }
-    
+	
+	IProgressCallback<DriveItem> callback = new IProgressCallback<DriveItem> () {
+		@Override
+		public void progress(final long current, final long max) {
+			//Check progress
+		}
+		@Override
+		public void success(final DriveItem result) {
+			//Handle the successful response
+			String finishedItemId = result.id;
+			Assert.assertNotNull(finishedItemId);
+		}
+		
+		@Override
+		public void failure(final ClientException ex) {
+			//Handle the failed upload
+			Assert.fail("Upload session failed");
+		}
+	};
     /**
      * Test large file upload. 
      * https://github.com/OneDrive/onedrive-sdk-csharp/blob/master/docs/chunked-uploads.md
@@ -43,25 +64,6 @@ public void testLargeFileUpload() throws IOException, InterruptedException {
     	InputStream uploadFile = OneDriveTests.class.getClassLoader().getResourceAsStream("hamilton.jpg");
     	long fileSize = (long) uploadFile.available();
 
-    	IProgressCallback<DriveItem> callback = new IProgressCallback<DriveItem> () {
-    		@Override
-        	public void progress(final long current, final long max) {
-        		//Check progress
-        	}
-    		@Override
-    		public void success(final DriveItem result) {
-    			//Handle the successful response
-    			String finishedItemId = result.id;
-    			Assert.assertNotNull(finishedItemId);
-    		}
-    		
-    		@Override
-    		public void failure(final ClientException ex) {
-    			//Handle the failed upload
-    			Assert.fail("Upload session failed");
-    		}
-    	};
-    	
     	UploadSession uploadSession = testBase
     			.graphClient
     			.me()
@@ -89,13 +91,38 @@ public void testDownloadWithCustomRequest() throws IOException {
 	}
 	@Test
 	public void downloadJsonFileFromOneDrive() throws Exception {
+		final DriveItemUploadableProperties item = new DriveItemUploadableProperties();
+		item.name = "test.json";
+		item.additionalDataManager().put("@microsoft.graph.conflictBehavior", new JsonPrimitive("replace"));
+
+		final InputStream uploadFile = new ByteArrayInputStream("{\"hehe\":\"haha\"}".getBytes(StandardCharsets.UTF_8));
+
+    	final long fileSize = (long) uploadFile.available();
+
+		final UploadSession session = testBase.graphClient.me()
+			.drive()
+			.root()
+			.itemWithPath(item.name)
+			.createUploadSession(item)
+			.buildRequest()
+			.post();
+
+		ChunkedUploadProvider<DriveItem> chunkedUploadProvider = new ChunkedUploadProvider<DriveItem>(
+    			session, 
+    			testBase.graphClient, 
+    			uploadFile, 
+    			fileSize, 
+    			DriveItem.class);
+		
+		chunkedUploadProvider.upload(callback);
+		
 		final InputStream stream = testBase.graphClient.me()
-		.drive()
-		.root()
-		.itemWithPath("test.json")
-		.content()
-		.buildRequest()
-		.get();
+			.drive()
+			.root()
+			.itemWithPath(item.name)
+			.content()
+			.buildRequest()
+			.get();
 
 		final String fileContent = CoreHttpProvider.streamToString(stream);