@Test public void get_methods_delegate_to_the_CEQueueStatus_instance() { assertThat(underTest.getPendingCount()).isEqualTo(PENDING_COUNT); assertThat(underTest.getInProgressCount()).isEqualTo(IN_PROGRESS_COUNT); assertThat(underTest.getErrorCount()).isEqualTo(ERROR_COUNT); assertThat(underTest.getSuccessCount()).isEqualTo(SUCCESS_COUNT); assertThat(underTest.getProcessingTime()).isEqualTo(PROCESSING_TIME); }
@Test public void getWorkerCount_delegates_to_the_CEConfiguration_instance() { assertThat(underTest.getWorkerCount()).isEqualTo(WORKER_COUNT); }
@Test public void getWorkerMaxCount_delegates_to_the_CEConfiguration_instance() { assertThat(underTest.getWorkerMaxCount()).isEqualTo(WORKER_MAX_COUNT); }
@Override public ProtobufSystemInfo.Section toProtobuf() { ProtobufSystemInfo.Section.Builder builder = ProtobufSystemInfo.Section.newBuilder(); builder.setName("Compute Engine Tasks"); builder.addAttributesBuilder().setKey("Pending").setLongValue(getPendingCount()).build(); builder.addAttributesBuilder().setKey("In Progress").setLongValue(getInProgressCount()).build(); builder.addAttributesBuilder().setKey("Processed With Error").setLongValue(getErrorCount()).build(); builder.addAttributesBuilder().setKey("Processed With Success").setLongValue(getSuccessCount()).build(); builder.addAttributesBuilder().setKey("Processing Time (ms)").setLongValue(getProcessingTime()).build(); builder.addAttributesBuilder().setKey("Worker Count").setLongValue(getWorkerCount()).build(); builder.addAttributesBuilder().setKey("Max Worker Count").setLongValue(getWorkerMaxCount()).build(); builder.addAttributesBuilder().setKey("Workers Paused").setBooleanValue(queueStatus.areWorkersPaused()).build(); return builder.build(); } }
@Test public void getWorkerUuids_returns_ordered_list_of_uuids_of_worker_from_CeWorkerFactory_instance() { List<String> workerUuids = underTest.getWorkerUuids(); assertThat(workerUuids).isEqualTo(WORKERS.stream().map(CeWorker::getUUID).sorted().collect(Collectors.toList())); // ImmutableSet can not be serialized assertThat(workerUuids).isNotInstanceOf(ImmutableSet.class); }
@Test public void getEnabledWorkerUuids_returns_ordered_list_of_uuids_of_worker_from_CeWorkerFactory_instance_filtered_on_enabled_ones() { int enabledWorkerCount = new Random().nextInt(WORKERS.size()); int i = 0; CeWorker[] enabledWorkers = new CeWorker[enabledWorkerCount]; for (CeWorker worker : WORKERS) { if (i < enabledWorkerCount) { enabledWorkers[i] = worker; when(ceWorkerController.isEnabled(worker)).thenReturn(true); } else { when(ceWorkerController.isEnabled(worker)).thenReturn(false); } i++; } List<String> enabledWorkerUuids = underTest.getEnabledWorkerUuids(); assertThat(enabledWorkerUuids).isEqualTo(Stream.of(enabledWorkers).map(CeWorker::getUUID).sorted().collect(Collectors.toList())); // ImmutableSet can not be serialized assertThat(enabledWorkerUuids).isNotInstanceOf(ImmutableSet.class); }
@Override public ProtobufSystemInfo.Section toProtobuf() { ProtobufSystemInfo.Section.Builder builder = ProtobufSystemInfo.Section.newBuilder(); builder.setName("Compute Engine Tasks"); builder.addAttributesBuilder().setKey("Pending").setLongValue(getPendingCount()).build(); builder.addAttributesBuilder().setKey("In Progress").setLongValue(getInProgressCount()).build(); builder.addAttributesBuilder().setKey("Processed With Error").setLongValue(getErrorCount()).build(); builder.addAttributesBuilder().setKey("Processed With Success").setLongValue(getSuccessCount()).build(); builder.addAttributesBuilder().setKey("Processing Time (ms)").setLongValue(getProcessingTime()).build(); builder.addAttributesBuilder().setKey("Worker Count").setLongValue(getWorkerCount()).build(); builder.addAttributesBuilder().setKey("Max Worker Count").setLongValue(getWorkerMaxCount()).build(); return builder.build(); } }