DATAGRAPH-285 adding application events for save and delete

spring-data-neo4j/src/main/java/org/springframework/data/neo4j/lifecycle/AfterSaveEvent.java

@@ -0,0 +1,38 @@
+/**
+ * Copyright 2011 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.data.neo4j.lifecycle;
+
+import org.springframework.context.ApplicationEvent;
+
+public class AfterSaveEvent<T> extends ApplicationEvent {
+    private final T entity;
+
+    /**
+     * Create a new ApplicationEvent.
+     *
+     * @param source the component that published the event (never <code>null</code>)
+     * @param entity
+     */
+    public AfterSaveEvent(Object source, T entity) {
+        super(source);
+
+        this.entity = entity;
+    }
+
+    public T getEntity() {
+        return entity;
+    }
+}

spring-data-neo4j/src/main/java/org/springframework/data/neo4j/lifecycle/BeforeSaveEvent.java

@@ -0,0 +1,36 @@
+/**
+ * Copyright 2011 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.data.neo4j.lifecycle;
+
+import org.springframework.context.ApplicationEvent;
+
+public class BeforeSaveEvent<T> extends ApplicationEvent {
+    private final T entity;
+
+    /**
+     * @param source the component that published the event (never <code>null</code>)
+     * @param entity the entity that is about to be saved
+     */
+    public BeforeSaveEvent(Object source, T entity) {
+        super(source);
+
+        this.entity = entity;
+    }
+
+    public T getEntity() {
+        return entity;
+    }
+}

spring-data-neo4j/src/main/java/org/springframework/data/neo4j/lifecycle/DeleteEvent.java

