Skip to content

Commit 7535525

Browse files
committed
Fixing review comments on PR : #2
1 parent 84a405a commit 7535525

File tree

6 files changed

+17
-15
lines changed

6 files changed

+17
-15
lines changed

src/main/java/com/microsoft/graph/content/MSBatchRequestContent.java

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,8 @@ public boolean addBatchRequestStep(MSBatchRequestStep batchRequestStep) {
4141
return batchRequestStepsArray.add(batchRequestStep);
4242
}
4343

44-
public void removeBatchRequesStepWithId(String requestId) {
44+
public boolean removeBatchRequesStepWithId(String requestId) {
45+
boolean ret = false;
4546
for (int i = batchRequestStepsArray.size()-1; i >= 0; i--)
4647
{
4748
MSBatchRequestStep requestStep = batchRequestStepsArray.get(i);
@@ -51,11 +52,15 @@ public void removeBatchRequesStepWithId(String requestId) {
5152
if(dependsOnId.compareTo(requestId) == 0)
5253
{
5354
requestStep.getArrayOfDependsOnIds().remove(j);
55+
ret = true;
5456
}
5557
}
56-
if(requestId.compareTo(requestStep.getRequestId()) == 0)
58+
if(requestId.compareTo(requestStep.getRequestId()) == 0) {
5759
batchRequestStepsArray.remove(i);
60+
ret = true;
61+
}
5862
}
63+
return ret;
5964
}
6065

6166
public String getBatchRequestContent() {

src/main/java/com/microsoft/graph/httpcore/AuthenticationHandler.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,8 @@ public AuthenticationHandler(IAuthenticationProvider authProvider) {
1717

1818
@Override
1919
public void process(HttpRequest request, HttpContext context) throws HttpException, IOException {
20-
authProvider.authenticateRequest(request);
20+
String token = authProvider.getAccessToken();
21+
request.addHeader("Authorization", "Bearer " + token);
2122
}
2223

2324
}

src/main/java/com/microsoft/graph/httpcore/IAuthenticationProvider.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,9 @@
44

55
public interface IAuthenticationProvider {
66
/**
7-
* Authenticates the request
7+
* Get Access Token
88
*
9-
* @param request the request to authenticate
109
*/
11-
void authenticateRequest(final HttpRequest request);
10+
11+
String getAccessToken();
1212
}

src/test/java/com/microsoft/graph/httpcore/AuthenticationHandlerTest.java

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -17,11 +17,8 @@ public class AuthenticationHandlerTest {
1717
static String token = "TEST-TOKEN";
1818

1919
public static class AuthProvider implements IAuthenticationProvider{
20-
public static String getToken() {
21-
return "Bearer " + token;
22-
}
23-
public void authenticateRequest(HttpRequest request) {
24-
request.addHeader("Authorization", AuthProvider.getToken());
20+
public String getAccessToken() {
21+
return token;
2522
}
2623
}
2724

src/test/java/com/microsoft/graph/httpcore/HttpClientsTest.java

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22

33
import static org.junit.Assert.assertTrue;
44

5-
import org.apache.http.HttpRequest;
65
import org.apache.http.impl.client.CloseableHttpClient;
76
import org.junit.Test;
87

@@ -12,8 +11,8 @@ public class HttpClientsTest {
1211
public void testHttpClientCreation() {
1312
IAuthenticationProvider authprovider = new IAuthenticationProvider() {
1413
@Override
15-
public void authenticateRequest(HttpRequest request) {
16-
request.addHeader("Authorization", "TOKEN");
14+
public String getAccessToken() {
15+
return "TOKEN";
1716
}
1817
};
1918
CloseableHttpClient httpclient = HttpClients.createDefault(authprovider);

src/test/java/com/microsoft/graph/httpcore/RetryHandlerTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ public void testRetryRequestWithExponentialBackOff() {
7777
HttpClientContext localContext = HttpClientContext.create();
7878
localContext.setAttribute(HttpCoreContext.HTTP_REQUEST, httppost);
7979
assertTrue(retryhandler.retryRequest(response, 1, localContext));
80-
assertTrue(retryhandler.getRetryInterval() == 20);
80+
assertTrue(retryhandler.getRetryInterval() == 2000);
8181

8282
} catch (UnsupportedEncodingException e) {
8383
e.printStackTrace();

0 commit comments

Comments
 (0)