mirror of
https://github.com/facefusion/facefusion.git
synced 2026-09-15 20:15:28 +02:00
convert content store (#1236)
* convert content store * add local_id variables * add local_id variables * fix misconcept of clear() in stores, pass session id to threads
This commit is contained in:
@@ -1,61 +1,60 @@
|
||||
import pytest
|
||||
|
||||
from facefusion.process_manager import clear_process_state, end, get_process_state, init_process_state, is_pending, is_processing, is_stopping, set_process_state, start, stop
|
||||
from facefusion.session_context import set_session_id
|
||||
from facefusion import session_context
|
||||
from facefusion.process_manager import clear, end, get_state, init, is_pending, is_processing, is_stopping, set_state, start, stop
|
||||
|
||||
|
||||
@pytest.fixture(scope = 'function', autouse = True)
|
||||
def before_each() -> None:
|
||||
set_session_id('session-a')
|
||||
clear_process_state()
|
||||
set_session_id('session-b')
|
||||
clear_process_state()
|
||||
set_session_id('session-a')
|
||||
session_context.set_session_id('session-a')
|
||||
clear()
|
||||
session_context.set_session_id('session-b')
|
||||
clear()
|
||||
session_context.set_session_id('session-a')
|
||||
|
||||
|
||||
def test_init_process_state() -> None:
|
||||
assert get_process_state() is None
|
||||
def test_init() -> None:
|
||||
set_state('processing')
|
||||
init()
|
||||
|
||||
init_process_state()
|
||||
|
||||
assert get_process_state() == 'pending'
|
||||
assert get_state() == 'pending'
|
||||
|
||||
|
||||
def test_get_process_state() -> None:
|
||||
set_process_state('processing')
|
||||
set_session_id('session-b')
|
||||
set_process_state('stopping')
|
||||
def test_get_state() -> None:
|
||||
set_state('processing')
|
||||
session_context.set_session_id('session-b')
|
||||
set_state('stopping')
|
||||
|
||||
assert get_process_state() == 'stopping'
|
||||
assert get_state() == 'stopping'
|
||||
|
||||
set_session_id('session-a')
|
||||
session_context.set_session_id('session-a')
|
||||
|
||||
assert get_process_state() == 'processing'
|
||||
|
||||
|
||||
def test_clear_process_state() -> None:
|
||||
set_process_state('processing')
|
||||
clear_process_state()
|
||||
|
||||
assert get_process_state() is None
|
||||
assert get_state() == 'processing'
|
||||
|
||||
|
||||
def test_start() -> None:
|
||||
set_process_state('pending')
|
||||
set_state('pending')
|
||||
start()
|
||||
|
||||
assert is_processing()
|
||||
|
||||
|
||||
def test_stop() -> None:
|
||||
set_process_state('processing')
|
||||
set_state('processing')
|
||||
stop()
|
||||
|
||||
assert is_stopping()
|
||||
|
||||
|
||||
def test_end() -> None:
|
||||
set_process_state('processing')
|
||||
set_state('processing')
|
||||
end()
|
||||
|
||||
assert is_pending()
|
||||
|
||||
|
||||
def test_clear() -> None:
|
||||
set_state('processing')
|
||||
clear()
|
||||
|
||||
assert get_state() == 'pending'
|
||||
|
||||
Reference in New Issue
Block a user