Skip to content

Commit bc0cf8e

Browse files
committed
Refactor test
1 parent a4e62a2 commit bc0cf8e

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

test/Grpc.Net.ClientFactory.Tests/DefaultGrpcClientFactoryTests.cs

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -61,13 +61,13 @@ public void CreateClient_Default_DefaultInvokerSet()
6161
public void CreateClient_Default_PrimaryHandlerIsSocketsHttpHandler()
6262
{
6363
// Arrange
64-
SocketsHttpHandler? socketsHttpHandler = null;
64+
HttpMessageHandler? clientPrimaryHandler = null;
6565
var services = new ServiceCollection();
6666
services
6767
.AddGrpcClient<TestGreeterClient>(o => o.Address = new Uri("http://localhost"))
6868
.ConfigurePrimaryHttpMessageHandler((primaryHandler, _) =>
6969
{
70-
socketsHttpHandler = (SocketsHttpHandler)primaryHandler;
70+
clientPrimaryHandler = primaryHandler;
7171
});
7272

7373
var serviceProvider = services.BuildServiceProvider(validateScopes: true);
@@ -78,8 +78,9 @@ public void CreateClient_Default_PrimaryHandlerIsSocketsHttpHandler()
7878
var client = clientFactory.CreateClient<TestGreeterClient>(nameof(TestGreeterClient));
7979

8080
// Assert
81-
Assert.NotNull(socketsHttpHandler);
82-
Assert.IsTrue(socketsHttpHandler!.EnableMultipleHttp2Connections);
81+
Assert.NotNull(clientPrimaryHandler);
82+
Assert.IsInstanceOf<SocketsHttpHandler>(clientPrimaryHandler);
83+
Assert.IsTrue(((SocketsHttpHandler)clientPrimaryHandler!).EnableMultipleHttp2Connections);
8384
}
8485
#endif
8586

0 commit comments

Comments
 (0)