@Test(groups = { "unit" }, timeOut = TIMEOUT)

public void shouldRetryWithPartitionIsMigratingException() {
    RxDocumentServiceRequest request = RxDocumentServiceRequest.create(OperationType.Read, ResourceType.Document);
    GoneAndRetryWithRetryPolicy goneAndRetryWithRetryPolicy = new GoneAndRetryWithRetryPolicy(request, 30);
    Single<IRetryPolicy.ShouldRetryResult> singleShouldRetry = goneAndRetryWithRetryPolicy
            .shouldRetry(new PartitionIsMigratingException());
    IRetryPolicy.ShouldRetryResult shouldRetryResult = singleShouldRetry.toBlocking().value();
    assertThat(shouldRetryResult.shouldRetry).isTrue();
    assertThat(request.forceCollectionRoutingMapRefresh).isTrue();
    assertThat(shouldRetryResult.policyArg.getValue0()).isTrue();
}