Jelajahi Sumber

调整tests

zhouhao 8 tahun lalu
induk
melakukan
97ded6383f

+ 6 - 1
hsweb-message/hsweb-message-api/src/test/java/org/hswebframework/web/message/SimpleMessageTests.java

@@ -18,6 +18,7 @@
 
 package org.hswebframework.web.message;
 
+import org.hswebframework.web.message.support.TextMessage;
 import org.junit.Assert;
 import org.junit.Test;
 
@@ -33,12 +34,16 @@ public class SimpleMessageTests {
     @Test
     public void simpleTest() throws InterruptedException {
         byte[] stat = new byte[1];
-        messager.subscribe(user("test"))
+
+       messager.subscribe(user("test"))
+                .onMessage(System.out::println)
                 .onMessage(msg -> stat[0] = 1);
+
         messager.publish(text("hello2"))
                 .to(user("test"))
                 .from(user("admin"))
                 .send();
+
         Thread.sleep(1000);
         Assert.assertEquals(stat[0], 1);
     }

+ 11 - 1
hsweb-message/hsweb-message-api/src/test/java/org/hswebframework/web/message/builder/SimpleMessageBuilder.java

@@ -31,7 +31,17 @@ import org.hswebframework.web.message.support.TextMessage;
 public class SimpleMessageBuilder implements MessageBuilder {
     @Override
     public TextMessage text(String msg) {
-        return (TextMessage) () -> msg;
+        return new TextMessage() {
+            @Override
+            public String getMessage() {
+                return msg;
+            }
+
+            @Override
+            public String toString() {
+                return msg;
+            }
+        };
     }
 
     @Override