1 files changed, 16 insertions, 1 deletions
diff --git a/juick-server-jdbc/src/test/java/com/juick/service/MessageServiceTest.java b/juick-server-jdbc/src/test/java/com/juick/service/MessageServiceTest.java
index fed5dad8..fae938cc 100644
--- a/juick-server-jdbc/src/test/java/com/juick/service/MessageServiceTest.java
+++ b/juick-server-jdbc/src/test/java/com/juick/service/MessageServiceTest.java
@@ -24,6 +24,9 @@ import org.springframework.test.context.junit4.AbstractJUnit4SpringContextTests;
 
 import javax.inject.Inject;
 
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.Matchers.equalTo;
+
 /**
  * Created by aalexeev on 11/25/16.
  */
@@ -31,10 +34,22 @@ import javax.inject.Inject;
 public class MessageServiceTest extends AbstractJUnit4SpringContextTests {
     @Inject
     private MessagesService messagesService;
-
+    @Inject
+    private UserService userService;
 
     @Test
     public void getMyFeed() {
         messagesService.getMyFeed(1, 1000000);
     }
+    @Test
+    public void likesCountShouldMatch() {
+        int uid = userService.createUser("ugnich", "secret");
+        int mid = messagesService.createMessage(uid, "test", null, null);
+        int mid2 = messagesService.createMessage(uid, "test2", null, null);
+        assertThat(messagesService.recommendMessage(mid, uid), equalTo(true));
+        assertThat(messagesService.getMessage(mid).getLikes(), equalTo(1));
+        assertThat(messagesService.recommendMessage(mid, uid), equalTo(false));
+        assertThat(messagesService.getMessage(mid).getLikes(), equalTo(0));
+        assertThat(messagesService.getAll(uid, 0).size(), equalTo(2));
+    }
 }