Skip to content

Commit 6af3abf

Browse files
darrelmillerpeombwa
authored andcommitted
Fix tests
1 parent 152e7c7 commit 6af3abf

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

tests/Microsoft.Graph.DotnetCore.Core.Test/Requests/GraphClientFactoryTests.cs

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -19,10 +19,16 @@ namespace Microsoft.Graph.DotnetCore.Core.Test.Requests
1919
public class GraphClientFactoryTests : IDisposable
2020
{
2121
private MockRedirectHandler testHttpMessageHandler;
22+
private DelegatingHandler[] handlers;
2223

2324
public GraphClientFactoryTests()
2425
{
2526
this.testHttpMessageHandler = new MockRedirectHandler();
27+
handlers = handlers = new DelegatingHandler[3];
28+
handlers[0] = new RetryHandler();
29+
handlers[1] = new RedirectHandler();
30+
handlers[2] = new AuthenticationHandler();
31+
2632
}
2733

2834
public void Dispose()
@@ -33,7 +39,7 @@ public void Dispose()
3339
[Fact]
3440
public void CreatePipelineWithoutHttpMessageHandlerInput()
3541
{
36-
using (RetryHandler retryHandler = (RetryHandler)GraphClientFactory.CreateDefaultHandlers())
42+
using (RetryHandler retryHandler = (RetryHandler)GraphClientFactory.CreatePipeline(handlers))
3743
using (RedirectHandler redirectHandler = (RedirectHandler)retryHandler.InnerHandler)
3844
using (AuthenticationHandler authenticationHandler = (AuthenticationHandler)redirectHandler.InnerHandler)
3945
using (HttpClientHandler innerMost = (HttpClientHandler)authenticationHandler.InnerHandler)
@@ -53,7 +59,7 @@ public void CreatePipelineWithoutHttpMessageHandlerInput()
5359
[Fact]
5460
public void CreatePipelineWithHttpMessageHandlerInput()
5561
{
56-
using (RetryHandler retryHandler = (RetryHandler)GraphClientFactory.CreatePipeline(null, this.testHttpMessageHandler))
62+
using (RetryHandler retryHandler = (RetryHandler)GraphClientFactory.CreatePipeline(handlers, this.testHttpMessageHandler))
5763
using (RedirectHandler redirectHandler = (RedirectHandler)retryHandler.InnerHandler)
5864
using (AuthenticationHandler authenticationHandler = (AuthenticationHandler)redirectHandler.InnerHandler)
5965
using (MockRedirectHandler innerMost = (MockRedirectHandler)authenticationHandler.InnerHandler)
@@ -234,11 +240,10 @@ public async Task SendRequest_UnauthorizedWithAuthenticationProvider()
234240

235241
testHttpMessageHandler.SetHttpResponse(unauthorizedResponse, okResponse);
236242

237-
var authHandler = new AuthenticationHandler(new MockAuthenticationProvider().Object);
238-
var pipeline = GraphClientFactory.CreateDefaultHandlers();
243+
handlers[2] = new AuthenticationHandler(new MockAuthenticationProvider().Object);
239244

240245
GraphClientFactory.DefaultHttpHandler = () => this.testHttpMessageHandler;
241-
using (HttpClient client = GraphClientFactory.Create())
246+
using (HttpClient client = GraphClientFactory.Create(handlers: handlers))
242247
{
243248
var response = await client.SendAsync(httpRequestMessage, new CancellationToken());
244249
Assert.Same(response, okResponse);

0 commit comments

Comments
 (0)