From 6cfe7cf51316dc10dbabd9cdfc10a00e506e3746 Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Sat, 12 Sep 2026 17:11:35 +0200
Subject: [PATCH] fix tests

---
 tests/test_rtc_store.py | 112 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 112 insertions(+)
 create mode 100644 tests/test_rtc_store.py

diff --git a/tests/test_rtc_store.py b/tests/test_rtc_store.py
new file mode 100644
index 00000000..09d15ff5
--- /dev/null
+++ b/tests/test_rtc_store.py
@@ -0,0 +1,112 @@
+import ctypes
+from typing import Iterator
+
+import pytest
+
+from facefusion import rtc, session_context, session_manager, state_manager
+from facefusion.libraries import datachannel as datachannel_module
+from facefusion.rtc_store import delete_peer, get_peer, has_peer, init, set_peer
+from facefusion.types import RtcPeer
+
+
+@pytest.fixture(scope = 'module', autouse = True)
+def before_all() -> None:
+	state_manager.init()
+	state_manager.init_item('download_providers', [ 'github', 'huggingface' ])
+
+	datachannel_module.pre_check()
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> Iterator[None]:
+	local_id = session_context.resolve_local_id()
+
+	session_context.set_session_id(local_id)
+	init()
+
+	yield
+
+	delete_peer()
+	session_context.set_session_id(local_id)
+
+
+def create_rtc_peer() -> RtcPeer:
+	peer_connection = rtc.create_peer_connection()
+	rtc_peer : RtcPeer =\
+	{
+		'peer_connection': peer_connection,
+		'video':
+		{
+			'sender_track': rtc.add_video_track(peer_connection, 'sendonly', 'vp8', 96),
+			'receiver_track': 0,
+			'codec': 'vp8'
+		},
+		'sender_bitrate': ctypes.c_uint(0),
+		'receiver_bitrate': ctypes.c_uint(0)
+	}
+
+	return rtc_peer
+
+
+def test_init() -> None:
+	local_id = session_context.resolve_local_id()
+	rtc_peer = create_rtc_peer()
+
+	set_peer(rtc_peer)
+	session_context.set_session_id('session-a')
+	init()
+
+	assert has_peer() is False
+
+	session_manager.fork_session()
+
+	assert has_peer() is False
+
+	session_manager.join_session()
+	session_context.set_session_id(local_id)
+
+	assert get_peer() is rtc_peer
+
+	session_context.set_session_id('session-a')
+	set_peer(create_rtc_peer())
+	session_manager.fork_session()
+
+	assert has_peer() is True
+
+	session_manager.join_session()
+	delete_peer()
+	session_context.set_session_id(local_id)
+
+
+def test_has_peer() -> None:
+	assert has_peer() is False
+
+	set_peer(create_rtc_peer())
+
+	assert has_peer() is True
+
+
+def test_get_peer() -> None:
+	rtc_peer = create_rtc_peer()
+
+	assert get_peer() is None
+
+	set_peer(rtc_peer)
+
+	assert get_peer() is rtc_peer
+
+
+def test_set_peer() -> None:
+	rtc_peer = create_rtc_peer()
+
+	set_peer(rtc_peer)
+
+	assert get_peer() is rtc_peer
+
+
+def test_delete_peer() -> None:
+	set_peer(create_rtc_peer())
+	delete_peer()
+
+	assert has_peer() is False
+	assert get_peer() is None