Session aware stores (#1235)

* session aware process manager via store creator

* remove unused clear_session_id

* restore original scopes

* stateless test helpers

* more updates

* more updates

* more updates

* more updates

---------

Co-authored-by: Claude Fable 5.1 <noreply@anthropic.com>
This commit is contained in:
Henry Ruhs
2026-09-11 16:48:23 +02:00
committed by GitHub
co-authored by Claude Fable 5.1
parent 5bc9c80c55
commit c358ba7cf6
14 changed files with 211 additions and 25 deletions
+64
View File
@@ -0,0 +1,64 @@
from facefusion.store_creator import create_store, delete_content, get_content, has_content, init_content, set_content
def test_create_store() -> None:
store = create_store({ 'total': 0 })
assert store.get('__init__') == { 'total': 0 }
assert store.get('content_set') == {}
def test_init_content() -> None:
store = create_store({ 'total': 0 })
init_content(store, 'session-a')
assert store.get('content_set').get('session-a') == { 'total': 0 }
def test_has_content() -> None:
store = create_store({ 'total': 0 })
assert has_content(store, 'session-a') is False
init_content(store, 'session-a')
assert has_content(store, 'session-a') is True
assert has_content(store, 'session-b') is False
def test_get_content() -> None:
store = create_store({ 'total': 0 })
init_content(store, 'session-a')
assert get_content(store, 'session-a') == { 'total': 0 }
assert get_content(store, 'session-b') is None
get_content(store, 'session-a')['total'] += 1
assert get_content(store, 'session-a') == { 'total': 1 }
def test_set_content() -> None:
store = create_store({ 'total': 0 })
set_content(store, 'session-a', { 'total': 3 })
assert get_content(store, 'session-a') == { 'total': 3 }
assert has_content(store, 'session-b') is False
def test_delete_content() -> None:
store = create_store({ 'total': 0 })
init_content(store, 'session-a')
init_content(store, 'session-b')
delete_content(store, 'session-a')
assert list(store.get('content_set').keys()) == [ 'session-b' ]
delete_content(store, 'session-a')
assert list(store.get('content_set').keys()) == [ 'session-b' ]