@@ -0,0 +1,32 @@
+/**
+ * Copyright 2011 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.data.neo4j.lifecycle;
+
+import org.springframework.context.ApplicationEvent;
+
+public class DeleteEvent<T> extends ApplicationEvent {
+    private final T entity;
+
+    public DeleteEvent(Object source, T entity) {
+        super(source);
+
+        this.entity = entity;
+    }
+
+    public T getEntity() {
+        return entity;
+    }
+}

spring-data-neo4j/src/test/java/org/springframework/data/neo4j/lifecycle/AfterSaveEventTests.java

@@ -0,0 +1,81 @@
+/**
+ * Copyright 2011 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.data.neo4j.lifecycle;
+
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.neo4j.graphdb.GraphDatabaseService;
+import org.neo4j.test.ImpermanentGraphDatabase;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.ApplicationListener;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.data.neo4j.annotation.GraphId;
+import org.springframework.data.neo4j.annotation.NodeEntity;
+import org.springframework.data.neo4j.config.EnableNeo4jRepositories;
+import org.springframework.data.neo4j.config.Neo4jConfiguration;
+import org.springframework.data.neo4j.support.Neo4jTemplate;
+import org.springframework.test.context.ContextConfiguration;
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
+import org.springframework.transaction.annotation.Transactional;
+
+import static org.hamcrest.CoreMatchers.is;
+import static org.junit.Assert.assertThat;
+import static org.neo4j.helpers.collection.IteratorUtil.single;
+
+@NodeEntity
+class Bar {
+    @GraphId
+    Long id;
+
+    String generatedId = "no event";
+}
+
+@RunWith(SpringJUnit4ClassRunner.class)
+@ContextConfiguration
+@Transactional
+public class AfterSaveEventTests {
+    @Configuration
+    @EnableNeo4jRepositories
+    static class TestConfig extends Neo4jConfiguration {
+        @Bean
+        GraphDatabaseService graphDatabaseService() {
+            return new ImpermanentGraphDatabase();
+        }
+
+        @Bean
+        ApplicationListener<AfterSaveEvent> beforeSaveEventApplicationListener() {
+            return new ApplicationListener<AfterSaveEvent>() {
+                @Override
+                public void onApplicationEvent(AfterSaveEvent event) {
+                    Bar bar = (Bar) event.getEntity();
+                    bar.generatedId = "after event";
+                }
+            };
+        }
+    }
+
+    @Autowired
+    private Neo4jTemplate template;
+
+    @Test
+    public void shouldFireAfterEntityIsSaved() throws Exception {
+        Bar entity = new Bar();
+        assertThat(template.save(entity).generatedId, is("no event"));
+        assertThat(entity.generatedId, is("after event"));
+        assertThat(single(template.findAll(Bar.class)).generatedId, is("no event"));
+    }
+}

spring-data-neo4j/src/test/java/org/springframework/data/neo4j/lifecycle/BeforeSaveEventTests.java

@@ -0,0 +1,79 @@
+/**
+ * Copyright 2011 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.data.neo4j.lifecycle;
+
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.neo4j.graphdb.GraphDatabaseService;
+import org.neo4j.test.ImpermanentGraphDatabase;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.ApplicationListener;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.data.neo4j.annotation.GraphId;
+import org.springframework.data.neo4j.annotation.NodeEntity;
+import org.springframework.data.neo4j.config.EnableNeo4jRepositories;
+import org.springframework.data.neo4j.config.Neo4jConfiguration;
+import org.springframework.data.neo4j.support.Neo4jTemplate;
+import org.springframework.test.context.ContextConfiguration;
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
+import org.springframework.transaction.annotation.Transactional;
+
+import static org.hamcrest.CoreMatchers.is;
+import static org.junit.Assert.assertThat;
+import static org.neo4j.helpers.collection.IteratorUtil.single;
+
+@NodeEntity
+class Foo {
+    @GraphId
+    Long id;
+
+    String generatedId = "no event";
+}
+
+@RunWith(SpringJUnit4ClassRunner.class)
+@ContextConfiguration
+@Transactional
+public class BeforeSaveEventTests {
+    @Configuration
+    @EnableNeo4jRepositories
+    static class TestConfig extends Neo4jConfiguration {
+        @Bean
+        GraphDatabaseService graphDatabaseService() {
+            return new ImpermanentGraphDatabase();
+        }
+
+        @Bean
+        ApplicationListener<BeforeSaveEvent> beforeSaveEventApplicationListener() {
+            return new ApplicationListener<BeforeSaveEvent>() {
+                @Override
+                public void onApplicationEvent(BeforeSaveEvent event) {
+                    Foo foo = (Foo) event.getEntity();
+                    foo.generatedId = "before event";
+                }
+            };
+        }
+    }
+
+    @Autowired
+    private Neo4jTemplate template;
+
+    @Test
+    public void shouldFireBeforeEntityIsSaved() throws Exception {
+        assertThat(template.save(new Foo()).generatedId, is("before event"));
+        assertThat(single(template.findAll(Foo.class)).generatedId, is("before event"));
+    }
+}

spring-data-neo4j/src/test/java/org/springframework/data/neo4j/lifecycle/DeleteEventTests.java

@@ -0,0 +1,107 @@
+/**
+ * Copyright 2011 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.springframework.data.neo4j.lifecycle;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.neo4j.graphdb.GraphDatabaseService;
+import org.neo4j.test.ImpermanentGraphDatabase;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.ApplicationListener;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.data.neo4j.annotation.GraphId;
+import org.springframework.data.neo4j.annotation.NodeEntity;
+import org.springframework.data.neo4j.config.EnableNeo4jRepositories;
+import org.springframework.data.neo4j.config.Neo4jConfiguration;
+import org.springframework.data.neo4j.support.Neo4jTemplate;
+import org.springframework.data.neo4j.support.node.Neo4jHelper;
+import org.springframework.test.context.ContextConfiguration;
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
+import org.springframework.test.context.transaction.BeforeTransaction;
+import org.springframework.transaction.annotation.Transactional;
+
+import java.util.LinkedList;
+
+import static org.junit.Assert.assertThat;
+import static org.junit.matchers.JUnitMatchers.hasItem;
+
+@NodeEntity
+class Program {
+    @GraphId
+    Long id;
+
+    String name;
+
+    Program() {
+    }
+
+    public Program(String name) {
+        this.name = name;
+    }
+}
+
+@RunWith(SpringJUnit4ClassRunner.class)
+@ContextConfiguration
+@Transactional
+public class DeleteEventTests {
+    @Configuration
+    @EnableNeo4jRepositories
+    static class TestConfig extends Neo4jConfiguration {
+        @Bean
+        GraphDatabaseService graphDatabaseService() {
+            return new ImpermanentGraphDatabase();
+        }
+
+        @Bean
+        ApplicationListener<DeleteEvent<Program>> deleteEventApplicationListener() {
+            return new ApplicationListener<DeleteEvent<Program>>() {
+                @Override
+                public void onApplicationEvent(DeleteEvent<Program> event) {
+                    deletions.add(event.getEntity());
+                }
+            };
+        }
+    }
+
+    @Autowired
+    Neo4jTemplate template;
+
+    @Autowired
+    GraphDatabaseService graphDatabaseService;
+
+    static final LinkedList<Program> deletions = new LinkedList<Program>();
+
+    @BeforeTransaction
+    public void beforeTransaction() {
+        Neo4jHelper.cleanDb(template);
+    }
+
+    @Before
+    public void before() {
+        deletions.clear();
+    }
+
+    @Test
+    public void shouldFireEventOnNodeDeletion() throws Exception {
+        Program sark = template.save(new Program("Sark"));
+
+        template.delete(sark);
+
+        assertThat(deletions, hasItem(sark));
+    }
+}