split api and cli sessions (#1244)

Claude-Session: https://claude.ai/code/session_014VuiA4Yfhy78naZK5UtfX5

Co-authored-by: Claude Fable 5.1 <noreply@anthropic.com>

tests/test_session_manager.py

@@ -1,11 +1,11 @@
 import secrets
-from datetime import datetime, timedelta
+from datetime import timedelta
 from typing import Iterator
 
 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
+from facefusion.session_manager import clear_api_session, clear_cli_session, create_api_session, create_cli_session, find_api_session_id, fork_session, get_api_session, get_cli_session, join_session, resolve_owner_id, set_api_session, set_cli_session, validate_api_session
 
 
 @pytest.fixture(scope = 'function', autouse = True)
@@ -25,77 +25,112 @@ def test_fork_session() -> None:
 
 	assert get_session_id() == fork_id
 	assert resolve_owner_id() == local_id
-	assert get_session(fork_id).get('owner_id') == local_id
-	assert get_session(fork_id).get('access_token') is None
-	assert get_session(fork_id).get('expires_at') is None
+	assert get_cli_session(fork_id).get('owner_id') == local_id
 
 	join_session()
 
 
-def test_get_and_set_session() -> None:
-	session = create_session()
-	session_id = secrets.token_urlsafe(16)
-
-	set_session(session_id, session)
-
-	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)
-
-	set_session(session_id, session)
-
-	assert validate_session(session_id) is True
-
-	set_session(session_id,
-	{
-		'access_token': session.get('access_token'),
-		'refresh_token': session.get('refresh_token'),
-		'created_at': session.get('created_at'),
-		'expires_at': session.get('expires_at') - timedelta(hours = 1)
-	})
-
-	assert validate_session(session_id) is False
-
-
-def test_clear_session() -> None:
-	session = create_session()
-	session_id = secrets.token_urlsafe(16)
-
-	set_session(session_id, session)
-
-	assert validate_session(session_id) is True
-
-	clear_session(session_id)
-
-	assert validate_session(session_id) is None
-
-
 def test_join_session() -> None:
 	local_id = resolve_local_id()
 	fork_id = fork_session()
 	join_session()
 
 	assert get_session_id() == local_id
-	assert get_session(fork_id) is None
+	assert get_cli_session(fork_id) is None
+
+
+def test_get_and_set_api_session() -> None:
+	api_session = create_api_session()
+	session_id = secrets.token_urlsafe(16)
+
+	set_api_session(session_id, api_session)
+
+	assert get_api_session(session_id) == api_session
+
+	clear_api_session(session_id)
+
+
+def test_get_and_set_cli_session() -> None:
+	local_id = resolve_local_id()
+	cli_session = create_cli_session()
+	session_id = secrets.token_urlsafe(16)
+
+	set_cli_session(session_id, cli_session)
+
+	assert get_cli_session(session_id) == cli_session
+	assert get_cli_session(session_id).get('owner_id') == local_id
+
+	clear_cli_session(session_id)
+
+
+def test_find_api_session_id() -> None:
+	api_session = create_api_session()
+	session_id = secrets.token_urlsafe(16)
+
+	set_api_session(session_id, api_session)
+
+	assert find_api_session_id(api_session.get('access_token')) == session_id
+	assert find_api_session_id('INVALID') is None
+
+	clear_api_session(session_id)
+
+
+def test_resolve_owner_id() -> None:
+	local_id = resolve_local_id()
+
+	assert resolve_owner_id() == local_id
+
+	set_api_session('session-a', create_api_session())
+	set_session_id('session-a')
+
+	assert resolve_owner_id() == 'session-a'
+
+	fork_session()
+
+	assert resolve_owner_id() == 'session-a'
+
+	join_session()
+	clear_api_session('session-a')
+
+
+def test_validate_api_session() -> None:
+	api_session = create_api_session()
+	session_id = secrets.token_urlsafe(16)
+
+	set_api_session(session_id, api_session)
+
+	assert validate_api_session(session_id) is True
+
+	set_api_session(session_id,
+	{
+		'access_token': api_session.get('access_token'),
+		'refresh_token': api_session.get('refresh_token'),
+		'created_at': api_session.get('created_at'),
+		'expires_at': api_session.get('expires_at') - timedelta(hours = 1)
+	})
+
+	assert validate_api_session(session_id) is False
+
+	clear_api_session(session_id)
+
+	assert validate_api_session(session_id) is False
+
+
+def test_clear_api_session() -> None:
+	api_session = create_api_session()
+	session_id = secrets.token_urlsafe(16)
+
+	set_api_session(session_id, api_session)
+	clear_api_session(session_id)
+
+	assert get_api_session(session_id) is None
+
+
+def test_clear_cli_session() -> None:
+	cli_session = create_cli_session()
+	session_id = secrets.token_urlsafe(16)
+
+	set_cli_session(session_id, cli_session)
+	clear_cli_session(session_id)
+
+	assert get_cli_session(session_id) is None