);
PartitionStore partitionStore = Mockito.mock(PartitionStore.class);
Mockito.when(service.getPartitionStore()).thenReturn(partitionStore);
Mockito.when(partitionStore.getPartitionIds()).thenReturn(
Lists.newArrayList(0, 1));
Partition partition = Mockito.mock(Partition.class);
Mockito.when(partition.getVertexCount()).thenReturn(Long.valueOf(1));
Mockito.when(partitionStore.getOrCreatePartition(0)).thenReturn(partition);
Mockito.when(partitionStore.getOrCreatePartition(1)).thenReturn(partition);
}