update xmpp status api

1 files changed, 22 insertions, 0 deletions

diff --git a/juick-server/src/test/java/com/juick/server/tests/ServerTests.java b/juick-server/src/test/java/com/juick/server/tests/ServerTests.java
index 297a3c88..abeb7424 100644
--- a/juick-server/src/test/java/com/juick/server/tests/ServerTests.java
+++ b/juick-server/src/test/java/com/juick/server/tests/ServerTests.java
@@ -28,6 +28,8 @@ import com.juick.server.helpers.CommandResult;
 import com.juick.server.helpers.TagStats;
 import com.juick.server.util.HttpUtils;
 import com.juick.server.util.ImageUtils;
+import com.juick.server.xmpp.helpers.XMPPStatus;
+import com.juick.server.xmpp.s2s.ConnectionIn;
 import com.juick.service.*;
 import com.juick.util.DateFormattersHolder;
 import com.juick.util.MessageUtils;
@@ -77,6 +79,7 @@ import javax.xml.parsers.DocumentBuilder;
 import javax.xml.parsers.DocumentBuilderFactory;
 import javax.xml.parsers.ParserConfigurationException;
 import java.io.*;
+import java.net.Socket;
 import java.net.URI;
 import java.net.URISyntaxException;
 import java.nio.file.*;
@@ -85,6 +88,7 @@ import java.time.Instant;
 import java.util.*;
 import java.util.function.BiFunction;
 import java.util.function.Function;
+import java.util.function.Supplier;
 import java.util.stream.IntStream;
 
 import static org.hamcrest.MatcherAssert.assertThat;
@@ -1203,4 +1207,22 @@ public class ServerTests {
         messagesService.recommendMessage(readerMid, recommenderId);
         assertThat(subscribers.apply(recommenderId, messagesService.getMessage(readerMid)).size(), is(0));
     }
+    @Test
+    public void xmppStatusApi() throws Exception {
+        Supplier<XMPPStatus> getStatus = () -> {
+            try {
+                MvcResult result = mockMvc.perform(get("/xmpp/status").with(httpBasic(ugnichName, ugnichPassword)))
+                        .andExpect(status().isOk()).andReturn();
+                return jsonMapper.readValue(result.getResponse().getContentAsString(), XMPPStatus.class);
+            } catch (Exception e) {
+                e.printStackTrace();
+                return null;
+            }
+        };
+        assertThat(getStatus.get().getInbound().size(), is(0));
+        ConnectionIn test = new ConnectionIn(server, new Socket("localhost", server.getServerPort()));
+        test.from.add(Jid.of("test"));
+        server.addConnectionIn(test);
+        assertThat(getStatus.get().getInbound().size(), is(1));
+    }
 }