convert rtc store (#1242)

tests/test_api_stream_manager.py

@@ -36,7 +36,7 @@ def before_all() -> None:
 def before_each() -> Iterator[None]:
 	local_id = resolve_local_id()
 
-	rtc_store.clear()
+	rtc_store.delete_peer()
 
 	yield
 
@@ -117,14 +117,15 @@ def test_process_video(video_codec : VideoCodec, session_id : str) -> None:
 	datachannel_module.create_static_library().rtcDeletePeerConnection(peer_connection)
 
 	with patch('facefusion.apis.stream_manager.threading.Thread'):
-		sdp_answer = process_video(session_id, sdp_offer)
+		set_session_id(session_id)
+		sdp_answer = process_video(sdp_offer)
 
 	assert sdp_answer
 	assert 'm=video' in sdp_answer
 	assert 'a=recvonly' in sdp_answer
 	assert 'a=sendonly' in sdp_answer
 
-	rtc_peer = rtc_store.get_peer(session_id)
+	rtc_peer = rtc_store.get_peer()
 	sender_bitrate = rtc_peer.get('sender_bitrate')
 	receiver_bitrate = rtc_peer.get('receiver_bitrate')
 
@@ -156,22 +157,22 @@ def test_run_peer_loop(video_codec : VideoCodec, payload_type : int, session_id
 		'receiver_bitrate': ctypes.c_uint(0)
 	}
 
-	rtc_store.set_peer(session_id, rtc_peer)
 	store_creator.set_content(state_manager.STATE_SET, session_id, state_manager.get_state())
 	set_session_id(session_id)
+	rtc_store.set_peer(rtc_peer)
 
-	assert rtc_store.has_peer(session_id) is True
+	assert rtc_store.has_peer() is True
 
 	with patch('facefusion.thread_helper.ThreadPoolExecutor') as thread_pool_executor_mock:
 		with patch('facefusion.apis.stream_manager.receive_video_frames'):
 			with patch('facefusion.apis.stream_manager.run_video_encode_loop'):
-				thread = threading.Thread(target = copy_context().run, args = (run_peer_loop, session_id, rtc_peer), daemon = True)
+				thread = threading.Thread(target = copy_context().run, args = (run_peer_loop, rtc_peer), daemon = True)
 				thread.start()
 				thread.join(timeout = 5.0)
 
 	thread_pool_executor_mock.assert_called_once_with(max_workers = 8, initializer = set_session_id, initargs = tuple([ session_id ]))
 
-	assert rtc_store.has_peer(session_id) is False
+	assert rtc_store.has_peer() is False
 
 
 def test_destroy_stream() -> None:
@@ -189,11 +190,11 @@ def test_destroy_stream() -> None:
 		'sender_bitrate': ctypes.c_uint(0),
 		'receiver_bitrate': ctypes.c_uint(0)
 	}
-	session_id = 'test-destroy-stream'
 
-	rtc_store.set_peer(session_id, rtc_peer)
+	set_session_id('test-destroy-stream')
+	rtc_store.set_peer(rtc_peer)
 
-	assert destroy_stream(session_id) is True
-	assert rtc_store.get_peer(session_id) is None
+	assert destroy_stream() is True
+	assert rtc_store.get_peer() is None
 
-	assert destroy_stream(session_id) is False
+	assert destroy_stream() is False