1 files changed, 6 insertions, 8 deletions
diff --git a/example/websockets2/app/test/src/ExampleTests.scala b/example/websockets2/app/test/src/ExampleTests.scala
index 11796ed..61038c5 100644
--- a/example/websockets2/app/test/src/ExampleTests.scala
+++ b/example/websockets2/app/test/src/ExampleTests.scala
@@ -36,14 +36,12 @@ object ExampleTests extends TestSuite{
         Thread.sleep(100)
         out ==> List("haoyi world", "haoyi hello")
 
-        var error: String = ""
-        val ws2 = cask.WsClient.connect("ws://localhost:8080/connect/nobody") {
-          case cask.Ws.Text(s) => out = s :: out
-          case cask.Ws.Error(t) => error += t.toString
-          case cask.Ws.Close(code, reason) => error += reason
-        }
-
-        assert(error.contains("403"))
+        val ex = intercept[Exception](
+          cask.WsClient.connect("ws://localhost:8080/connect/nobody") {
+            case _ => /*do nothing*/
+          }
+        )
+        assert(ex.getMessage.contains("403"))
       }finally ws.close()
     }