|
14 | 14 | import com.azure.cosmos.implementation.ResourceType; |
15 | 15 | import com.azure.cosmos.implementation.RxDocumentServiceRequest; |
16 | 16 | import com.azure.cosmos.implementation.SerializationDiagnosticsContext; |
17 | | -import com.azure.cosmos.implementation.apachecommons.collections.list.UnmodifiableList; |
18 | 17 | import com.azure.cosmos.implementation.perPartitionAutomaticFailover.PerPartitionAutomaticFailoverInfoHolder; |
19 | 18 | import com.azure.cosmos.implementation.perPartitionCircuitBreaker.GlobalPartitionEndpointManagerForPerPartitionCircuitBreaker; |
20 | 19 | import com.azure.cosmos.implementation.perPartitionCircuitBreaker.LocationHealthStatus; |
@@ -214,8 +213,8 @@ public void recordHealthyToHealthyWithFailuresStatusTransition(String partitionL |
214 | 213 | maxExclusive, |
215 | 214 | LocationEastUs2EndpointToLocationPair.getKey()); |
216 | 215 |
|
217 | | - Mockito.when(this.globalEndpointManagerMock.getReadEndpoints()).thenReturn((UnmodifiableList<RegionalRoutingContext>) UnmodifiableList.unmodifiableList(applicableReadWriteEndpoints)); |
218 | | - Mockito.when(this.globalEndpointManagerMock.getWriteEndpoints()).thenReturn((UnmodifiableList<RegionalRoutingContext>) UnmodifiableList.unmodifiableList(applicableReadWriteEndpoints)); |
| 216 | + Mockito.when(this.globalEndpointManagerMock.getReadEndpoints()).thenReturn(Collections.unmodifiableList(applicableReadWriteEndpoints)); |
| 217 | + Mockito.when(this.globalEndpointManagerMock.getWriteEndpoints()).thenReturn(Collections.unmodifiableList(applicableReadWriteEndpoints)); |
219 | 218 |
|
220 | 219 | globalPartitionEndpointManagerForCircuitBreaker |
221 | 220 | .handleLocationExceptionForPartitionKeyRange(request, new RegionalRoutingContext(LocationEastUs2EndpointToLocationPair.getKey()), false); |
@@ -284,8 +283,8 @@ public void recordHealthyWithFailuresToUnavailableStatusTransition(String partit |
284 | 283 | maxExclusive, |
285 | 284 | LocationEastUs2EndpointToLocationPair.getKey()); |
286 | 285 |
|
287 | | - Mockito.when(this.globalEndpointManagerMock.getApplicableWriteRegionalRoutingContexts(Mockito.anyList())).thenReturn((UnmodifiableList<RegionalRoutingContext>) UnmodifiableList.unmodifiableList(applicableReadWriteEndpoints)); |
288 | | - Mockito.when(this.globalEndpointManagerMock.getApplicableReadRegionalRoutingContexts(Mockito.anyList())).thenReturn((UnmodifiableList<RegionalRoutingContext>) UnmodifiableList.unmodifiableList(applicableReadWriteEndpoints)); |
| 286 | + Mockito.when(this.globalEndpointManagerMock.getApplicableWriteRegionalRoutingContexts(Mockito.anyList())).thenReturn(Collections.unmodifiableList(applicableReadWriteEndpoints)); |
| 287 | + Mockito.when(this.globalEndpointManagerMock.getApplicableReadRegionalRoutingContexts(Mockito.anyList())).thenReturn(Collections.unmodifiableList(applicableReadWriteEndpoints)); |
289 | 288 |
|
290 | 289 | int exceptionCountToHandle |
291 | 290 | = globalPartitionEndpointManagerForCircuitBreaker.getConsecutiveExceptionBasedCircuitBreaker().getAllowedExceptionCountToMaintainStatus(LocationHealthStatus.HealthyWithFailures, readOperationTrue); |
@@ -361,8 +360,8 @@ public void recordUnavailableToHealthyTentativeStatusTransition(String partition |
361 | 360 | maxExclusive, |
362 | 361 | LocationEastUs2EndpointToLocationPair.getKey()); |
363 | 362 |
|
364 | | - Mockito.when(this.globalEndpointManagerMock.getApplicableWriteRegionalRoutingContexts(Mockito.anyList())).thenReturn((UnmodifiableList<RegionalRoutingContext>) UnmodifiableList.unmodifiableList(applicableReadWriteEndpoints)); |
365 | | - Mockito.when(this.globalEndpointManagerMock.getApplicableReadRegionalRoutingContexts(Mockito.anyList())).thenReturn((UnmodifiableList<RegionalRoutingContext>) UnmodifiableList.unmodifiableList(applicableReadWriteEndpoints)); |
| 363 | + Mockito.when(this.globalEndpointManagerMock.getApplicableWriteRegionalRoutingContexts(Mockito.anyList())).thenReturn(Collections.unmodifiableList(applicableReadWriteEndpoints)); |
| 364 | + Mockito.when(this.globalEndpointManagerMock.getApplicableReadRegionalRoutingContexts(Mockito.anyList())).thenReturn(Collections.unmodifiableList(applicableReadWriteEndpoints)); |
366 | 365 |
|
367 | 366 | int exceptionCountToHandle |
368 | 367 | = globalPartitionEndpointManagerForCircuitBreaker.getConsecutiveExceptionBasedCircuitBreaker().getAllowedExceptionCountToMaintainStatus(LocationHealthStatus.HealthyWithFailures, readOperationTrue); |
@@ -449,8 +448,8 @@ public void recordHealthyTentativeToHealthyStatusTransition(String partitionLeve |
449 | 448 | maxExclusive, |
450 | 449 | LocationEastUs2EndpointToLocationPair.getKey()); |
451 | 450 |
|
452 | | - Mockito.when(this.globalEndpointManagerMock.getApplicableWriteRegionalRoutingContexts(Mockito.anyList())).thenReturn((UnmodifiableList<RegionalRoutingContext>) UnmodifiableList.unmodifiableList(applicableReadWriteEndpoints)); |
453 | | - Mockito.when(this.globalEndpointManagerMock.getApplicableReadRegionalRoutingContexts(Mockito.anyList())).thenReturn((UnmodifiableList<RegionalRoutingContext>) UnmodifiableList.unmodifiableList(applicableReadWriteEndpoints)); |
| 451 | + Mockito.when(this.globalEndpointManagerMock.getApplicableWriteRegionalRoutingContexts(Mockito.anyList())).thenReturn(Collections.unmodifiableList(applicableReadWriteEndpoints)); |
| 452 | + Mockito.when(this.globalEndpointManagerMock.getApplicableReadRegionalRoutingContexts(Mockito.anyList())).thenReturn(Collections.unmodifiableList(applicableReadWriteEndpoints)); |
454 | 453 |
|
455 | 454 | int exceptionCountToHandle |
456 | 455 | = globalPartitionEndpointManagerForCircuitBreaker.getConsecutiveExceptionBasedCircuitBreaker().getAllowedExceptionCountToMaintainStatus(LocationHealthStatus.HealthyWithFailures, readOperationTrue); |
@@ -545,8 +544,8 @@ public void recordHealthyTentativeToUnavailableTransition(String partitionLevelC |
545 | 544 | maxExclusive, |
546 | 545 | LocationEastUs2EndpointToLocationPair.getKey()); |
547 | 546 |
|
548 | | - Mockito.when(this.globalEndpointManagerMock.getApplicableWriteRegionalRoutingContexts(Mockito.anyList())).thenReturn((UnmodifiableList<RegionalRoutingContext>) UnmodifiableList.unmodifiableList(applicableReadWriteEndpoints)); |
549 | | - Mockito.when(this.globalEndpointManagerMock.getApplicableReadRegionalRoutingContexts(Mockito.anyList())).thenReturn((UnmodifiableList<RegionalRoutingContext>) UnmodifiableList.unmodifiableList(applicableReadWriteEndpoints)); |
| 547 | + Mockito.when(this.globalEndpointManagerMock.getApplicableWriteRegionalRoutingContexts(Mockito.anyList())).thenReturn(Collections.unmodifiableList(applicableReadWriteEndpoints)); |
| 548 | + Mockito.when(this.globalEndpointManagerMock.getApplicableReadRegionalRoutingContexts(Mockito.anyList())).thenReturn(Collections.unmodifiableList(applicableReadWriteEndpoints)); |
550 | 549 |
|
551 | 550 | int exceptionCountToHandle |
552 | 551 | = globalPartitionEndpointManagerForCircuitBreaker.getConsecutiveExceptionBasedCircuitBreaker().getAllowedExceptionCountToMaintainStatus(LocationHealthStatus.HealthyWithFailures, readOperationTrue); |
@@ -639,8 +638,8 @@ public void allRegionsUnavailableHandling(String partitionLevelCircuitBreakerCon |
639 | 638 | maxExclusive, |
640 | 639 | LocationEastUs2EndpointToLocationPair.getKey()); |
641 | 640 |
|
642 | | - Mockito.when(this.globalEndpointManagerMock.getApplicableWriteRegionalRoutingContexts(Mockito.anyList())).thenReturn((UnmodifiableList<RegionalRoutingContext>) UnmodifiableList.unmodifiableList(applicableReadWriteEndpoints)); |
643 | | - Mockito.when(this.globalEndpointManagerMock.getApplicableReadRegionalRoutingContexts(Mockito.anyList())).thenReturn((UnmodifiableList<RegionalRoutingContext>) UnmodifiableList.unmodifiableList(applicableReadWriteEndpoints)); |
| 641 | + Mockito.when(this.globalEndpointManagerMock.getApplicableWriteRegionalRoutingContexts(Mockito.anyList())).thenReturn(Collections.unmodifiableList(applicableReadWriteEndpoints)); |
| 642 | + Mockito.when(this.globalEndpointManagerMock.getApplicableReadRegionalRoutingContexts(Mockito.anyList())).thenReturn(Collections.unmodifiableList(applicableReadWriteEndpoints)); |
644 | 643 |
|
645 | 644 | int exceptionCountToHandle |
646 | 645 | = globalPartitionEndpointManagerForCircuitBreaker |
@@ -724,8 +723,8 @@ public void multiContainerBothWithSinglePartitionHealthyToUnavailableHandling(St |
724 | 723 | maxExclusive, |
725 | 724 | LocationEastUs2EndpointToLocationPair.getKey()); |
726 | 725 |
|
727 | | - Mockito.when(this.globalEndpointManagerMock.getApplicableWriteRegionalRoutingContexts(Mockito.anyList())).thenReturn((UnmodifiableList<RegionalRoutingContext>) UnmodifiableList.unmodifiableList(applicableReadWriteEndpoints)); |
728 | | - Mockito.when(this.globalEndpointManagerMock.getApplicableReadRegionalRoutingContexts(Mockito.anyList())).thenReturn((UnmodifiableList<RegionalRoutingContext>) UnmodifiableList.unmodifiableList(applicableReadWriteEndpoints)); |
| 726 | + Mockito.when(this.globalEndpointManagerMock.getApplicableWriteRegionalRoutingContexts(Mockito.anyList())).thenReturn(Collections.unmodifiableList(applicableReadWriteEndpoints)); |
| 727 | + Mockito.when(this.globalEndpointManagerMock.getApplicableReadRegionalRoutingContexts(Mockito.anyList())).thenReturn(Collections.unmodifiableList(applicableReadWriteEndpoints)); |
729 | 728 |
|
730 | 729 | int exceptionCountToHandle |
731 | 730 | = globalPartitionEndpointManagerForCircuitBreaker.getConsecutiveExceptionBasedCircuitBreaker().getAllowedExceptionCountToMaintainStatus(LocationHealthStatus.HealthyWithFailures, readOperationTrue); |
@@ -821,8 +820,8 @@ public void allRegionsUnavailableHandlingWithMultiThreading(String partitionLeve |
821 | 820 | .map(uriStringPair -> new RegionalRoutingContext(uriStringPair.getLeft())) |
822 | 821 | .collect(Collectors.toList()); |
823 | 822 |
|
824 | | - Mockito.when(this.globalEndpointManagerMock.getApplicableWriteRegionalRoutingContexts(Mockito.anyList())).thenReturn((UnmodifiableList<RegionalRoutingContext>) UnmodifiableList.unmodifiableList(applicableReadWriteEndpoints)); |
825 | | - Mockito.when(this.globalEndpointManagerMock.getApplicableReadRegionalRoutingContexts(Mockito.anyList())).thenReturn((UnmodifiableList<RegionalRoutingContext>) UnmodifiableList.unmodifiableList(applicableReadWriteEndpoints)); |
| 823 | + Mockito.when(this.globalEndpointManagerMock.getApplicableWriteRegionalRoutingContexts(Mockito.anyList())).thenReturn(Collections.unmodifiableList(applicableReadWriteEndpoints)); |
| 824 | + Mockito.when(this.globalEndpointManagerMock.getApplicableReadRegionalRoutingContexts(Mockito.anyList())).thenReturn(Collections.unmodifiableList(applicableReadWriteEndpoints)); |
826 | 825 |
|
827 | 826 | RxDocumentServiceRequest requestCentralUs = constructRxDocumentServiceRequestInstance( |
828 | 827 | readOperationTrue ? OperationType.Read : OperationType.Create, |
|
0 commit comments