From 5e7f4ab84f38469cb5cfef31e5357a1a697bc6bb Mon Sep 17 00:00:00 2001
From: coach88 <alexsemenyuk88@gmail.com>
Date: Sat, 1 Aug 2015 23:49:32 +0300
Subject: [PATCH] Added tests with save as insert and as update. Fixed test
 with FindAll with Pageable

---
 .../MongoTemplateIntegrationTest.java         |  7 +++++++
 .../UserRepositoryIntegrationTest.java        | 20 ++++++++++++++-----
 2 files changed, 22 insertions(+), 5 deletions(-)

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<User> users = userRepository.findAll(pageableRequest);
-
-        assertThat(users.getTotalPages(), is(1));
-        assertThat(users.iterator().next().getName(), is("Brendan"));
+        final Page<User> page = userRepository.findAll(pageableRequest);
+        List<User> users = page.getContent();
+       
+        assertThat(users.size(), is(1));
+        assertThat(page.getTotalPages(), is(2));
     }
 }