Add test

Author: roznawsk

fishjam/api/_fishjam_client.py

@@ -26,8 +26,8 @@
     PeerConfig,
     PeerDetailsResponse,
     PeerOptionsAgent,
-    PeerOptionsWebRTC,
     PeerOptionsVapi,
+    PeerOptionsWebRTC,
     PeerRefreshTokenResponse,
     PeerType,
     RoomConfig,

scripts.py

@@ -24,6 +24,7 @@ def check_exit_code(command):
     if exit_code != 0:
         sys.exit(exit_code)
 
+
 def run_tests():
     check_exit_code("uv run pytest")
 

tests/test_room_api.py

@@ -7,6 +7,7 @@
     FishjamClient,
     Peer,
     PeerOptions,
+    PeerOptionsVapi,
     Room,
     RoomOptions,
 )
@@ -251,6 +252,27 @@ def test_peer_limit_reached(self, room_api: FishjamClient):
             room_api.create_peer(room.id)
 
 
+class TestCreateVapiAgent:
+    def test_create_vapi_agent(self, room_api: FishjamClient):
+        room = room_api.create_room()
+        options = PeerOptionsVapi(api_key="test_api_key", call_id="test_call_id")
+        peer = room_api.create_vapi_agent(room.id, options=options)
+
+        assert peer.type_ == PeerType.VAPI
+
+        room = room_api.get_room(room.id)
+        assert any(p.id == peer.id for p in room.peers)
+
+    def test_delete_vapi_agent(self, room_api: FishjamClient):
+        room = room_api.create_room()
+        options = PeerOptionsVapi(api_key="test_api_key", call_id="test_call_id")
+        peer = room_api.create_vapi_agent(room.id, options=options)
+
+        room_api.delete_peer(room.id, peer.id)
+
+        assert [] == room_api.get_room(room.id).peers
+
+
 class TestDeletePeer:
     def test_valid(self, room_api: FishjamClient):
         room = room_api.create_room()