1 files changed, 5 insertions, 6 deletions
diff --git a/example/websockets/app/test/src/ExampleTests.scala b/example/websockets/app/test/src/ExampleTests.scala
index e8889b0..413937d 100644
--- a/example/websockets/app/test/src/ExampleTests.scala
+++ b/example/websockets/app/test/src/ExampleTests.scala
@@ -39,14 +39,13 @@ 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
+        val ex = intercept[Exception] {
+          cask.WsClient.connect("ws://localhost:8080/connect/nobody") {
+            case _ => /*do nothing*/
+          }
         }
+        assert(ex.getMessage.contains("403"))
 
-        assert(error.contains("403"))
       }finally ws.close()
     }