@Override
public PrimitiveIntIterator nodeGetLabels(KernelStatement state, long nodeId)
throws EntityNotFoundException {
if (state.hasTxStateWithChanges()) {
return nodeGetLabels(storeLayer, state.txState(), nodeId);
}
return storeLayer.nodeGetLabels(nodeId);
}
@Test
public void correctlySaysNodeIsDeleted() throws Exception {
// Given
state.nodeDoDelete(1l);
Node node = mock(Node.class);
when(node.getId()).thenReturn(1l);
when(ops.nodeGetAllProperties(1l)).thenReturn(IteratorUtil.<DefinedProperty>emptyIterator());
when(ops.nodeGetLabels(1l)).thenReturn(PrimitiveIntCollections.emptyIterator());
// When & Then
assertThat(snapshot().isDeleted(node), equalTo(true));
}
public static PrimitiveIntIterator nodeGetLabels(
StoreReadLayer storeLayer, ReadableTxState txState, long nodeId)
throws EntityNotFoundException {
if (txState.nodeIsDeletedInThisTx(nodeId)) {
return PrimitiveIntCollections.emptyIterator();
}
if (txState.nodeIsAddedInThisTx(nodeId)) {
return PrimitiveIntCollections.toPrimitiveIterator(
txState.nodeStateLabelDiffSets(nodeId).getAdded().iterator());
}
return txState.nodeStateLabelDiffSets(nodeId).augment(storeLayer.nodeGetLabels(nodeId));
}
@Override
public PrimitiveIntIterator nodeGetLabels(KernelStatement state, long nodeId)
throws EntityNotFoundException {
if (state.hasTxStateWithChanges()) {
if (state.txState().nodeIsDeletedInThisTx(nodeId)) {
return PrimitiveIntCollections.emptyIterator();
}
if (state.txState().nodeIsAddedInThisTx(nodeId)) {
return PrimitiveIntCollections.toPrimitiveIterator(
state.txState().nodeStateLabelDiffSets(nodeId).getAdded().iterator());
}
return state
.txState()
.nodeStateLabelDiffSets(nodeId)
.augment(storeLayer.nodeGetLabels(nodeId));
}
return storeLayer.nodeGetLabels(nodeId);
}
@Test
public void shouldListAddedLabels() throws Exception {
// Given
state.nodeDoAddLabel(2, 1l);
when(ops.labelGetName(2)).thenReturn("theLabel");
when(ops.nodeGetLabels(1l)).thenReturn(PrimitiveIntCollections.emptyIterator());
// When
Iterable<LabelEntry> labelEntries = snapshot().assignedLabels();
// Then
LabelEntry entry = single(labelEntries);
assertThat(entry.label().name(), equalTo("theLabel"));
assertThat(entry.node().getId(), equalTo(1l));
}
@Test
public void showsDeletedNodes() throws Exception {
// Given
state.nodeDoDelete(1l);
state.nodeDoDelete(2l);
when(ops.nodeGetAllProperties(1l)).thenReturn(IteratorUtil.<DefinedProperty>emptyIterator());
when(ops.nodeGetAllProperties(2l))
.thenReturn(asList(Property.stringProperty(1, "p")).iterator());
when(ops.nodeGetLabels(1l)).thenReturn(PrimitiveIntCollections.emptyIterator());
when(ops.nodeGetLabels(2l)).thenReturn(PrimitiveIntCollections.iterator(15));
when(ops.propertyKeyGetName(1)).thenReturn("key");
when(ops.labelGetName(15)).thenReturn("label");
// When & Then
TxStateTransactionDataSnapshot snapshot = snapshot();
assertThat(idList(snapshot.deletedNodes()), equalTo(asList(1l, 2l)));
assertThat(single(snapshot.removedLabels()).label().name(), equalTo("label"));
assertThat(single(snapshot.removedNodeProperties()).key(), equalTo("key"));
}