private String printPartitionRuntimeInfo(PartitionRuntimeInformation runtimeInformation) { if (runtimeInformation == null) { return "[PartitionRuntimeInformation: null]"; } StringBuilder stringBuilder = new StringBuilder(); stringBuilder.append("[PartitionRuntimeInformation:"); stringBuilder.append(" eventHubPath=").append(runtimeInformation.getEventHubPath()); stringBuilder.append(" partitionId=").append(runtimeInformation.getPartitionId()); stringBuilder.append(" lastEnqueuedTimeUtc=").append(runtimeInformation.getLastEnqueuedTimeUtc().toString()); stringBuilder.append(" lastEnqueuedOffset=").append(runtimeInformation.getLastEnqueuedOffset()); // calculate the number of messages in the queue stringBuilder.append(" numMessages=") .append(runtimeInformation.getLastEnqueuedSequenceNumber() - runtimeInformation.getBeginSequenceNumber()); stringBuilder.append("]"); return stringBuilder.toString(); }
@Override public CompletableFuture<PartitionRuntimeInformation> apply(Map<String, Object> rawData) { CompletableFuture<PartitionRuntimeInformation> future2 = new CompletableFuture<PartitionRuntimeInformation>(); future2.complete(new PartitionRuntimeInformation( (String) rawData.get(ClientConstants.MANAGEMENT_ENTITY_NAME_KEY), (String) rawData.get(ClientConstants.MANAGEMENT_PARTITION_NAME_KEY), (long) rawData.get(ClientConstants.MANAGEMENT_RESULT_BEGIN_SEQUENCE_NUMBER), (long) rawData.get(ClientConstants.MANAGEMENT_RESULT_LAST_ENQUEUED_SEQUENCE_NUMBER), (String) rawData.get(ClientConstants.MANAGEMENT_RESULT_LAST_ENQUEUED_OFFSET), ((Date) rawData.get(ClientConstants.MANAGEMENT_RESULT_LAST_ENQUEUED_TIME_UTC)).toInstant(), (boolean) rawData.get(ClientConstants.MANAGEMENT_RESULT_PARTITION_IS_EMPTY))); return future2; } }, this.executor);
String newestOffset = ehPartitionInfo.getLastEnqueuedOffset(); String upcomingOffset = EventHubSystemConsumer.END_OF_STREAM; SystemStreamPartitionMetadata sspMetadata = new SystemStreamPartitionMetadata(startingOffset, newestOffset,
}); PartitionRuntimeInformation mockPartitionRuntimeInfo = PowerMockito.mock(PartitionRuntimeInformation.class); PowerMockito.when(mockPartitionRuntimeInfo.getLastEnqueuedOffset()) .thenReturn(EventHubSystemConsumer.START_OF_STREAM); CompletableFuture<PartitionRuntimeInformation> partitionFuture = new MockPartitionFuture(mockPartitionRuntimeInfo);