mirror of
https://github.com/facefusion/facefusion.git
synced 2026-09-15 20:15:28 +02:00
* 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>
65 lines
1.6 KiB
Python
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' ]
|
|
|