/** * Validates and builds a {@link MongoItemWriter}. * * @return a {@link MongoItemWriter} */ public MongoItemWriter<T> build() { Assert.notNull(this.template, "template is required."); MongoItemWriter<T> writer = new MongoItemWriter<>(); writer.setTemplate(this.template); writer.setDelete(this.delete); writer.setCollection(this.collection); return writer; }
@Test public void testRemoveNoTransactionNoCollection() throws Exception { writer.setDelete(true); List<Object> items = new ArrayList<Object>() {{ add(new Object()); add(new Object()); }}; writer.write(items); verify(template).remove(items.get(0)); verify(template).remove(items.get(1)); }
@Test public void testRemoveNoTransactionWithCollection() throws Exception { writer.setDelete(true); List<Object> items = new ArrayList<Object>() {{ add(new Object()); add(new Object()); }}; writer.setCollection("collection"); writer.write(items); verify(template).remove(items.get(0), "collection"); verify(template).remove(items.get(1), "collection"); }
/** * Validates and builds a {@link MongoItemWriter}. * * @return a {@link MongoItemWriter} */ public MongoItemWriter<T> build() { Assert.notNull(this.template, "template is required."); MongoItemWriter<T> writer = new MongoItemWriter<>(); writer.setTemplate(this.template); writer.setDelete(this.delete); writer.setCollection(this.collection); return writer; }