Added event on saving User cascade save EmailAddress

spring-data-mongodb/src/test/java/org/baeldung/mongotemplate/DocumentQueryIntegrationTest.java

@@ -7,6 +7,7 @@ import java.util.Iterator;
 import java.util.List;
 
 import org.baeldung.config.MongoConfig;
+import org.baeldung.model.EmailAddress;
 import org.baeldung.model.User;
 import org.junit.After;
 import org.junit.Before;
@@ -31,11 +32,14 @@ public class DocumentQueryIntegrationTest {
 
     @Before
     public void testSetup() {
-        mongoTemplate.createCollection(User.class);
+        if (!mongoTemplate.collectionExists(User.class)) {
+            mongoTemplate.createCollection(User.class);
+        }
     }
 
     @After
     public void tearDown() {
+        mongoTemplate.dropCollection(EmailAddress.class);
         mongoTemplate.dropCollection(User.class);
     }
 
@@ -102,7 +106,7 @@ public class DocumentQueryIntegrationTest {
         query.addCriteria(Criteria.where("name").regex("^A"));
 
         List<User> users = mongoTemplate.find(query, User.class);
-        
+
         assertThat(users.size(), is(2));
     }
 
@@ -127,7 +131,7 @@ public class DocumentQueryIntegrationTest {
         query.addCriteria(Criteria.where("name").regex("c$"));
 
         List<User> users = mongoTemplate.find(query, User.class);
-        
+
         assertThat(users.size(), is(1));
     }
 
@@ -153,7 +157,7 @@ public class DocumentQueryIntegrationTest {
         query.with(pageableRequest);
 
         List<User> users = mongoTemplate.find(query, User.class);
-        
+
         assertThat(users.size(), is(2));
     }
 
@@ -178,11 +182,11 @@ public class DocumentQueryIntegrationTest {
         query.with(new Sort(Sort.Direction.ASC, "age"));
 
         List<User> users = mongoTemplate.find(query, User.class);
-        
+
         Iterator<User> iter = users.iterator();
-        assertThat(users.size(), is(3)); 
+        assertThat(users.size(), is(3));
         assertThat(iter.next().getName(), is("Antony"));
         assertThat(iter.next().getName(), is("Alice"));
-        assertThat(iter.next().getName(), is("Eric"));  
+        assertThat(iter.next().getName(), is("Eric"));
     }
 }

spring-data-mongodb/src/test/java/org/baeldung/mongotemplate/MongoTemplateQueryIntegrationTest.java

@@ -6,6 +6,7 @@ import static org.junit.Assert.assertThat;
 import java.util.List;
 
 import org.baeldung.config.MongoConfig;
+import org.baeldung.model.EmailAddress;
 import org.baeldung.model.User;
 import org.junit.After;
 import org.junit.Before;
@@ -15,7 +16,10 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.data.domain.PageRequest;
 import org.springframework.data.domain.Pageable;
 import org.springframework.data.domain.Sort;
+import org.springframework.data.domain.Sort.Direction;
 import org.springframework.data.mongodb.core.MongoTemplate;
+import org.springframework.data.mongodb.core.index.Index;
+import org.springframework.data.mongodb.core.index.IndexInfo;
 import org.springframework.data.mongodb.core.query.Criteria;
 import org.springframework.data.mongodb.core.query.Query;
 import org.springframework.test.context.ContextConfiguration;
@@ -128,4 +132,42 @@ public class MongoTemplateQueryIntegrationTest {
         List<User> users = mongoTemplate.find(query, User.class);
         assertThat(users.size(), is(3));
     }
+
+    @Test
+    public void givenUserExistsWithIndexAddedFromMapping_whenCheckingIndex_thenIndexIsExisted() {
+        User user = new User();
+        user.setName("Brendan");
+        EmailAddress emailAddress = new EmailAddress();
+        emailAddress.setValue("a@gmail.com");
+        user.setEmailAddress(emailAddress);
+        mongoTemplate.insert(user);
+
+        List<IndexInfo> indexInfos = mongoTemplate.indexOps("user").getIndexInfo();
+
+        assertThat(indexInfos.size(), is(1));
+    }
+
+    @Test
+    public void givenExisted() {
+        User user = new User();
+        user.setName("Brendan");
+        EmailAddress emailAddress = new EmailAddress();
+        emailAddress.setValue("b@gmail.com");
+        user.setEmailAddress(emailAddress);
+        mongoTemplate.insert(user);
+
+        assertThat(mongoTemplate.findOne(Query.query(Criteria.where("name").is("Brendan")), User.class).getEmailAddress().getValue(), is("b@gmail.com"));
+    }
+
+    @Test
+    public void givenUserExistsWithIndexAddedFromCode_whenCheckingIndex_thenIndexIsExisted() {
+        User user = new User();
+        user.setName("Brendan");
+        mongoTemplate.indexOps(User.class).ensureIndex(new Index().on("name", Direction.ASC));
+        mongoTemplate.insert(user);
+
+        List<IndexInfo> indexInfos = mongoTemplate.indexOps("user").getIndexInfo();
+
+        assertThat(indexInfos.size(), is(2));
+    }
 }

spring-data-mongodb/src/test/java/org/baeldung/repository/UserRepositoryIntegrationTest.java

@@ -6,7 +6,6 @@ import static org.junit.Assert.assertThat;
 import java.util.List;
 
 import org.baeldung.config.MongoConfig;
-import org.baeldung.model.EmailAddress;
 import org.baeldung.model.User;
 import org.junit.After;
 import org.junit.Before;
@@ -17,10 +16,7 @@ import org.springframework.data.domain.Page;
 import org.springframework.data.domain.PageRequest;
 import org.springframework.data.domain.Pageable;
 import org.springframework.data.domain.Sort;
-import org.springframework.data.domain.Sort.Direction;
 import org.springframework.data.mongodb.core.MongoOperations;
-import org.springframework.data.mongodb.core.index.Index;
-import org.springframework.data.mongodb.core.index.IndexInfo;
 import org.springframework.data.mongodb.core.query.Criteria;
 import org.springframework.data.mongodb.core.query.Query;
 import org.springframework.test.context.ContextConfiguration;
@@ -149,30 +145,5 @@ public class UserRepositoryIntegrationTest {
         assertThat(users.size(), is(1));
         assertThat(page.getTotalPages(), is(2));
     }
-    
-    @Test
-    public void givenUserExistsWithIndexAddedFromMapping_whenCheckingIndex_thenIndexIsExisted() {
-        User user = new User();
-        user.setName("Brendan");
-        EmailAddress emailAddress = new EmailAddress("a@gmail.com");
-        mongoOps.insert(emailAddress);
-        user.setEmailAddress(emailAddress);
-        mongoOps.insert(user);
-        
-        List<IndexInfo> indexInfos = mongoOps.indexOps("user").getIndexInfo();
 
-        assertThat(indexInfos.size(), is(1));
-    }
-    
-    @Test
-    public void givenUserExistsWithIndexAddedFromCode_whenCheckingIndex_thenIndexIsExisted() {
-        User user = new User();
-        user.setName("Brendan");
-        mongoOps.indexOps(User.class).ensureIndex(new Index().on("name", Direction.ASC));
-        mongoOps.insert(user);
-        
-        List<IndexInfo> indexInfos = mongoOps.indexOps("user").getIndexInfo();
-
-        assertThat(indexInfos.size(), is(2));
-    }
 }