fix tests

xinlian12 · Dec 4, 2024 · 6fc52b8 · 6fc52b8
1 parent 31bf7af
commit 6fc52b8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/...-cosmos-tests/src/test/java/com/azure/cosmos/implementation/RxDocumentClientImplTest.java b/...-cosmos-tests/src/test/java/com/azure/cosmos/implementation/RxDocumentClientImplTest.java
@@ -10,6 +10,7 @@
 import com.azure.cosmos.CosmosDiagnostics;
 import com.azure.cosmos.CosmosEndToEndOperationLatencyPolicyConfig;
 import com.azure.cosmos.CosmosItemSerializer;
+import com.azure.cosmos.Http2ConnectionConfig;
 import com.azure.cosmos.SessionRetryOptions;
 import com.azure.cosmos.implementation.apachecommons.lang.tuple.ImmutablePair;
 import com.azure.cosmos.implementation.caches.RxClientCollectionCache;
@@ -179,6 +180,7 @@ public void readMany() {
         Mockito.when(this.connectionPolicyMock.getMaxConnectionPoolSize()).thenReturn(dummyInt);
         Mockito.when(this.connectionPolicyMock.getProxy()).thenReturn(dummyProxyOptions);
         Mockito.when(this.connectionPolicyMock.getHttpNetworkRequestTimeout()).thenReturn(dummyDuration);
+        Mockito.when(this.connectionPolicyMock.getHttp2ConnectionConfig()).thenReturn(new Http2ConnectionConfig());
 
         httpClientMock
             .when(() -> HttpClient.createFixed(Mockito.any(HttpClientConfig.class)))

diff --git a/...mos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/http/HttpClientConfig.java b/...mos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/http/HttpClientConfig.java
@@ -172,6 +172,6 @@ public String toDiagnosticsString() {
             maxIdleConnectionTimeout,
             connectionAcquireTimeout,
             proxy != null,
-            http2ConnectionConfig.toDiagnosticsString());
+            http2ConnectionConfig == null ? null : http2ConnectionConfig.toDiagnosticsString());
     }
 }