Update old tests

Author: AHGIJMKLKKZNPJKQR

tests/test_notifier.py

@@ -8,6 +8,7 @@
 
 import pytest
 import requests
+import websockets
 
 from fishjam import FishjamClient, FishjamNotifier, RoomOptions
 from fishjam.events import (
@@ -68,8 +69,9 @@ def handle_notitifcation(_notification):
 
         notifier_task = asyncio.create_task(notifier.connect())
         await notifier.wait_ready()
-        # pylint: disable=protected-access
-        assert notifier._websocket.open
+        assert (
+            notifier._websocket and notifier._websocket.state == websockets.State.OPEN
+        )
 
         await cancel(notifier_task)
 

tests/test_room_api.py

@@ -58,7 +58,7 @@ def room_api():
 
 
 class TestCreateRoom:
-    def test_no_params(self, room_api):
+    def test_no_params(self, room_api: FishjamClient):
         room = room_api.create_room()
 
         config = RoomConfig(
@@ -71,6 +71,7 @@ def test_no_params(self, room_api):
         config.__setitem__(
             "peerlessPurgeTimeout", room.config.__getitem__("peerlessPurgeTimeout")
         )
+        config.__setitem__("geoLoc", room.config.__getitem__("geoLoc"))
 
         assert room == Room(
             config=config,
@@ -98,6 +99,7 @@ def test_valid_params(self, room_api):
         config.__setitem__(
             "peerlessPurgeTimeout", room.config.__getitem__("peerlessPurgeTimeout")
         )
+        config.__setitem__("geoLoc", room.config.__getitem__("geoLoc"))
 
         assert room == Room(
             config=config,
@@ -158,6 +160,7 @@ def test_valid(self, room_api: FishjamClient):
         config.__setitem__(
             "peerlessPurgeTimeout", room.config.__getitem__("peerlessPurgeTimeout")
         )
+        config.__setitem__("geoLoc", room.config.__getitem__("geoLoc"))
 
         assert Room(
             peers=[],