Files
facefusion/tests/test_store_creator.py
Henry RuhsandClaude Fable 5.1 c358ba7cf6 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>
2026-09-11 16:48:23 +02:00

65 lines
1.6 KiB
Python

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' ]