public async Task Test_UserAgentHttpHandler_CustomHeader()
{
IHttpExtensionHandler extensionHandler = Substitute.For <IHttpExtensionHandler>();
extensionHandler.SendAsync(
null,
default(CancellationToken),
null).ReturnsForAnyArgs(
(callInfo) =>
{
HttpRequestMessage requestMessage = callInfo[0] as HttpRequestMessage;
foreach (ProductInfoHeaderValue value in requestMessage.Headers.UserAgent)
{
Assert.AreEqual(
"Graph-ManagedAPI-abc",
value.Product.Name);
}
return(Task.FromResult(new HttpResponseMessage(HttpStatusCode.OK)));
});
HttpRequestContext context = new HttpRequestContext();
context.HttpExtensionHandler = extensionHandler;
context.AuthorizationProvider = Substitute.For <IAuthorizationProvider>();
using (HttpRequest request = HttpRequest.Get(new Uri("http://localhost"), context))
{
request.UserAgent = "abc";
await request.GetHttpResponseAsync();
}
}
public async Task Test_AuthZHttpHandler_SuccessfullyAuthenticates()
{
IHttpExtensionHandler extensionHandler = Substitute.For <IHttpExtensionHandler>();
extensionHandler.SendAsync(
new HttpRequestMessage(HttpMethod.Get, new Uri("http://localhost")),
default(CancellationToken),
null).ReturnsForAnyArgs(
(callInfo) =>
{
return(Task.FromResult(new HttpResponseMessage(HttpStatusCode.Unauthorized)));
},
(callInfo) =>
{
HttpRequestMessage requestMessage = callInfo[0] as HttpRequestMessage;
Assert.AreEqual(
"Bearer",
requestMessage.Headers.Authorization.Scheme);
Assert.AreEqual(
"123",
requestMessage.Headers.Authorization.Parameter);
return(Task.FromResult(new HttpResponseMessage(HttpStatusCode.OK)));
});
IAuthorizationProvider authorizationProvider = Substitute.For <IAuthorizationProvider>();
authorizationProvider.GetAuthenticationHeader().Returns(new AuthenticationHeaderValue("Bearer", "123"));
HttpRequestContext context = new HttpRequestContext();
context.HttpExtensionHandler = extensionHandler;
context.AuthorizationProvider = authorizationProvider;
using (HttpRequest get = HttpRequest.Get(new Uri("http://localhost"), context))
{
HttpResponse response = await get.GetHttpResponseAsync();
IEnumerable <string> header = response.ResponseHeaders.GetValues("X-RetryAttempt-HttpAuthZHandler");
Assert.IsNotNull(
header);
foreach (string s in header)
{
int i = 0;
Assert.IsTrue(
int.TryParse(s, out i));
Assert.AreEqual(
1,
i);
}
}
}
public async Task Test_AuthZHttpHandler_RetryCountExceeded()
{
IHttpExtensionHandler extensionHandler = Substitute.For <IHttpExtensionHandler>();
extensionHandler.SendAsync(
null,
default(CancellationToken),
null).ReturnsForAnyArgs(
Task.FromResult(
new HttpResponseMessage(HttpStatusCode.Unauthorized)));
HttpRequestContext context = new HttpRequestContext();
context.HttpExtensionHandler = extensionHandler;
context.AuthorizationProvider = Substitute.For <IAuthorizationProvider>();
using (HttpRequest get = HttpRequest.Get(new Uri("http://localhost"), context))
{
HttpResponse response = await get.GetHttpResponseAsync();
IEnumerable <string> headerValues;
Assert.IsTrue(
response.ResponseHeaders.TryGetValues("X-RetryAttempt-HttpAuthZHandler", out headerValues));
foreach (string s in headerValues)
{
Assert.AreEqual(
"3",
s);
}
Assert.IsTrue(
response.ResponseHeaders.TryGetValues("X-TotalDelayApplied-HttpAuthZHandler", out headerValues));
foreach (string s in headerValues)
{
Assert.AreEqual(
"0",
s);
}
}
}
public async Task Test_HttpRequestHeaderHandler()
{
IHttpExtensionHandler extensionHandler = Substitute.For <IHttpExtensionHandler>();
extensionHandler.SendAsync(
null,
default(CancellationToken),
null).ReturnsForAnyArgs(
(callInfo) =>
{
HttpRequestMessage requestMessage = callInfo[0] as HttpRequestMessage;
IEnumerable <string> headers;
Assert.IsTrue(requestMessage.Headers.TryGetValues("SdkVersion", out headers));
foreach (string header in headers)
{
Assert.IsTrue(header.StartsWith("Graph-ManagedAPI/"));
}
Assert.IsTrue(requestMessage.Headers.TryGetValues("client-request-id", out headers));
foreach (string header in headers)
{
Guid clientRequestId;
Assert.IsTrue(Guid.TryParse(header, out clientRequestId));
}
return(Task.FromResult(new HttpResponseMessage(HttpStatusCode.OK)));
});
HttpRequestContext context = new HttpRequestContext();
context.HttpExtensionHandler = extensionHandler;
context.AuthorizationProvider = Substitute.For <IAuthorizationProvider>();
using (HttpRequest request = HttpRequest.Get(new Uri("http://localhost"), context))
{
await request.GetHttpResponseAsync();
}
}
public async Task Test_ThrottlingHttpHandler_SuccessfulRetry()
{
IHttpExtensionHandler extensionHandler = Substitute.For <IHttpExtensionHandler>();
extensionHandler.SendAsync(
new HttpRequestMessage(HttpMethod.Get, new Uri("http://localhost")),
default(CancellationToken),
null).ReturnsForAnyArgs(
(callInfo) =>
{
return(Task.FromResult(new HttpResponseMessage((HttpStatusCode)429)));
},
(callInfo) =>
{
return(Task.FromResult(new HttpResponseMessage(HttpStatusCode.ServiceUnavailable)));
},
(callInfo) =>
{
return(Task.FromResult(new HttpResponseMessage(HttpStatusCode.OK)));
});
IAuthorizationProvider authorizationProvider = Substitute.For <IAuthorizationProvider>();
authorizationProvider.GetAuthenticationHeader().Returns(new AuthenticationHeaderValue("Bearer", "123"));
HttpRequestContext context = new HttpRequestContext();
context.HttpExtensionHandler = extensionHandler;
context.AuthorizationProvider = authorizationProvider;
using (HttpRequest get = HttpRequest.Get(new Uri("http://localhost"), context))
{
HttpResponse response = await get.GetHttpResponseAsync();
IEnumerable <string> header = response.ResponseHeaders.GetValues("X-RetryAttempt-HttpThrottlingHandler");
Assert.IsNotNull(
header);
foreach (string s in header)
{
int i = 0;
Assert.IsTrue(
int.TryParse(s, out i));
Assert.AreEqual(
2,
i);
}
header = response.ResponseHeaders.GetValues("X-TotalDelayApplied-HttpThrottlingHandler");
Assert.IsNotNull(
header);
foreach (string s in header)
{
int i = 0;
Assert.IsTrue(
int.TryParse(s, out i));
Assert.AreEqual(
10,
i);
}
}
}
