diff --git a/spring-data-mongodb/src/test/java/org/baeldung/mongotemplate/MongoTemplateIntegrationTest.java b/spring-data-mongodb/src/test/java/org/baeldung/mongotemplate/MongoTemplateIntegrationTest.java index 9a9e5b87f8..8330b8d0b3 100644 --- a/spring-data-mongodb/src/test/java/org/baeldung/mongotemplate/MongoTemplateIntegrationTest.java +++ b/spring-data-mongodb/src/test/java/org/baeldung/mongotemplate/MongoTemplateIntegrationTest.java @@ -41,6 +41,13 @@ public class MongoTemplateIntegrationTest { mongoTemplate.insert(user); } + @Test + public void whenSavingNewUser_thenUserIsInserted() { + final User user = new User(); + user.setName("Aaron"); + mongoTemplate.save(user); + } + @Test public void givenUserExists_whenSavingExistUser_thenUserIsUpdated() { User user = new User(); diff --git a/spring-data-mongodb/src/test/java/org/baeldung/repository/UserRepositoryIntegrationTest.java b/spring-data-mongodb/src/test/java/org/baeldung/repository/UserRepositoryIntegrationTest.java index b1ef3f1f82..009191f03e 100644 --- a/spring-data-mongodb/src/test/java/org/baeldung/repository/UserRepositoryIntegrationTest.java +++ b/spring-data-mongodb/src/test/java/org/baeldung/repository/UserRepositoryIntegrationTest.java @@ -51,6 +51,15 @@ public class UserRepositoryIntegrationTest { assertThat(mongoOps.findOne(Query.query(Criteria.where("name").is("Jon")), User.class).getName(), is("Jon")); } + @Test + public void whenSavingNewUser_thenUserIsInserted() { + final User user = new User(); + user.setName("Albert"); + userRepository.save(user); + + assertThat(mongoOps.findOne(Query.query(Criteria.where("name").is("Albert")), User.class).getName(), is("Albert")); + } + @Test public void givenUserExists_whenSavingExistUser_thenUserIsUpdated() { User user = new User(); @@ -128,11 +137,12 @@ public class UserRepositoryIntegrationTest { user.setName("Adam"); mongoOps.insert(user); - final Pageable pageableRequest = new PageRequest(0, 2); + final Pageable pageableRequest = new PageRequest(0, 1); - final Page users = userRepository.findAll(pageableRequest); + final Page page = userRepository.findAll(pageableRequest); + List users = page.getContent(); - assertThat(users.getTotalPages(), is(1)); - assertThat(users.iterator().next().getName(), is("Brendan")); + assertThat(users.size(), is(1)); + assertThat(page.getTotalPages(), is(2)); } }