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..6e0a666fbe 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 @@ -40,6 +40,13 @@ public class MongoTemplateIntegrationTest { user.setName("Jon"); 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() { 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..5744f47cb7 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 @@ -50,6 +50,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() { @@ -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); - - assertThat(users.getTotalPages(), is(1)); - assertThat(users.iterator().next().getName(), is("Brendan")); + final Page page = userRepository.findAll(pageableRequest); + List users = page.getContent(); + + assertThat(users.size(), is(1)); + assertThat(page.getTotalPages(), is(2)); } }