api: message entities

1 files changed, 12 insertions, 0 deletions

diff --git a/src/test/java/com/juick/MessageTest.java b/src/test/java/com/juick/MessageTest.java
index 6197f861..f76054cd 100644
--- a/src/test/java/com/juick/MessageTest.java
+++ b/src/test/java/com/juick/MessageTest.java
@@ -17,12 +17,17 @@
 
 package com.juick;
 
+import com.juick.model.Entity;
+import com.juick.test.util.MockUtils;
 import com.juick.util.MessageUtils;
 import org.apache.commons.lang3.RandomUtils;
 import org.apache.commons.lang3.StringUtils;
 import org.junit.Test;
 
+import java.util.ArrayList;
 import java.util.List;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
 
 import static org.hamcrest.MatcherAssert.assertThat;
 import static org.hamcrest.Matchers.*;
@@ -180,4 +185,11 @@ public class MessageTest {
         msg.setText("And dick is @ugnich@jabber.zp.ua");
         assertThat(MessageUtils.getGlobalMentions(msg).size(), is(1));
     }
+    @Test
+    public void entitiesTest() {
+        String msg = "http://google.com - there will @ugnich ask questions from #4321, then go to http://stackoverflow.com";
+        Message testMessage = MockUtils.mockMessage(514, MockUtils.mockUser(5432, "fmap", "secret"), msg);
+        List<Entity> entities = MessageUtils.getEntities(testMessage);
+        assertThat(entities.size(), is(4));
+    }
 }