fix websockets exception in unit tests #nolog

miguelgrinberg · Jul 29, 2019 · e4709df · e4709df
1 parent 7548f70
commit e4709df
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/asyncio/test_asyncio_client.py b/tests/asyncio/test_asyncio_client.py
@@ -423,7 +423,7 @@ def test_polling_connection_not_upgraded(self):
 
     @mock.patch('engineio.client.time.time', return_value=123.456)
     @mock.patch('engineio.asyncio_client.websockets.connect', new=AsyncMock(
-        side_effect=[websockets.exceptions.InvalidURI]))
+        side_effect=[websockets.exceptions.InvalidURI('foo')]))
     def test_websocket_connection_failed(self, _time):
         c = asyncio_client.AsyncClient()
         self.assertRaises(
@@ -436,7 +436,7 @@ def test_websocket_connection_failed(self, _time):
 
     @mock.patch('engineio.client.time.time', return_value=123.456)
     @mock.patch('engineio.asyncio_client.websockets.connect', new=AsyncMock(
-        side_effect=[websockets.exceptions.InvalidURI]))
+        side_effect=[websockets.exceptions.InvalidURI('foo')]))
     def test_websocket_upgrade_failed(self, _time):
         c = asyncio_client.AsyncClient()
         c.sid = '123'