update test related to ping fix

websocket-client · Jun 17, 2023 · ec4fb07 · ec4fb07
1 parent 323baea
commit ec4fb07
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/websocket/tests/test_app.py b/websocket/tests/test_app.py
@@ -310,8 +310,8 @@ def on_pong(app, msg):
         app.run_forever(ping_interval=2, ping_timeout=1, reconnect=3)
 
         self.assertEqual(pong_count, 2)
-        self.assertIsInstance(exc, ValueError)
-        self.assertEqual(str(exc), "Invalid file object: None")
+        self.assertIsInstance(exc, ws.WebSocketTimeoutException)
+        self.assertEqual(str(exc), "ping/pong timed out")
 
 
 if __name__ == "__main__":