Examples with JobSchedulerAdapter org.neo4j.test.scheduler.JobSchedulerAdapter used on opensource projects

Example 1 with JobSchedulerAdapter

use of org.neo4j.test.scheduler.JobSchedulerAdapter in project neo4j by neo4j.

the class BlockBasedIndexPopulatorTest method shouldScheduleMergeOnJobSchedulerWithCorrectGroup.

@Test
void shouldScheduleMergeOnJobSchedulerWithCorrectGroup() throws IndexEntryConflictException, IOException {
    // given
    BlockBasedIndexPopulator<GenericKey, NativeIndexValue> populator = instantiatePopulator(NO_MONITOR);
    boolean closed = false;
    try {
        populator.add(batchOfUpdates(), NULL);
        // when
        MutableBoolean called = new MutableBoolean();
        JobScheduler trackingJobScheduler = new JobSchedulerAdapter() {

            @Override
            public <T> JobHandle<T> schedule(Group group, JobMonitoringParams jobMonitoringParams, Callable<T> job) {
                called.setTrue();
                assertThat(group).isSameAs(Group.INDEX_POPULATION_WORK);
                return jobScheduler.schedule(group, jobMonitoringParams, job);
            }
        };
        populator.scanCompleted(nullInstance, wrapScheduler(trackingJobScheduler), NULL);
        assertTrue(called.booleanValue());
        populator.close(true, NULL);
        closed = true;
    } finally {
        if (!closed) {
            populator.close(true, NULL);
        }
    }
}