fork and join sessions (#1239)

* introduce fork and join session, fix inference manager race condition

* destroy stream on DELETE

tests/test_session_manager.py

@@ -1,7 +1,35 @@
 import secrets
-from datetime import timedelta
+from datetime import datetime, timedelta
+from typing import Iterator
 
-from facefusion.session_manager import clear_session, create_session, get_session, set_session, validate_session
+import pytest
+
+from facefusion.session_context import get_session_id, resolve_local_id, set_session_id
+from facefusion.session_manager import clear_session, create_session, fork_session, get_session, join_session, resolve_owner_id, set_session, validate_session
+
+
+@pytest.fixture(scope = 'function', autouse = True)
+def before_each() -> Iterator[None]:
+	local_id = resolve_local_id()
+
+	set_session_id(local_id)
+
+	yield
+
+	set_session_id(local_id)
+
+
+def test_fork_session() -> None:
+	local_id = resolve_local_id()
+	session_id = fork_session()
+
+	assert get_session_id() == session_id
+	assert resolve_owner_id() == local_id
+	assert get_session(session_id).get('owner_id') == local_id
+	assert get_session(session_id).get('access_token') is None
+	assert get_session(session_id).get('expires_at') is None
+
+	join_session()
 
 
 def test_get_and_set_session() -> None:
@@ -13,6 +41,25 @@ def test_get_and_set_session() -> None:
 	assert get_session(session_id) == session
 
 
+def test_resolve_owner_id() -> None:
+	local_id = resolve_local_id()
+
+	assert resolve_owner_id() == local_id
+
+	set_session('session-a', create_session())
+	set_session('session-a1', { 'owner_id': 'session-a', 'created_at': datetime.now() })
+	set_session_id('session-a')
+
+	assert resolve_owner_id() == 'session-a'
+
+	set_session_id('session-a1')
+
+	assert resolve_owner_id() == 'session-a'
+
+	clear_session('session-a1')
+	clear_session('session-a')
+
+
 def test_validate_session() -> None:
 	session = create_session()
 	session_id = secrets.token_urlsafe(16)
@@ -43,3 +90,12 @@ def test_clear_session() -> None:
 	clear_session(session_id)
 
 	assert validate_session(session_id) is None
+
+
+def test_join_session() -> None:
+	local_id = resolve_local_id()
+	session_id = fork_session()
+	join_session()
+
+	assert get_session_id() == local_id
+	assert get_session(session_id) is None