/** * Validates and builds a {@link org.springframework.batch.item.data.Neo4jItemWriter}. * * @return a {@link Neo4jItemWriter} */ public Neo4jItemWriter<T> build() { Assert.notNull(sessionFactory, "sessionFactory is required."); Neo4jItemWriter<T> writer = new Neo4jItemWriter<>(); writer.setDelete(this.delete); writer.setSessionFactory(this.sessionFactory); return writer; } }
@Test public void testDeleteItemsWithSession() throws Exception { writer = new Neo4jItemWriter<>(); writer.setSessionFactory(this.sessionFactory); writer.afterPropertiesSet(); List<String> items = new ArrayList<>(); items.add("foo"); items.add("bar"); writer.setDelete(true); when(this.sessionFactory.openSession()).thenReturn(this.session); writer.write(items); verify(this.session).delete("foo"); verify(this.session).delete("bar"); } }
/** * Validates and builds a {@link org.springframework.batch.item.data.Neo4jItemWriter}. * * @return a {@link Neo4jItemWriter} */ public Neo4jItemWriter<T> build() { Assert.notNull(sessionFactory, "sessionFactory is required."); Neo4jItemWriter<T> writer = new Neo4jItemWriter<>(); writer.setDelete(this.delete); writer.setSessionFactory(this.sessionFactory); return writer; } }