BatchTests.java

package com.microsoft.graph.functional;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;

import java.io.IOException;

import com.microsoft.graph.content.MSBatchRequestContent;
import com.microsoft.graph.content.MSBatchResponseContent;
import com.microsoft.graph.http.HttpMethod;
import com.microsoft.graph.httpcore.HttpClients;
import com.microsoft.graph.httpcore.ICoreAuthenticationProvider;
import com.microsoft.graph.models.extensions.IGraphServiceClient;
import com.microsoft.graph.models.extensions.User;
import com.microsoft.graph.requests.extensions.UserRequest;

import org.junit.Before;
import org.junit.Ignore;
import org.junit.Test;

import okhttp3.MediaType;
import okhttp3.OkHttpClient;
import okhttp3.Request;
import okhttp3.RequestBody;
import okhttp3.Response;

@Ignore
public class BatchTests {
    IGraphServiceClient graphServiceClient = null;

	@Before
	public void setUp() {
		final TestBase testBase = new TestBase();
		graphServiceClient = testBase.graphClient;
    }
    @Test
    public void GetsABatchFromRequests() throws IOException{
        final MSBatchRequestContent batchContent = new MSBatchRequestContent();
        final String meGetId = batchContent.addBatchRequestStep(graphServiceClient.me()
                                        .buildRequest()
                                        .withHttpMethod(HttpMethod.GET)
                                        .getHttpRequest());
        assertNotNull(meGetId);
        final String usersGetId = batchContent.addBatchRequestStep(graphServiceClient.users()
                        .buildRequest()
                        .filter("accountEnabled eq true")
                        .expand("manager")
                        .top(5)
                        .withHttpMethod(HttpMethod.GET)
                        .getHttpRequest(),
                        meGetId);
        final User userToAdd = new User();
        userToAdd.givenName = "Darrel";
        final String userPostId = batchContent.addBatchRequestStep(graphServiceClient.users()
                        .buildRequest()
                        .withHttpMethod(HttpMethod.POST)
                        .getHttpRequest(userToAdd), usersGetId);

        final String serializedBatchContent = batchContent.getBatchRequestContent();

        final Request batchRequest = new Request.Builder()
                        .url("https://graph.microsoft.com/v1.0/$batch")
                        .post(RequestBody.create(MediaType.parse("application/json"), serializedBatchContent))
                        .build();
        
        final OkHttpClient client = HttpClients.createDefault((ICoreAuthenticationProvider)graphServiceClient.getAuthenticationProvider());
        final Response batchResponse = client.newCall(batchRequest).execute();
        assertEquals(200, batchResponse.code());

        final MSBatchResponseContent responseContent = new MSBatchResponseContent(batchResponse);
        batchResponse.close();

        assertEquals(400, responseContent.getResponseById(userPostId).code()); //400:we're not providing enough properties for the call to go through
        assertEquals(200, responseContent.getResponseById(meGetId).code());
        assertEquals(200, responseContent.getResponseById(usersGetId).code());
    }
}