use of org.neo4j.ogm.domain.gh576.DataItem in project neo4j-ogm by neo4j.
the class CypherContextTest method shouldDeregisterRelationshipEntities.
// GH-576
@Test
public void shouldDeregisterRelationshipEntities() {
Collection<DataItem> dataItems;
FormulaItem formulaItem;
Filter filter = new Filter("nodeId", ComparisonOperator.EQUALS, "m1");
dataItems = session.loadAll(DataItem.class, filter);
assertThat(dataItems).hasSize(1);
formulaItem = (FormulaItem) dataItems.iterator().next();
assertThat(formulaItem.getVariables()).hasSize(3);
Predicate<Variable> isVariableAWithDataItemM2 = v -> v.getVariable().equals("A") && v.getDataItem().getNodeId().equals("m2");
formulaItem.getVariables().removeIf(isVariableAWithDataItemM2);
assertThat(formulaItem.getVariables()).hasSize(2);
session.save(formulaItem);
dataItems = session.loadAll(DataItem.class, filter);
assertThat(dataItems).hasSize(1);
formulaItem = (FormulaItem) dataItems.iterator().next();
assertThat(formulaItem.getVariables()).hasSize(2);
}
Aggregations